@@ -210,10 +210,10 @@ def _init_grpo(self):
210210 logger .info (f'Auto-configured soft_max_length = max_completion_length { self .max_completion_length } ' )
211211 if self .use_vllm :
212212 # set vllm mode
213- if self .vllm_server_host is not None :
213+ if self .vllm_server_host is not None or self . vllm_server_base_url is not None :
214214 if self .vllm_mode != 'server' :
215215 self .vllm_mode = 'server'
216- logger .warning ('set vllm_mode to `server` since vllm_server_host is provided' )
216+ logger .warning ('set vllm_mode to `server` since vllm server host/base_url is provided' )
217217 else :
218218 if self .vllm_mode != 'colocate' :
219219 self .vllm_mode = 'colocate'
@@ -250,7 +250,7 @@ def _init_rm(self):
250250 self .num_labels = 1
251251
252252 def _init_external_vllm (self ):
253- if self .rlhf_type != 'grpo' or self .vllm_server_host is None :
253+ if self .rlhf_type != 'grpo' or ( self .vllm_server_host is None and self . vllm_server_base_url is None ) :
254254 return
255255 from swift .trainers .rlhf_trainer .vllm_client import VLLMClient
256256 if is_master ():
@@ -310,7 +310,7 @@ def _check_grpo(self):
310310 assert is_liger_kernel_available (), (
311311 'Please install/update liger-kernel by running: pip install -U liger-kernel' )
312312 if self .vllm_mode == 'server' :
313- assert not self .use_vllm or self .vllm_server_host is not None
313+ assert not self .use_vllm or self .vllm_server_host is not None or self . vllm_server_base_url is not None
314314
315315 if self .async_generate :
316316 assert self .vllm_mode == 'server' , 'async generate require vllm_mode == server, '
0 commit comments