from transformers import PretrainedConfig, BertModel class MyBertClassifierConfig(PretrainedConfig): model_type = "my_bert_classifier" def __init__( self, **kwargs, ): super().__init__(**kwargs)