Skip to content

Commit b29a26a

Browse files
authored
[grpo] fix gym check (#4998)
1 parent f364cf4 commit b29a26a

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

swift/trainers/rlhf_trainer/grpo_trainer.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -233,9 +233,10 @@ def __init__(self,
233233
if is_peft_model(self.model):
234234
self.parameter_groups, self.parameter_groups_no_lora = self.split_batches()
235235
self.use_fast_infer = self.use_vllm # whether to use the PT backend
236-
# gym engine
237236
self.vllm_use_async_engine = False
238237
self.enable_offload = False
238+
# gym engine
239+
self.use_gym_env = False
239240
if self.use_vllm:
240241
if not is_vllm_available():
241242
raise ImportError('vLLM is not available and `use_vllm` is set to True. '

0 commit comments

Comments
 (0)