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
parent 2457739b5e
commit 85c03c75ca
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

@ -30,8 +30,8 @@ class BaseTTS(BaseTrainerModel):
def __init__(
self,
config: Coqpit,
ap: "AudioProcessor",
tokenizer: "TTSTokenizer",
ap: "AudioProcessor" = None,
speaker_manager: SpeakerManager = None,
language_manager: LanguageManager = None,
):
@ -309,7 +309,7 @@ class BaseTTS(BaseTrainerModel):
compute_f0=config.get("compute_f0", False),
f0_cache_path=config.get("f0_cache_path", None),
samples=samples,
ap=self.ap,
audio_config=self.config.audio,
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,
min_text_len=config.min_text_len,