CuMo-7b-zero / requirements.txt
jiachenl
update
c3f3b0b
raw
history blame
315 Bytes
torch==2.1.2
torchvision==0.16.2
transformers==4.37.2
tokenizers==0.15.1
sentencepiece==0.1.99
shortuuid
accelerate==0.21.0
pydantic
markdown2[all]
numpy
scikit-learn
gradio==4.16.0
gradio_client==0.8.1
requests
httpx==0.23.3
uvicorn
fastapi
einops==0.6.1
einops-exts==0.0.4
deepspeed==0.13.4
ninja
wheel
setuptools