relik-entity-linking / requirements.txt
riccorl's picture
Upload models
8197b11
#------- Core dependencies -------
torch>=2.0
transformers[sentencepiece]>=4.33,<4.34
rich>=13.0.0,<14.0.0
scikit-learn
overrides
#------- Optional dependencies -------
# train
lightning>=2.0,<2.1
hydra-core>=1.3,<1.4
hydra_colorlog
wandb>=0.15,<0.16
datasets>=2.13,<2.15
# faiss
faiss-cpu==1.7.4 # needed by: faiss
# serve
fastapi>=0.103,<0.104 # needed by: serve
uvicorn[standard]==0.23.2 # needed by: serve
gunicorn==21.2.0 # needed by: serve
ray[serve]>=2.7,<=2.8 # needed by: serve
ipa-core # needed by: serve
streamlit>=1.27,<1.28 # needed by: serve
streamlit_extras>=0.3,<0.4 # needed by: serve
# retriever
# reader
# dev
pre-commit
black[d]
isort