From d9b31ee7642af78d1c22a818af2c0be8d2cd3c0e Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Wed, 19 Feb 2025 12:51:22 -0800 Subject: [PATCH] Update rl.py --- unsloth/models/rl.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/unsloth/models/rl.py b/unsloth/models/rl.py index 85b66e3f80..48f04412fe 100644 --- a/unsloth/models/rl.py +++ b/unsloth/models/rl.py @@ -519,7 +519,7 @@ def patch_functions(RLTrainer, trainer_file, RLTrainer_name, all_imports, import replacer = replacer[0] vllm_setter = "\n" + " "*8 + \ "if hasattr(model, 'vllm_engine') and "\ - "hasattr(self.args, 'use_vllm') and (getattr(self.args, 'use_vllm', False) == False): "\ + "hasattr(args, 'use_vllm') and (getattr(args, 'use_vllm', False) == False): "\ "args.use_vllm = True\n" init = init.replace(replacer, replacer + vllm_setter) pass