Update requirements.txt
Browse files- 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
|