File size: 254 Bytes
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)