multilabel-classification-bert-conll03 / configuration_multilabelbert.py
jvaquet's picture
Upload MultilabelNerPipeline
f0bb526 verified
raw
history blame contribute delete
230 Bytes
from transformers import BertConfig, AutoConfig
class MultiLabelBertConfig(BertConfig):
model_type = 'MultiLabelBert'
AutoConfig.register('MultiLabelBert', MultiLabelBertConfig)
MultiLabelBertConfig.register_for_auto_class()