from transformers import PretrainedConfig


class OnsidesConfig(PretrainedConfig):
    model_type = "onsides"

    def __init__(self, classifier_dropout=0.5, num_labels=2, **kwargs):
        self.classifier_dropout = classifier_dropout
        self.num_labels = num_labels
        super().__init__(**kwargs)