jbilcke-hf HF staff commited on
Commit
cd0ccad
1 Parent(s): e3dcab5

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +6 -1
requirements.txt CHANGED
@@ -17,4 +17,9 @@ loguru==0.7.2
17
  imageio==2.34.0
18
  imageio-ffmpeg==0.5.1
19
  safetensors==0.4.3
20
- gradio==4.43.0
 
 
 
 
 
 
17
  imageio==2.34.0
18
  imageio-ffmpeg==0.5.1
19
  safetensors==0.4.3
20
+ gradio==4.43.0
21
+
22
+ # taken from the HunyuanVideo README:
23
+ # Install flash attention v2 for acceleration (requires CUDA 11.8 or above)
24
+ # python -m pip install git+https://github.com/Dao-AILab/flash-attention.git@v2.5.9.post1
25
+ flash-attn==2.5.9.post1