| | from ..llama_configs import LLAMA_CONFIGS |
| |
|
| |
|
| | class T3Config: |
| | def __init__(self, text_tokens_dict_size=704): |
| | self.start_text_token = 255 |
| | self.stop_text_token = 0 |
| | self.text_tokens_dict_size = text_tokens_dict_size |
| | self.max_text_tokens = 2048 |
| |
|
| | self.start_speech_token = 6561 |
| | self.stop_speech_token = 6562 |
| | self.speech_tokens_dict_size = 8194 |
| | self.max_speech_tokens = 4096 |
| |
|
| | self.llama_config_name = "Llama_520M" |
| | self.input_pos_emb = "learned" |
| | self.speech_cond_prompt_len = 150 |
| |
|
| | self.encoder_type = "voice_encoder" |
| | self.speaker_embed_size = 256 |
| | self.use_perceiver_resampler = True |
| | self.emotion_adv = True |
| |
|
| | @property |
| | def n_channels(self): |
| | return LLAMA_CONFIGS[self.llama_config_name]["hidden_size"] |
| | |
| | @property |
| | def is_multilingual(self): |
| | return self.text_tokens_dict_size == 2454 |
| |
|
| | @classmethod |
| | def english_only(cls): |
| | """Create configuration for English-only TTS model.""" |
| | return cls(text_tokens_dict_size=704) |
| | |
| | @classmethod |
| | def multilingual(cls): |
| | """Create configuration for multilingual TTS model.""" |
| | return cls(text_tokens_dict_size=2454) |
