/ repo3-fine-tuning-template / requirements.txt
requirements.txt
 1  torch>=2.0.0
 2  transformers>=4.36.0
 3  trl>=0.7.0
 4  datasets>=2.14.0
 5  accelerate>=0.24.0
 6  peft>=0.6.0
 7  bitsandbytes>=0.41.0
 8  wandb>=0.15.0
 9  tqdm>=4.65.0
10  numpy>=1.24.0
11  scipy>=1.10.0
12  scikit-learn>=1.3.0