requirements.txt

--extra-index-url https://download.pytorch.org/whl/cu121
torch==2.3.1+cu121
torchvision==0.18.1+cu121
transformers==4.45.2
datasets==2.20.0
accelerate==0.26.1
diffusers==0.30.3
timm==0.9.16
peft==0.9.0
safetensors==0.4.5
gradio==5.4.0
numpy==1.26.3