Files
Open-Assistant/model/supervised_finetuning/requirements.txt
T
2023-01-10 20:16:02 -05:00

11 lines
174 B
Plaintext

accelerate==0.15.0
bitsandbytes==0.36.0.post2
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