NEOX / requirements /requirements.txt
akswelh's picture
Upload 251 files
d90b3a8 verified
deepspeed@git+https://github.com/EleutherAI/DeeperSpeed.git@02e2ebf7dee6aaab3d89094ed470a4609763c742#egg=deepspeed
ftfy>=6.0.1
huggingface_hub>=0.11.0
jinja2==3.1.4
lm_dataformat@git+https://github.com/EleutherAI/lm_dataformat.git@4eec05349977071bf67fc072290b95e31c8dd836
lm_eval>=0.4.0,<=0.4.1
mpi4py>=3.0.3
numpy<2.0
pybind11>=2.6.2
regex
sentencepiece
six
tiktoken>=0.1.2
tokenizers>=0.12.1
transformers==4.38.0