Fix unit tests

This commit is contained in:
Edresson Casanova 2022-03-11 19:55:29 -03:00
parent e33819b7de
commit 25da4d9b74
2 changed files with 2 additions and 2 deletions

View File

@ -257,7 +257,7 @@ class EmbeddingManager(BaseIDManager):
embedding = _compute(wav_file) embedding = _compute(wav_file)
return embedding[0].tolist() 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. """Compute embedding from features.
Args: Args:

View File

@ -210,7 +210,7 @@ class Synthesizer(object):
# handle multi-speaker # handle multi-speaker
speaker_embedding = None speaker_embedding = None
speaker_id = 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 speaker_name and isinstance(speaker_name, str):
if self.tts_config.use_d_vector_file: if self.tts_config.use_d_vector_file:
# get the average speaker embedding from the saved embeddings. # get the average speaker embedding from the saved embeddings.