datasets>=2.5.2
numpy>=1.23.1
pandas>=1.4.4
scikit_learn>=1.1.1
scipy>=1.9.0
torch>=1.10.2
tqdm>=4.64.1
chugchug>=0.1.3
transformers>=4.22.1
accelerate>=0.20.1

[lora]
peft>=0.14.0
