danschr commited on
Commit
464e8a1
1 Parent(s): 34dd635

Upload DebertaArgClassifier

Browse files
config.json CHANGED
@@ -29,7 +29,7 @@
29
  "19": "Universalism: objectivity"
30
  },
31
  "label2id": null,
32
- "model_type": "deberta_multilabel",
33
  "number_labels": 20,
34
  "torch_dtype": "float32",
35
  "transformers_version": "4.26.1"
 
29
  "19": "Universalism: objectivity"
30
  },
31
  "label2id": null,
32
+ "model_type": "deberta_arg_classifier",
33
  "number_labels": 20,
34
  "torch_dtype": "float32",
35
  "transformers_version": "4.26.1"
configuration_deberta_arg_classifier.py CHANGED
@@ -1,7 +1,7 @@
1
  from transformers import PretrainedConfig
2
 
3
  class DebertaConfig(PretrainedConfig):
4
- model_type = "deberta_multilabel"
5
 
6
  def __init__(self, num_labels: int=20, **kwargs):
7
  self.number_labels = num_labels
 
1
  from transformers import PretrainedConfig
2
 
3
  class DebertaConfig(PretrainedConfig):
4
+ model_type = "deberta_arg_classifier"
5
 
6
  def __init__(self, num_labels: int=20, **kwargs):
7
  self.number_labels = num_labels