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