File size: 367 Bytes
fdfbe63 |
1 2 3 4 5 6 7 8 9 10 11 |
from transformers import BertConfig
class NomBertConfig(BertConfig):
def __init__(self, unk_id=0, id_start=1, output_vocab_size=7430, lm_head_dict={}, **kwargs):
super().__init__(**kwargs)
self.unk_id = unk_id
self.id_start = id_start
self.output_vocab_size = output_vocab_size
self.lm_head_dict = lm_head_dict
|