from transformers import PretrainedConfig from typing import List class BertConfig(PretrainedConfig): model_type="bert" def __init__( self, **kwargs, ): super().__init__(**kwargs)