From 2123ff33a235af585f0fd725d914213ac3a37b72 Mon Sep 17 00:00:00 2001
From: FangYin Cheng <staneyffer@gmail.com>
Date: Thu, 12 Oct 2023 10:28:27 +0800
Subject: [PATCH] feat(core): Support read model path from model name prefix

---
 pilot/model/cluster/worker/manager.py | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/pilot/model/cluster/worker/manager.py b/pilot/model/cluster/worker/manager.py
index bd941781d..5648c8e01 100644
--- a/pilot/model/cluster/worker/manager.py
+++ b/pilot/model/cluster/worker/manager.py
@@ -729,8 +729,15 @@ def _parse_worker_params(
     model_name: str = None, model_path: str = None, **kwargs
 ) -> ModelWorkerParameters:
     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(
-        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)
     # Read parameters agein with prefix of model name.