--index-url https://download.pytorch.org/whl/cu128
--extra-index-url https://pypi.org/simple

torch
huggingface-hub
transformers==4.45.1
numpy
hydra-core
hydra_colorlog
datasets==2.21.0
accelerate
bitsandbytes
rouge-score
scipy
tensorboard
scikit-learn
deepspeed
lm_eval==0.4.5
trl==0.11.4
