requirements.txt

evaluate 
transformers 
scikit-learn 
accelerate 
rouge_score 
wandb 
jsonlines 
fire 
fraction 
vllm==0.4.0.post1 
bitsandbytes 
deepspeed
peft@git+https://github.com/huggingface/peft.git
torch-scatter torch-sparse -f https://data.pyg.org/whl/torch-2.1.2+cu121.html