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