CogVideo/requirements.txt
2024-08-19 16:47:51 +08:00

14 lines
590 B
Plaintext

git+https://github.com/huggingface/diffusers.git@main#egg=diffusers
transformers==4.44.0
git+https://github.com/huggingface/accelerate.git@main#egg=accelerate
sentencepiece==0.2.0 # T5
SwissArmyTransformer==0.4.12 # Inference
torch==2.4.0 # Tested in 2.2 2.3 2.4 and 2.5
torchvision==0.19.0
gradio==4.40.0 # For HF gradio demo
pillow==9.5.0 # For HF gradio demo
streamlit==1.37.0 # For streamlit web demo
imageio==2.34.2 # For diffusers inference export video
imageio-ffmpeg==0.5.1 # For diffusers inference export video
openai==1.40.6 # For prompt refiner
moviepy==1.0.3 # For export video