Files
Open-Assistant/model/supervised_finetuning/requirements.txt
T
Sotirios Anagnostidis d3952354e2 pre commits
2023-01-06 22:09:24 +01:00

10 lines
147 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