multilabel-classification-bert-ace2004 / configuration_multilabelbert.py
jvaquet's picture
Upload MultilabelNerPipeline
8804863 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()