a3b24ec 464e8a1 a3b24ec
1
2
3
4
5
6
7
8
9
from transformers import PretrainedConfig class DebertaConfig(PretrainedConfig): model_type = "deberta_arg_classifier" def __init__(self, num_labels: int=20, **kwargs): self.number_labels = num_labels super().__init__(**kwargs)