From c0933dfb777e898371f5e1e9b47e51b750bb2717 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Sat, 15 Feb 2025 16:45:57 -0800 Subject: [PATCH] Update rl.py --- unsloth/models/rl.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/unsloth/models/rl.py b/unsloth/models/rl.py index d53c9606d2..ac1b836673 100644 --- a/unsloth/models/rl.py +++ b/unsloth/models/rl.py @@ -535,8 +535,8 @@ def patch_functions(RLTrainer, trainer_file, RLTrainer_name, all_imports, import sampling_params # Add spaces new_vllm_part = \ f"\n{' '*8}if {args}.use_vllm:\n{sampling_params} "\ - f"if getattr(args, 'sampling_params', None) is None else "\ - f"getattr(args, 'sampling_params', None)\n{' '*8}else:\n" + f"if getattr(args, 'vllm_sampling_params', None) is None else "\ + f"getattr(args, 'vllm_sampling_params', None)\n{' '*8}else:\n" init = init.replace(vllm_part, new_vllm_part) pass pass