File size: 308 Bytes
f25282b
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
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)