Deberta_Human_Value_Detector / configuration_deberta_arg_classifier.py
danschr's picture
Upload DebertaArgClassifier
464e8a1
raw
history blame
No virus
254 Bytes
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)