ceos/unsloth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Initialize parent class in UnslothTrainingArguments constructor

Browse source
This commit is contained in:
Erland366 2025-07-03 15:47:58 +00:00
commit 09ddc2ed5c
1 changed files with 2 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

6
unsloth/trainer.py
View file

@ -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