From a2984fb43510060245af181f73b83de55bd4e73b Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Eren=20G=C3=B6lge?= Date: Fri, 7 Jul 2023 20:23:27 +0200 Subject: [PATCH] Fix #2745 (#2748) --- TTS/tts/models/tortoise.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/TTS/tts/models/tortoise.py b/TTS/tts/models/tortoise.py index 88817725..af7ea583 100644 --- a/TTS/tts/models/tortoise.py +++ b/TTS/tts/models/tortoise.py @@ -497,14 +497,14 @@ class Tortoise(BaseTTS): with torch.no_grad(): return self.rlg_auto(torch.tensor([0.0])), self.rlg_diffusion(torch.tensor([0.0])) - def synthesize(self, text, config, speaker_id="random", extra_voice_dirs=None, **kwargs): + def synthesize(self, text, config, speaker_id="random", voice_dirs=None, **kwargs): """Synthesize speech with the given input text. Args: text (str): Input text. config (TortoiseConfig): Config with inference parameters. speaker_id (str): One of the available speaker names. If `random`, it generates a random speaker. - extra_voice_dirs (List[str]): List of paths that host reference audio files for speakers. Defaults to None. + voice_dirs (List[str]): List of paths that host reference audio files for speakers. Defaults to None. **kwargs: Inference settings. See `inference()`. Returns: @@ -513,9 +513,9 @@ class Tortoise(BaseTTS): as latents used at inference. """ - if extra_voice_dirs is not None: - extra_voice_dirs = [extra_voice_dirs] - voice_samples, conditioning_latents = load_voice(speaker_id, extra_voice_dirs) + if voice_dirs is not None: + voice_dirs = [voice_dirs] + voice_samples, conditioning_latents = load_voice(speaker_id, voice_dirs) else: voice_samples, conditioning_latents = load_voice(speaker_id)