--extra-index-url https://download.pytorch.org/whl/cu124
torch>=2.1.0
torchvision
diffusers>=0.28.2
transformers==4.44.2
sentencepiece>=0.1.96
wheel==0.44.0
einops==0.8.0
accelerate==1.1.1
matplotlib
imageio[ffmpeg]
gradio==5.7.1
gradio_toggle==2.0.2
--extra-index-url https://download.pytorch.org/whl/nightly/cpu
torchao