From 09ddc2ed5ca746752c053c78eb0560107915fc4e Mon Sep 17 00:00:00 2001 From: Erland366 Date: Thu, 3 Jul 2025 15:47:58 +0000 Subject: [PATCH] Initialize parent class in UnslothTrainingArguments constructor --- unsloth/trainer.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/unsloth/trainer.py b/unsloth/trainer.py index 3231353296..6a42fe4a2f 100644 --- a/unsloth/trainer.py +++ b/unsloth/trainer.py @@ -69,10 +69,8 @@ pass class UnslothTrainingArguments(TrainingArguments): def __init__(self, embedding_learning_rate: float = None, *args, **kwargs): - embedding_learning_rate : Optional[float] = field( - default = embedding_learning_rate, - metadata = {"help" : "Different learning rates for embeddings and lm_head."} - ) + embedding_learning_rate = embedding_learning_rate + super().__init__(*args, **kwargs) pass