Spaces:
Running
Running
This Pull Request set a better parameter
Browse filesWith this parameter, the space will corretly resolve the dependencies of the space
Click on _Merge_ to add this feature or synchronize your space by clicking on _Settings_ and _Synchronize_.
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,
|