Trust remote code (#3)

substratusai · Sep 4, 2024 · e6d24ba · e6d24ba
1 parent a02de08
commit e6d24ba
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/main.py b/main.py
@@ -40,7 +40,7 @@ class EmbeddingResponse(BaseModel):
 
 @asynccontextmanager
 async def lifespan(app: FastAPI):
-    models[model_name] = SentenceTransformer(model_name)
+    models[model_name] = SentenceTransformer(model_name, trust_remote_code=True)
     yield
 
 

diff --git a/requirements.txt b/requirements.txt
@@ -1,3 +1,4 @@
 fastapi
-sentence-transformers==2.2.2
+sentence-transformers==2.4.0
 uvicorn[standard]
+einops