Files
Open-Assistant/model/supervised_finetuning/requirements.txt
T
2023-01-09 22:07:06 -05:00

10 lines
173 B
Plaintext

accelerate==0.15.0
datasets==2.8.0
deepspeed==0.7.7
mpi4py==3.1.4
numpy==1.23.0
PyYAML==6.0
scikit_learn==1.2.0
torch==1.13.1
transformers==4.25.1
bitsandbytes==0.36.0.post2