diff --git a/src/tasknet/models.py b/src/tasknet/models.py index f6f39d3..1122d7a 100755 --- 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)