| --extra-index-url https://download.pytorch.org/whl/cu128 | |
| --extra-index-url https://pypi.nvidia.com | |
| torch==2.8.0+cu128 | |
| torchvision==0.23.0+cu128 | |
| transformers==4.57.1 | |
| accelerate==1.12.0 | |
| flash-attn==2.8.1 | |
| torchcodec==0.7.0 | |
| numpy==2.4.3 | |
| pillow==12.1.1 | |
| joblib==1.5.2 | |
| einops==0.8.2 | |
| ninja==1.13.0 | |
| packaging==26.0 | |