from transformers import MistralConfig


class MistralDenseFormerConfig(MistralConfig):
    model_type = "mistral_denseformer"

    def __init__(self, *args, dilation=4, dwa_period=5, **kwargs):
        self.dilation = dilation
        self.dwa_period = dwa_period
        super().__init__(*args, **kwargs)