sync with cht-llama-cpp

premAI-io · Nov 3, 2023 · a8e4707 · a8e4707
1 parent 0f310d6
commit a8e4707
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/ebd-all-minilm/models.py b/ebd-all-minilm/models.py
@@ -12,12 +12,10 @@ def embeddings(cls, texts):
         return values.tolist()
 
     @classmethod
-    def get_model(cls, model=None):
+    def get_model(cls, model_path):
         if cls.model is None:
-            if model is None or not os.path.exists(model):
-                model = os.getenv("MODEL_ID", "all-MiniLM-L6-v2")
             cls.model = SentenceTransformer(
-                model,
+                model_path,
                 device=os.getenv("DEVICE", "cpu"),
             )
         return cls.model