From 022d5718357f10a7a8befa3e7d8dfdc4df613cfb Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Wed, 13 Nov 2024 18:48:59 -0800 Subject: [PATCH] Update trainer.py --- unsloth/trainer.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/unsloth/trainer.py b/unsloth/trainer.py index 896db35ecd..25b8999959 100644 --- a/unsloth/trainer.py +++ b/unsloth/trainer.py @@ -201,7 +201,10 @@ pass def _patch_trl_trainer(): - if hasattr(trl, "__UNSLOTH_BACKWARDS_COMPATIBLE__"): return + try: + trl.__UNSLOTH_BACKWARDS_COMPATIBLE__ + return + except: pass if Version(trl.__version__) <= Version("0.11.0"): return import trl.trainer