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:worker_register_host parameter does not work

Browse Source 
Close #526
This commit is contained in:
aries_ckt 2023-09-03 18:14:51 +08:00
parent e37ce04232
commit 858a1caefd
1 changed files with 5 additions and 1 deletions
Show Stats 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 from pilot.utils.net_utils import _get_ip_address
client = ModelRegistryClient(worker_params.controller_addr) 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 port = worker_params.port
async def register_func(worker_run_data: WorkerRunData): async def register_func(worker_run_data: WorkerRunData):