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)
|