From 858a1caefd693c5162144193fcfef3f7ac60fab6 Mon Sep 17 00:00:00 2001
From: aries_ckt <916701291@qq.com>
Date: Sun, 3 Sep 2023 18:14:51 +0800
Subject: [PATCH] fix:worker_register_host parameter does not work Close #526

---
 pilot/model/worker/manager.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/pilot/model/worker/manager.py b/pilot/model/worker/manager.py
index f39aba585..79c455f54 100644
--- a/pilot/model/worker/manager.py
+++ b/pilot/model/worker/manager.py
@@ -587,7 +587,11 @@ def _create_local_model_manager(
         from pilot.utils.net_utils import _get_ip_address
 
         client = ModelRegistryClient(worker_params.controller_addr)
-        host = _get_ip_address()
+        host = (
+            worker_params.worker_register_host
+            if worker_params.worker_register_host
+            else _get_ip_address()
+        )
         port = worker_params.port
 
         async def register_func(worker_run_data: WorkerRunData):