From 24274c58f8eb53d4753045cdeca7a9e8f075e79d Mon Sep 17 00:00:00 2001 From: Edresson Casanova Date: Fri, 11 Mar 2022 19:55:29 -0300 Subject: [PATCH] Fix unit tests --- TTS/tts/utils/managers.py | 2 +- TTS/utils/synthesizer.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/TTS/tts/utils/managers.py b/TTS/tts/utils/managers.py index 480c4c98..963226e0 100644 --- a/TTS/tts/utils/managers.py +++ b/TTS/tts/utils/managers.py @@ -257,7 +257,7 @@ class EmbeddingManager(BaseIDManager): embedding = _compute(wav_file) return embedding[0].tolist() - def compute_embedding(self, feats: Union[torch.Tensor, np.ndarray]) -> List: + def compute_embeddings(self, feats: Union[torch.Tensor, np.ndarray]) -> List: """Compute embedding from features. Args: diff --git a/TTS/utils/synthesizer.py b/TTS/utils/synthesizer.py index 54676635..c695446b 100644 --- a/TTS/utils/synthesizer.py +++ b/TTS/utils/synthesizer.py @@ -210,7 +210,7 @@ class Synthesizer(object): # handle multi-speaker speaker_embedding = None speaker_id = None - if self.tts_speakers_file or hasattr(self.tts_model.speaker_manager, "speaker_ids"): + if self.tts_speakers_file or hasattr(self.tts_model.speaker_manager, "ids"): if speaker_name and isinstance(speaker_name, str): if self.tts_config.use_d_vector_file: # get the average speaker embedding from the saved d_vectors.