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,
|