From ce95b13c0a2ab4ef9540772800f66b276559ed3d Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Thu, 13 Feb 2025 16:37:05 -0800 Subject: [PATCH] Update rl.py --- unsloth/models/rl.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/unsloth/models/rl.py b/unsloth/models/rl.py index fadae874db..048ec7bb02 100644 --- a/unsloth/models/rl.py +++ b/unsloth/models/rl.py @@ -566,8 +566,8 @@ pass def PatchFastRL(algorithm = None, FastLanguageModel = None): - return - # if FastLanguageModel is not None: PatchRL(FastLanguageModel) - # patch_trl_rl_trainers() - # if algorithm is not None: PatchRLStatistics(algorithm) + if FastLanguageModel is not None: PatchRL(FastLanguageModel) + patch_trl_rl_trainers() + if type(algorithm) is str and algorithm.islower(): + PatchRLStatistics(algorithm) pass