File size: 256 Bytes
2efddce | 1 2 3 4 5 6 7 8 9 10 |
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
|