From dbe5eb992e2a60340591d58e132aeb669642627b Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Eren=20G=C3=B6lge?= <egolge@coqui.ai>
Date: Tue, 19 Apr 2022 09:22:08 +0200
Subject: [PATCH] Make AP optional in BaseTTS

---
 TTS/tts/models/base_tts.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/TTS/tts/models/base_tts.py b/TTS/tts/models/base_tts.py
index 6069eb26..cad67654 100644
--- a/TTS/tts/models/base_tts.py
+++ b/TTS/tts/models/base_tts.py
@@ -31,8 +31,8 @@ class BaseTTS(BaseTrainerModel):
     def __init__(
         self,
         config: Coqpit,
-        ap: "AudioProcessor",
         tokenizer: "TTSTokenizer",
+        ap: "AudioProcessor" = None,
         speaker_manager: SpeakerManager = None,
         language_manager: LanguageManager = None,
     ):
@@ -314,7 +314,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,