fine-tuning / requirements.txt
Hunzla's picture
Update requirements.txt
24effb1
raw
history blame
No virus
112 Bytes
accelerate==0.21.0
peft==0.4.0
bitsandbytes==0.40.2
transformers==4.31.0
trl==0.4.7
torch
datasets
gradio
scipy