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(