model-finetuning-with-own-data / requirements.txt
jonACE's picture
Update requirements.txt
fb8a2ec verified
raw
history blame
160 Bytes
PyMuPDF==1.21.1
numpy==1.26.4
transformers==4.31.0
datasets==2.13.0
torch==2.0.1
accelerate==0.21.0
peft==0.3.0
bitsandbytes==0.41.1
scipy==1.10.1
triton==2.0.0