mirror of
https://github.com/huggingface/diffusers.git
synced 2026-01-29 07:22:12 +03:00
add 10 second models support
This commit is contained in:
@@ -361,7 +361,8 @@ class Kandinsky5SDPAAttentionProcessor(nn.Module):
|
||||
|
||||
class Kandinsky5NablaAttentionProcessor(nn.Module):
|
||||
"""Custom attention processor for Nabla attention"""
|
||||
|
||||
|
||||
@torch.compile(mode="max-autotune-no-cudagraphs", dynamic=True)
|
||||
def __call__(
|
||||
self,
|
||||
attn,
|
||||
|
||||
Reference in New Issue
Block a user