Stabilises the run and fixes the finetuner error (#170)

* Stabilises the run and fixes the finetuner error * back out my changes --------- Co-authored-by: Eric Pugh <[email protected]>
querqy · Sep 20, 2024 · b1a6148 · b1a6148
1 parent 1905206
commit b1a6148
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/embeddings/app/main.py b/embeddings/app/main.py
@@ -1,11 +1,11 @@
 from fastapi import FastAPI
 from clipL14 import router as router_clip
 from minilm import router as router_minilm
-#from finetuned import router as router_finetuned
+from finetuned import router as router_finetuned
 
 app = FastAPI()
 
 app.include_router(router_clip.router, prefix="/clip")
 app.include_router(router_minilm.router, prefix="/minilm")
-#app.include_router(router_finetuned.sbert_router, prefix="/ftbert")
-#app.include_router(router_finetuned.clip_router, prefix="/ftclip")
+app.include_router(router_finetuned.sbert_router, prefix="/ftbert")
+app.include_router(router_finetuned.clip_router, prefix="/ftclip")
diff --git a/embeddings/requirements.txt b/embeddings/requirements.txt
@@ -1,9 +1,9 @@
-docarray==0.21.0 # See https://github.com/querqy/chorus/issues/169
 sentence-transformers
 fastapi
 uvicorn[standard]
 gunicorn
 pydantic
 torch
 git+https://github.com/openai/CLIP.git
-finetuner[full]
+finetuner[full]==0.7.8
+docarray[commons]==0.21.0
diff --git a/finetuning/requirements.txt b/finetuning/requirements.txt
@@ -1,4 +1,4 @@
-finetuner[full]=0.7.8
+finetuner[full]==0.7.8
 docarray[commons]==0.21.0
 click==8.1.3