From ec31dacbb70d2356a34fe3ffa955028a8f9ca57d Mon Sep 17 00:00:00 2001 From: Edresson Date: Mon, 22 Nov 2021 08:57:13 -0300 Subject: [PATCH] Remove unusable speaker manager function --- TTS/tts/models/vits.py | 2 +- TTS/tts/utils/speakers.py | 14 -------------- 2 files changed, 1 insertion(+), 15 deletions(-) diff --git a/TTS/tts/models/vits.py b/TTS/tts/models/vits.py index 0abf0ca3..7b27bc73 100644 --- a/TTS/tts/models/vits.py +++ b/TTS/tts/models/vits.py @@ -519,7 +519,7 @@ class Vits(BaseTTS): if speaker_name is None: d_vector = self.speaker_manager.get_random_d_vector() else: - d_vector = self.speaker_manager.get_d_vector_by_speaker(speaker_name) + d_vector = self.speaker_manager.get_mean_d_vector(speaker_name, num_samples=1, randomize=False) elif config.use_speaker_embedding: if speaker_name is None: speaker_id = self.speaker_manager.get_random_speaker_id() diff --git a/TTS/tts/utils/speakers.py b/TTS/tts/utils/speakers.py index 8c248658..828abede 100644 --- a/TTS/tts/utils/speakers.py +++ b/TTS/tts/utils/speakers.py @@ -194,20 +194,6 @@ class SpeakerManager: """ return [x["embedding"] for x in self.d_vectors.values() if x["name"] == speaker_idx] - def get_d_vector_by_speaker(self, speaker_idx: str) -> np.ndarray: - """Get a d_vector of a speaker. - - Args: - speaker_idx (str): Target speaker ID. - - Returns: - np.ndarray: d_vector. - """ - for x in self.d_vectors.values(): - if x["name"] == speaker_idx: - return x["embedding"] - return None - def get_mean_d_vector(self, speaker_idx: str, num_samples: int = None, randomize: bool = False) -> np.ndarray: """Get mean d_vector of a speaker ID.