from transformers.models.seamless_m4t_v2.configuration_seamless_m4t_v2 import (
    SeamlessM4Tv2Config,
)
from transformers.models.auto.configuration_auto import AutoConfig

MODEL_TYPE = "seamlessm4t-v2-large-speech_encoder"


class SeamlessM4Tv2EncoderConfig(SeamlessM4Tv2Config):
    model_type = MODEL_TYPE

    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)


AutoConfig.register(MODEL_TYPE, SeamlessM4Tv2EncoderConfig)

SeamlessM4Tv2EncoderConfig.register_for_auto_class()