File size: 511 Bytes
65a0eff |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
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()
|