Spaces:

Fabrice-TIERCELIN
/

HunyuanVideo

Running

Fabrice-TIERCELIN commited on Apr 19

Commit

add0d35

verified ·

1 Parent(s): b3893eb

flash

Files changed (1) hide show

hyvideo/modules/attenion.py CHANGED Viewed

@@ -61,7 +61,7 @@ def attention(
     q,
     k,
     v,
-    mode="torch",
     drop_rate=0,
     attn_mask=None,
     causal=False,

     q,
     k,
     v,
+    mode="flash",
     drop_rate=0,
     attn_mask=None,
     causal=False,