From 9921fb0ffdf0a45d8199e5e5c3fbef64b2b7389c Mon Sep 17 00:00:00 2001
From: Erland366 <erland.pg366@gmail.com>
Date: Thu, 3 Jul 2025 14:08:04 +0000
Subject: [PATCH] Always use SFTConfig

---
 unsloth/trainer.py | 8 ++------
 1 file changed, 2 insertions(+), 6 deletions(-)

diff --git a/unsloth/trainer.py b/unsloth/trainer.py
index 75fdd410e1..a360205432 100644
--- a/unsloth/trainer.py
+++ b/unsloth/trainer.py
@@ -61,13 +61,9 @@ else:
     pass
 pass
 
-try:
-    from trl import SFTConfig as TrainingArguments
-except:
-    from transformers import TrainingArguments
-pass
+from trl import SFTConfig
 @dataclass
-class UnslothTrainingArguments(TrainingArguments):
+class UnslothTrainingArguments(SFTConfig):
     embedding_learning_rate : Optional[float] = field(
         default = None,
         metadata = {"help" : "Different learning rates for embeddings and lm_head."}