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()