File size: 519 Bytes
cb9ad2c | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 | from transformers import WavLMConfig
class DualCTCConfig(WavLMConfig):
model_type = "dual_ctc"
def __init__(
self,
kana_vocab_size=84,
phoneme_vocab_size=44,
kana_ctc_layer=11,
phoneme_ctc_layer=6,
**kwargs
):
super().__init__(**kwargs)
self.kana_vocab_size = kana_vocab_size
self.phoneme_vocab_size = phoneme_vocab_size
self.kana_ctc_layer = kana_ctc_layer
self.phoneme_ctc_layer = phoneme_ctc_layer |