jbilcke-hf HF staff commited on
Commit
3f46a58
·
verified ·
1 Parent(s): 1a6f91c

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +22 -3
requirements.txt CHANGED
@@ -1,4 +1,23 @@
 
1
  git+https://github.com/huggingface/diffusers.git@main
2
- torch==2.2.0
3
- torchvision
4
- pillow==9.5.0
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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