github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-08-02 16:50:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Aries-ckt 2023-10-20 17:31:10 +08:00 committed by GitHub
commit d6285b53e4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
pilot/model/cluster/manager_base.py
View File

@ -28,7 +28,11 @@ class WorkerRunData:
def _to_print_key(self): def _to_print_key(self):
model_name = self.model_params.model_name 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 host = self.host
port = self.port port = self.port
return f"model {model_name}@{model_type}({host}:{port})" return f"model {model_name}@{model_type}({host}:{port})"

2
setup.py
View File

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