requirements.txt

--extra-index-url https://download.pytorch.org/whl/cu128
torch>=2.7
accelerate==0.34.2
bitsandbytes>=0.46.0
scipy==1.14.1
sentencepiece==0.2.0
transformers==4.56.2
diffusers>=0.35.0
einops==0.8.0
safetensors==0.4.5
huggingface_hub