From 285efeb30f582798ef515e62ff97463ec8e73558 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Sat, 15 Feb 2025 18:34:25 -0800 Subject: [PATCH] Update rl.py --- unsloth/models/rl.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/unsloth/models/rl.py b/unsloth/models/rl.py index 2facd3ccb0..df1f2f110d 100644 --- a/unsloth/models/rl.py +++ b/unsloth/models/rl.py @@ -622,9 +622,9 @@ def patch_functions(RLTrainer, trainer_file, RLTrainer_name, all_imports, import pass # Remove non editted functions - for remove in remover: - RLTrainer_source = RLTrainer_source.replace(remove, "") - pass + # for remove in remover: + # RLTrainer_source = RLTrainer_source.replace(remove, "") + # pass RLTrainer_source = RLTrainer_source.replace( f"class {RLTrainer_name}", f"class _Unsloth{RLTrainer_name}", 1