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

feat(core): Support read model path from model name prefix

Browse Source
This commit is contained in:
FangYin Cheng 2023-10-12 10:28:27 +08:00
parent 2a46909eac
commit 2123ff33a2
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
pilot/model/cluster/worker/manager.py
View File

@ -729,8 +729,15 @@ def _parse_worker_params(
model_name: str = None, model_path: str = None, **kwargs model_name: str = None, model_path: str = None, **kwargs
) -> ModelWorkerParameters: ) -> ModelWorkerParameters:
worker_args = EnvArgumentParser() worker_args = EnvArgumentParser()
env_prefix = None
if model_name:
env_prefix = EnvArgumentParser.get_env_prefix(model_name)
worker_params: ModelWorkerParameters = worker_args.parse_args_into_dataclass( worker_params: ModelWorkerParameters = worker_args.parse_args_into_dataclass(
ModelWorkerParameters, model_name=model_name, model_path=model_path, **kwargs ModelWorkerParameters,
env_prefix=env_prefix,
model_name=model_name,
model_path=model_path,
**kwargs,
) )
env_prefix = EnvArgumentParser.get_env_prefix(worker_params.model_name) env_prefix = EnvArgumentParser.get_env_prefix(worker_params.model_name)
# Read parameters agein with prefix of model name. # Read parameters agein with prefix of model name.