Update models.py

sileod · Nov 2, 2023 · 3bd3910 · 3bd3910
1 parent a3bb814
commit 3bd3910
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/tasknet/models.py b/src/tasknet/models.py
@@ -342,6 +342,7 @@ class default:
             save_steps = 1000000
             label_names = ["labels"]
             include_inputs_for_metrics = True
+            model_name = "sileod/deberta-v3-base-tasksource-nli"
 
         default, hparams = to_dict(default), to_dict(hparams)
         self.p = hparams.get('p', 1)