--extra-index-url https://download.pytorch.org/whl/cu121 torch==2.1.1 torchvision==0.16.1 torchaudio==2.1.1 imageio==2.36.0 xformers==0.0.23 numpy==1.24.4 imageio-ffmpeg transformers accelerate opencv-python gdown einops decord omegaconf safetensors gradio wandb triton termcolor huggingface_hub==0.25.2