from transformers import BertConfig, AutoConfig

class MultiLabelBertConfig(BertConfig):
    model_type = 'MultiLabelBert'

AutoConfig.register('MultiLabelBert', MultiLabelBertConfig)
MultiLabelBertConfig.register_for_auto_class()