from transformers import PretrainedConfig

class CustomConfig(PretrainedConfig):
    model_type = "custom_model"

    def __init__(self,
                transformer_type = "microsoft/graphcodebert-base",
                transformer_output_dim = 768,
                **kwargs):
        super().__init__(**kwargs)
        self.transformer_type = transformer_type
        self.transformer_output_dim = transformer_output_dim