| 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/flash-attention.git@v2.5.9.post1 | |
| #flash-attn==2.5.9.post1 |