update

Jintao-Huang · Feb 18, 2025 · 2db85ef · 2db85ef
1 parent 1188fe3
commit 2db85ef
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
@@ -219,7 +219,7 @@ def _template_context(template):
     def _move_model_to_vllm_lmdeploy(self):
         from accelerate.utils.other import is_compiled_module
         with unwrap_model_for_generation(
-                self.model_wrapped, self.accelerator,
+                self.model, self.accelerator,
                 gather_deepspeed3_params=self.args.ds3_gather_for_generation) as unwrapped_model:
             if is_compiled_module(unwrapped_model):
                 unwrapped_model = unwrapped_model._orig_mod