File size: 230 Bytes
5bb7ef5 | 1 2 3 4 5 6 7 | from transformers import BertConfig, AutoConfig
class MultiLabelBertConfig(BertConfig):
model_type = 'MultiLabelBert'
AutoConfig.register('MultiLabelBert', MultiLabelBertConfig)
MultiLabelBertConfig.register_for_auto_class() |