DialogSumLlama2 / requirements.txt
shenoy's picture
Fixing dependencies
e2da97b
raw
history blame
96 Bytes
accelerate==0.21.0
peft==0.4.0
bitsandbytes==0.40.2
transformers==4.30.2
trl==0.4.7
scikit-learn