requirements.txt

accelerate==0.33.0
einops==0.8.0
gradio==4.39.0
pillow==10.4.0
transformers==4.43.3
torchvision==0.17.0 --extra-index-url https://download.pytorch.org/whl/cu121