danswer/backend/requirements/model_server.txt
Yuhong Sun 7433dddac3
Model Server (#695)
Provides the ability to pull out the NLP models into a separate model server which can then be hosted on a GPU instance if desired.
2023-11-06 16:36:09 -08:00

9 lines
151 B
Plaintext

fastapi==0.103.0
pydantic==1.10.7
safetensors==0.3.1
sentence-transformers==2.2.2
tensorflow==2.13.0
torch==2.0.1
transformers==4.30.1
uvicorn==0.21.1