Update requirements.txt
Browse files- requirements.txt +22 -3
requirements.txt
CHANGED
@@ -1,4 +1,23 @@
|
|
|
|
1 |
git+https://github.com/huggingface/diffusers.git@main
|
2 |
-
|
3 |
-
|
4 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
opencv-python==4.9.0.80
|
2 |
git+https://github.com/huggingface/diffusers.git@main
|
3 |
+
|
4 |
+
# using the 4.46.1 because we have this error:
|
5 |
+
# "huggingface-inference-toolkit 0.5.2 requires transformers[audio,sentencepiece,sklearn,vision]==4.46.1,
|
6 |
+
# but you have transformers 4.46.3 which is incompatible.""
|
7 |
+
#transformers==4.46.3
|
8 |
+
transformers==4.46.1
|
9 |
+
|
10 |
+
tokenizers==0.20.3
|
11 |
+
accelerate==1.1.1
|
12 |
+
pandas==2.0.3
|
13 |
+
numpy==1.24.4
|
14 |
+
einops==0.7.0
|
15 |
+
tqdm==4.66.2
|
16 |
+
loguru==0.7.2
|
17 |
+
imageio==2.34.0
|
18 |
+
imageio-ffmpeg==0.5.1
|
19 |
+
safetensors==0.4.3
|
20 |
+
|
21 |
+
# Install flash attention v2 for acceleration (requires CUDA 11.8 or above)
|
22 |
+
# python -m pip install git+https://github.com/Dao-AILab/[email protected]
|
23 |
+
#flash-attn==2.5.9.post1
|