from transformers import PretrainedConfig


class avGFPConfig(PretrainedConfig):
    model_type = "onehot_cnn"
    
    def __init__(
            self,
            vocab_size: int = 20,
            kernel_size: int = 5,
            input_size: int = 237,
            **kwargs
    ):
        self.vocab_size = vocab_size
        self.kernel_size = kernel_size
        self.input_size = input_size
        super().__init__(**kwargs)