mode="torch",
Browse files
hyvideo/modules/attenion.py
CHANGED
|
@@ -61,7 +61,7 @@ def attention(
|
|
| 61 |
q,
|
| 62 |
k,
|
| 63 |
v,
|
| 64 |
-
mode="
|
| 65 |
drop_rate=0,
|
| 66 |
attn_mask=None,
|
| 67 |
causal=False,
|
|
|
|
| 61 |
q,
|
| 62 |
k,
|
| 63 |
v,
|
| 64 |
+
mode="torch",
|
| 65 |
drop_rate=0,
|
| 66 |
attn_mask=None,
|
| 67 |
causal=False,
|