From c8b303ea5ba59ff66261345204bd6f0df21295c5 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Thu, 13 Feb 2025 04:34:13 -0800 Subject: [PATCH] Update rl_replacements.py --- unsloth/models/rl_replacements.py | 1 + 1 file changed, 1 insertion(+) diff --git a/unsloth/models/rl_replacements.py b/unsloth/models/rl_replacements.py index 968f2b19fd..5da57c44bf 100644 --- a/unsloth/models/rl_replacements.py +++ b/unsloth/models/rl_replacements.py @@ -115,6 +115,7 @@ def sft_trainer_compute_loss(function_name, function): if function_name != "compute_loss": return function function = inspect.getsource(_sft_trainer_compute_loss) + function = function.replace("def _sft_trainer_compute_loss", "def compute_loss") return function pass RL_FUNCTIONS["sft_trainer"].append(sft_trainer_compute_loss)