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

fix:worker_register_host parameter does not work (#527)

Browse Source 
fix:worker_register_host parameter does not work
Close #526
This commit is contained in:
FangYin Cheng
2023-09-03 18:32:10 +08:00
committed by GitHub
parent e37ce04232 858a1caefd
commit b8f09df45e
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
pilot/model/worker/manager.py
View File

@@ -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):