from transformers import PretrainedConfig

class PhyloGPNConfig(PretrainedConfig):
    model_type = "phylogpn"

    def __init__(self, outer_dim: int = 960, inner_dim: int = 480, kernel_size: int = 5, stack_size: int = 2, num_stacks: int = 20, **kwargs):
        self.outer_dim = outer_dim
        self.inner_dim = inner_dim
        self.kernel_size = kernel_size
        self.stack_size = stack_size
        self.num_stacks = num_stacks
        super().__init__(**kwargs)