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)