Skip to content

Commit

Permalink
Fix #2745 (#2748)
Browse files Browse the repository at this point in the history
  • Loading branch information
erogol authored Jul 7, 2023
1 parent 7b5c842 commit a2984fb
Showing 1 changed file with 5 additions and 5 deletions.
10 changes: 5 additions & 5 deletions TTS/tts/models/tortoise.py
Original file line number Diff line number Diff line change
Expand Up @@ -497,14 +497,14 @@ def get_random_conditioning_latents(self):
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:
Expand All @@ -513,9 +513,9 @@ def synthesize(self, text, config, speaker_id="random", extra_voice_dirs=None, *
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)

Expand Down

0 comments on commit a2984fb

Please sign in to comment.