From c2ddabcfd654f4f50dd6da5298f156337eed54af Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Wed, 5 Mar 2025 01:11:03 -0800 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 55c2daa323..7462d55944 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: + if fast_sft_prepare_dataset is not None and "pack_examples" in function: params = inspect.signature(fast_sft_prepare_dataset).parameters.keys() params = ".*?".join(params) matched = re.match(