Skip to content

Commit 1050701

Browse files
authored
[grpo] fix base url (#4463)
1 parent 730ecc9 commit 1050701

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

swift/llm/argument/rlhf_args.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -200,7 +200,10 @@ def _init_external_vllm(self):
200200
from swift.trainers.rlhf_trainer.vllm_client import VLLMClient
201201
if is_master():
202202
self.vllm_client = VLLMClient(
203-
self.vllm_server_host, self.vllm_server_port, connection_timeout=self.vllm_server_timeout)
203+
base_url=self.vllm_server_base_url,
204+
host=self.vllm_server_host,
205+
server_port=self.vllm_server_port,
206+
connection_timeout=self.vllm_server_timeout)
204207
self.vllm_client.init_communicator()
205208

206209
def _set_default(self):

0 commit comments

Comments
 (0)