--extra-index-url https://download.pytorch.org/whl/cu129
accelerate
datasets
numpy
safetensors>=0.4.0
torch>=2.6.0
transformers<5.0
