--extra-index-url https://download.pytorch.org/whl/cu126
torch==2.6.0+cu126
torchvision
torchaudio
huggingface_hub
torchao==0.9.0
diffusers==0.32.2
peft
transformers<=4.48.3
numpy<2
scipy
Pillow
sentencepiece
protobuf
triton
schedule
para-attn==0.3.23