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