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