/ 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