from transformers import PretrainedConfig

class MosNetConfig(PretrainedConfig):
    """
    Это конфигурация для модели MosNet.
    Она хранит параметры, определяющие архитектуру модели.
    """
    model_type = "mosnet"

    def __init__(
        self,
        sample_rate: int = 16000,
        fft_size: int = 512,
        hop_length: int = 256,
        win_length: int = 512,
        dropout: float = 0.3,
        **kwargs,
    ):
        super().__init__(**kwargs)
        self.sample_rate = sample_rate
        self.fft_size = fft_size
        self.hop_length = hop_length
        self.win_length = win_length
        self.dropout = dropout