requirements.txt

accelerate>=0.33.0 #git+https://github.com/huggingface/accelerate.git@main#egg=accelerate is suggested
diffusers>=0.30.1 #git+https://github.com/huggingface/diffusers.git@main#egg=diffusers is suggested
gradio>=4.42.0 # For HF gradio demo
imageio==2.34.2 # For diffusers inference export video
imageio-ffmpeg==0.5.1 # For diffusers inference export video
moviepy==1.0.3 # For export video
numpy==1.26.0
openai>=1.42.0 # For prompt refiner
pillow==9.5.0
sentencepiece>=0.2.0 # T5 used
streamlit>=1.38.0 # For streamlit web demo
SwissArmyTransformer>=0.4.12
# torch>=2.4.0 # Tested in 2.2 2.3 2.4 and 2.5, The development team is working on version 2.4.0.
# torchvision>=0.19.0 # The development team is working on version 0.19.0.
transformers>=4.44.2  # The development team is working on version 4.44.2
omegaconf==2.3.0
pytorch_lightning==2.5.0.post0
kornia==0.8.0
scipy==1.15.1
torchdiffeq==0.2.5
beartype==0.19.0
PyAV==14.1.0
av==14.1.0
wandb