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