diff --git a/src/diffusers/models/transformers/transformer_kandinsky.py b/src/diffusers/models/transformers/transformer_kandinsky.py index ac2fe58d60..8d2bae11cb 100644 --- a/src/diffusers/models/transformers/transformer_kandinsky.py +++ b/src/diffusers/models/transformers/transformer_kandinsky.py @@ -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,