--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