From a97ba4c6a49bf38a8076a9fbbd914b35cb006ee7 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Wed, 13 Nov 2024 17:38:26 -0800 Subject: [PATCH] Update __init__.py --- unsloth/__init__.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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()