requirements.txt

# python.version == 3.11.13
--extra-index-url https://download.pytorch.org/whl/cu121

torch==2.5.1+cu121
torchvision==0.20.1+cu121
torchaudio==2.5.1+cu121

numpy==2.1.2
pandas==2.3.1
safetensors==0.5.3
transformers==4.53.3
deepspeed==0.17.2
dataset==1.6.2
datasets==4.0.0
protobuf==6.31.1
sentencepiece==0.2.0
accelerate==1.9.0