accelerate>=1.1.1
datasets>=3.1.0
numpy>=1.26.4
rouge>=1.0.1
torch>=2.0.0
triton>=2.0.0
safetensors>=0.4.5
transformers>=4.46.2
tqdm>=4.67.0
threadpoolctl>=3.5.0
packaging>=24.2
setuptools>=75.5.0
device-smi>=0.0.1.dev0