File size: 238 Bytes
9f6d03f | 1 2 3 4 5 6 7 8 9 10 |
from transformers import PretrainedConfig
class CustomConfig(PretrainedConfig):
model_type = "custom"
def __init__(self, hidden_size=768, **kwargs):
self.hidden_size = hidden_size
super().__init__(**kwargs)
|