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