requirements.txt

torch==2.1.0
torchaudio==2.1.0
torchvision==0.16.0
accelerate==0.32.1
diffusers==0.29.2
einops==0.8.0
ftfy==6.2.0
imageio==2.34.2
iopath==0.1.10
pytorchvideo==0.1.5
timm==1.0.7
tqdm==4.66.4
transformers==4.42.4
gradio==4.39.0