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