fix grpo trainer (#3120)

modelscope · Feb 14, 2025 · 235c117 · 235c117
1 parent e438a5e
commit 235c117
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/swift/trainers/rlhf_trainer/grpo_trainer.py b/swift/trainers/rlhf_trainer/grpo_trainer.py
@@ -131,7 +131,7 @@ def __init__(self,
                 profiling_patch = patch(
                     'vllm.worker.worker.Worker._assert_memory_footprint_increased_during_profiling', return_value=None)
                 from swift.tuners import Swift
-                with world_size_patch, profiling_patch, Swift.grpo_context(model, self.processor):
+                with world_size_patch, profiling_patch, Swift.grpo_context(model, self.template.processor):
                     self.engine = VllmEngine(
                         model.model_dir,
                         model.model_info.torch_dtype,