fix d-vector

pull/1032/head
WeberJulian 2021-11-18 00:17:42 +01:00 committed by Eren Gölge
parent da6c1e858c
commit 631addf33b
1 changed files with 1 additions and 1 deletions

View File

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