--extra-index-url https://download.pytorch.org/whl/cu121
torch==2.1.1
torchvision==0.16.1
torchaudio==2.1.1
diffusers==0.24.0
imageio==2.36.0
imageio-ffmpeg
transformers==4.45.2
accelerate==1.0.1
opencv-python
gdown
einops
decord
omegaconf
safetensors
gradio==5.1.0
wandb
triton
huggingface_hub==0.25.2