jbilcke-hf HF Staff commited on
Commit
ba384df
·
verified ·
1 Parent(s): cf4050b

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +3 -0
requirements.txt CHANGED
@@ -5,6 +5,9 @@ diffusers==0.31.0
5
  transformers>=4.49.0
6
  tokenizers>=0.20.3
7
  accelerate>=1.1.1
 
 
 
8
  tqdm
9
  imageio
10
  easydict
 
5
  transformers>=4.49.0
6
  tokenizers>=0.20.3
7
  accelerate>=1.1.1
8
+
9
+ flash-attn @ https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.4.post1/flash_attn-2.7.4.post1+cu12torch2.7cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
10
+
11
  tqdm
12
  imageio
13
  easydict