From 9bbba3a511f23386581e7c5f36ed478bf1a29ef9 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Wed, 26 Mar 2025 05:18:18 -0700 Subject: [PATCH] Update rl_replacements.py --- unsloth/models/rl_replacements.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/unsloth/models/rl_replacements.py b/unsloth/models/rl_replacements.py index a3b2d1de8a..376d1e9a28 100644 --- a/unsloth/models/rl_replacements.py +++ b/unsloth/models/rl_replacements.py @@ -79,7 +79,7 @@ def sft_trainer_prepare_dataset(function_name, function): function_name != "_prepare_dataset": return function fast_sft_prepare_dataset = RL_REPLACEMENTS.get("sft_prepare_dataset", None) - if fast_sft_prepare_dataset is not None and "pack_examples" in function: + if fast_sft_prepare_dataset is not None: params = inspect.signature(fast_sft_prepare_dataset).parameters.keys() params = ".*?".join(params) matched = re.match(