| #------- 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 | |