fix(model): Fix remote embedding model start error (#708)

eosphoros-ai · Oct 20, 2023 · d6285b5 · d6285b5
2 parents af59282 + b3b486b
commit d6285b5
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/pilot/model/cluster/manager_base.py b/pilot/model/cluster/manager_base.py
@@ -28,7 +28,11 @@ class WorkerRunData:
 
     def _to_print_key(self):
         model_name = self.model_params.model_name
-        model_type = self.model_params.model_type
+        model_type = (
+            self.model_params.model_type
+            if hasattr(self.model_params, "model_type")
+            else "text2vec"
+        )
         host = self.host
         port = self.port
         return f"model {model_name}@{model_type}({host}:{port})"

diff --git a/setup.py b/setup.py
@@ -316,6 +316,7 @@ def core_requires():
         "jsonschema",
         # TODO move transformers to default
         "transformers>=4.31.0",
+        "alembic==1.12.0",
     ]
 
 
@@ -424,7 +425,6 @@ def default_requires():
         "dashscope",
         "chardet",
         "GitPython",
-        "alembic==1.12.0",
     ]
     setup_spec.extras["default"] += setup_spec.extras["framework"]
     setup_spec.extras["default"] += setup_spec.extras["knowledge"]