| --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 | |
| numpy | |
| scipy | |
| Pillow | |
| sentencepiece | |
| protobuf | |
| triton | |
| pymongo | |
| simple_parsing | |
| timm |
| --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 | |
| numpy | |
| scipy | |
| Pillow | |
| sentencepiece | |
| protobuf | |
| triton | |
| pymongo | |
| simple_parsing | |
| timm |