File size: 250 Bytes
a3b24ec |
1 2 3 4 5 6 7 8 9 |
from transformers import PretrainedConfig
class DebertaConfig(PretrainedConfig):
model_type = "deberta_multilabel"
def __init__(self, num_labels: int=20, **kwargs):
self.number_labels = num_labels
super().__init__(**kwargs)
|