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