requirements.txt 247 Bytes
Newer Older
silencealiang's avatar
silencealiang committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
einops
flask-restful
nltk
pytest
pytest_asyncio
pytest-cov
pytest_mock
pytest-random-order
sentencepiece
tiktoken
wrapt
zarr
wandb
triton==2.1.0
tensorstore!=0.1.46,!=0.1.72
nvidia-modelopt[torch]>=0.19.0; sys_platform != "darwin"