2efddce 4afb295 2efddce
1
2
3
4
5
6
7
8
9
10
11
from transformers import BertConfig class NewModelConfig(BertConfig): model_type = "new-model" def __init__(self, *args, new_hidden_size=12, **kwargs): super().__init__(*args, **kwargs) self.new_hidden_size = new_hidden_size
