From 34969ca1e130742c5797d4376f0fa864c5658a6c Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Thu, 13 Feb 2025 15:02:50 -0800 Subject: [PATCH] Update rl.py --- unsloth/models/rl.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/unsloth/models/rl.py b/unsloth/models/rl.py index 466101d16c..fc094b0839 100644 --- a/unsloth/models/rl.py +++ b/unsloth/models/rl.py @@ -565,7 +565,8 @@ pass def PatchFastRL(algorithm = None, FastLanguageModel = None): - if FastLanguageModel is not None: PatchRL(FastLanguageModel) - patch_trl_rl_trainers() - if algorithm is not None: PatchRLStatistics(algorithm) + return + # if FastLanguageModel is not None: PatchRL(FastLanguageModel) + # patch_trl_rl_trainers() + # if algorithm is not None: PatchRLStatistics(algorithm) pass