up

Author: tolgacangoz

src/diffusers/models/transformers/transformer_skyreels_v2.py

@@ -124,7 +124,7 @@ def apply_rotary_emb(
             key = apply_rotary_emb(key, *rotary_emb)
 
         if not attn.is_cross_attention:
-            attention_backend = "_native_flash"
+            attention_backend = "flash"
         else:
             attention_backend = "flash_varlen"