From b1a614820aab0abc28f679df7c829a7cf4156b60 Mon Sep 17 00:00:00 2001 From: Atita Arora Date: Fri, 20 Sep 2024 11:50:10 +0200 Subject: [PATCH] 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 --- embeddings/app/main.py | 6 +++--- embeddings/requirements.txt | 4 ++-- finetuning/requirements.txt | 2 +- 3 files changed, 6 insertions(+), 6 deletions(-) diff --git a/embeddings/app/main.py b/embeddings/app/main.py index f0da4eb..27ea6ff 100644 --- 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 index 56b84c3..3040f9b 100644 --- a/embeddings/requirements.txt +++ b/embeddings/requirements.txt @@ -1,4 +1,3 @@ -docarray==0.21.0 # See https://github.com/querqy/chorus/issues/169 sentence-transformers fastapi uvicorn[standard] @@ -6,4 +5,5 @@ 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 index 82d8e9e..8ea65f3 100644 --- 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