requirements.txt

accelerate>=1.8.0
diffusers>=0.33.1
fastapi
ftfy
gradio==5.34.1
numpy==1.26.4
optimum
peft==0.6.0
pydantic==2.11.7
spaces
transformers >= 4.34.0
--extra-index-url https://download.pytorch.org/whl/cu121
torch==2.1.0
torchvision>=0.16.0
torchaudio>=2.1.0
xformers>=0.0.22.post7