Update requirements.txt

requirements.txt

@@ -1,5 +1,5 @@
-#------- Core dependencies -------
-torch>=2.0
+--extra-index-url https://download.pytorch.org/whl/cu118
+torch==2.1.0
 transformers[sentencepiece]>=4.33,<4.34
 rich>=13.0.0,<14.0.0
 scikit-learn
@@ -21,15 +21,14 @@ datasets>=2.13,<2.15
 faiss-cpu==1.7.4 # needed by: faiss
 
 # serve
+fastapi>=0.104,<0.105 # needed by: serve, ray
+uvicorn[standard]==0.23.2 # needed by: serve, ray
+gunicorn==21.2.0 # needed by: serve, ray
 spacy>=3.7,<3.8 # needed by: serve, ray
-fastapi>=0.103,<0.104 # needed by: serve
-uvicorn[standard]==0.23.2 # needed by: serve
-gunicorn==21.2.0 # needed by: serve
+streamlit>=1.28,<1.29 # needed by: serve, ray
+streamlit_extras>=0.3,<0.4 # needed by: serve, ray
 pyvis # needed by: serve, ray
-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
+ray[serve]>=2.7,<=2.8 # needed by: ray
 
 # retriever