Fabrice-TIERCELIN commited on
Commit
6817e9b
·
verified ·
1 Parent(s): 8395168

This Pull Request set a better parameter

Browse files

With 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_.

Files changed (1) hide show
  1. hyvideo/modules/attenion.py +1 -1
hyvideo/modules/attenion.py CHANGED
@@ -61,7 +61,7 @@ def attention(
61
  q,
62
  k,
63
  v,
64
- mode="flash",
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,