seamless-m4t-v2-large-speech-encoder / configuration_seamless_m4t_v2_speech_encoder.py
fdschmidt93's picture
initial commit
65a0eff
raw
history blame contribute delete
511 Bytes
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()