Update models.py

sileod · Nov 2, 2023 · f89bbce · f89bbce
1 parent 3bd3910
commit f89bbce
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/tasknet/models.py b/src/tasknet/models.py
@@ -350,7 +350,7 @@ class default:
  self.batched = hparams.get('batched',False)
 
  trainer_args = transformers.TrainingArguments(
-  **{**default, **fc.project(hparams, dir(transformers.TrainingArguments))},
+ **fc.project({**default,**hparams}, dir(transformers.TrainingArguments))
  )
  if not tokenizer:
  tokenizer = AutoTokenizer.from_pretrained(hparams["model_name"])