mirror of https://github.com/coqui-ai/TTS.git
fix d-vector
parent
da6c1e858c
commit
631addf33b
|
@ -415,7 +415,7 @@ class TTSDataset(Dataset):
|
||||||
language_ids = None
|
language_ids = None
|
||||||
# get pre-computed d-vectors
|
# get pre-computed d-vectors
|
||||||
if self.d_vector_mapping is not None:
|
if self.d_vector_mapping is not None:
|
||||||
wav_files_names = [batch["wav_file_name"][idx] for idx in ids_sorted_decreasing]
|
wav_files_names = list(batch["wav_file_name"])
|
||||||
d_vectors = [self.d_vector_mapping[w]["embedding"] for w in wav_files_names]
|
d_vectors = [self.d_vector_mapping[w]["embedding"] for w in wav_files_names]
|
||||||
else:
|
else:
|
||||||
d_vectors = None
|
d_vectors = None
|
||||||
|
|
Loading…
Reference in New Issue