coqui-ai
/
TTS
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Remove unusable speaker manager function

pull/1032/head
Edresson 2021-11-22 08:57:13 -03:00 committed by Eren Gölge
parent 8c22d5ac49
commit 9daa33d1fd
2 changed files with 1 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
TTS/tts/models/vits.py
View File

@ -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()

14
TTS/tts/utils/speakers.py
View File

@ -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.