requirements.txt

torch>=2.2
torchvision>=0.17
accelerate>=0.28.0
transformers>=4.40.0
diffusers>=0.31.0
safetensors>=0.4.2
sentencepiece>=0.2.0
gradio>=4.32.0
imageio>=2.34.0
imageio-ffmpeg>=0.4.9