File size: 662 Bytes
626eca0 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 |
#------- Core dependencies ------- torch>=2.0 transformers[sentencepiece]>=4.34,<4.35 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 |