| 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 | |
| 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 | |