requirements.txt

pip==23.3.1
torch==2.1.2 --index-url https://download.pytorch.org/whl/cu121
torchvision==0.16.2 --index-url https://download.pytorch.org/whl/cu121
torchaudio==2.1.2 --index-url https://download.pytorch.org/whl/cu121
tqdm==4.66.1
mediapipe==0.10.8
Pillow==10.1.0
diffusers==0.18.2
transformers==4.30.2
fire==0.5.0
huggingface_hub==0.16.4
imageio==2.31.1
imageio-ffmpeg==0.4.9
PyYAML==6.0.1
safetensors==0.4.1
wandb==0.16.0
accelerate==0.24.1
triton==2.1.0
gradio==4.7.1
plyfile==0.8.1
# xformers==0.0.23.post1 --index-url https://download.pytorch.org/whl/cu121