fix(model): Fix embedding start error

This commit is contained in:
FangYin Cheng
2023-10-20 17:25:35 +08:00
parent af592823d0
commit b3b486bd24
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"]