requirements.txt

accelerate==0.32.1
auto_gptq==0.7.1
bitsandbytes==0.43.1
deepspeed==0.14.4
diffusers==0.29.2
einops==0.8.0
flow_vis==0.1
gradio==3.50.2
huggingface_hub==0.23.4
imageio==2.34.2
matplotlib==3.9.1
numpy==2.0.0
omegaconf==2.3.0
packaging==24.1
Pillow==10.4.0
PyYAML==6.0.1
Requests==2.32.3
safetensors==0.4.3
scipy==1.14.0
torch==2.3.1
torch_xla==2.3.0
torchvision==0.18.1
tqdm==4.65.0
transformers==4.42.3
imageio-ffmpeg
av2
peft