Medix-AI/coqui-tts
Medix-AI
/
coqui-tts
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Make AP optional in BaseTTS

This commit is contained in:
Eren Gölge 2022-04-19 09:22:08 +02:00 committed by Eren G??lge
parent 6a53b77a95
commit dbe5eb992e
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
TTS/tts/models/base_tts.py
View File

@ -31,8 +31,8 @@ class BaseTTS(BaseTrainerModel):
def __init__( def __init__(
self, self,
config: Coqpit, config: Coqpit,
ap: "AudioProcessor",
tokenizer: "TTSTokenizer", tokenizer: "TTSTokenizer",
ap: "AudioProcessor" = None,
speaker_manager: SpeakerManager = None, speaker_manager: SpeakerManager = None,
language_manager: LanguageManager = None, language_manager: LanguageManager = None,
): ):
@ -314,7 +314,7 @@ class BaseTTS(BaseTrainerModel):
compute_f0=config.get("compute_f0", False), compute_f0=config.get("compute_f0", False),
f0_cache_path=config.get("f0_cache_path", None), f0_cache_path=config.get("f0_cache_path", None),
samples=samples, samples=samples,
ap=self.ap, audio_config=self.config.audio,
return_wav=config.return_wav if "return_wav" in config else False, return_wav=config.return_wav if "return_wav" in config else False,
batch_group_size=0 if is_eval else config.batch_group_size * config.batch_size, batch_group_size=0 if is_eval else config.batch_group_size * config.batch_size,
min_text_len=config.min_text_len, min_text_len=config.min_text_len,