File size: 773 Bytes
50ae4c6
 
a1ccb04
45927c2
3f46a58
 
 
 
 
 
 
d79d5cb
 
 
06050fe
3f46a58
d79d5cb
 
 
 
93db5ee
3f46a58
425d81c
e9d66cc
3f46a58
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
diffusers @ git+https://github.com/huggingface/diffusers.git@main
varnish @ git+https://github.com/jbilcke-hf/varnish.git@main

opencv-python>=4.10.0.84

# using the 4.46.1 because we have this error:
# "huggingface-inference-toolkit 0.5.2 requires transformers[audio,sentencepiece,sklearn,vision]==4.46.1,
# but you have transformers 4.46.3 which is incompatible.""
#transformers==4.46.3
transformers==4.46.1

tokenizers>=0.20.3
accelerate>=1.1.1
pandas>=2.0.3
numpy
einops==0.7.0
tqdm>=4.66.5
loguru>=0.7.2
imageio>=2.34.2
imageio-ffmpeg>=0.5.1
safetensors>=0.4.5

moviepy==1.0.3

# Install flash attention v2 for acceleration (requires CUDA 11.8 or above)
# python -m pip install git+https://github.com/Dao-AILab/[email protected]
#flash-attn==2.5.9.post1