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) | |
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) | |