From 613bcf90e58f3ab81d3827e7fc572319908db9fb Mon Sep 17 00:00:00 2001 From: Matthew Hambrecht <14303543+matthambrecht@users.noreply.github.com> Date: Wed, 22 Oct 2025 09:55:26 -0400 Subject: [PATCH] fix: enable_sleep_mode -> vllm_enable_sleep_mode (#3225) Co-authored-by: Matthew Hambrecht --- src/axolotl/core/trainers/grpo/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/axolotl/core/trainers/grpo/__init__.py b/src/axolotl/core/trainers/grpo/__init__.py index d1a6b7fd9..bd77489eb 100644 --- a/src/axolotl/core/trainers/grpo/__init__.py +++ b/src/axolotl/core/trainers/grpo/__init__.py @@ -52,7 +52,7 @@ class GRPOStrategy: if trl.vllm_mode: grpo_args_kwargs["vllm_mode"] = trl.vllm_mode if trl.vllm_mode == "colocate": - grpo_args_kwargs["enable_sleep_mode"] = trl.vllm_enable_sleep_mode # type: ignore[attr-defined] + grpo_args_kwargs["vllm_enable_sleep_mode"] = trl.vllm_enable_sleep_mode # type: ignore[attr-defined] grpo_args_kwargs["vllm_gpu_memory_utilization"] = ( vllm_cfg.gpu_memory_utilization )