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

This commit is contained in:
Aries-ckt
2023-10-20 17:31:10 +08:00
committed by GitHub
2 changed files with 6 additions and 2 deletions

View File

@@ -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})"

View File

@@ -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"]