diff --git a/unsloth/__init__.py b/unsloth/__init__.py index d94eeb8973..745b210208 100644 --- a/unsloth/__init__.py +++ b/unsloth/__init__.py @@ -177,5 +177,5 @@ from .chat_templates import * from .tokenizer_utils import * from .trainer import * -# patch sft trainer -_patch_trl_trainer() \ No newline at end of file +# Patch TRL trainers for backwards compatibility +_patch_trl_trainer()