fix d-vector

This commit is contained in:
WeberJulian 2021-11-18 00:17:42 +01:00 committed by Eren Gölge
parent 868cf6424f
commit 390096fe0f
1 changed files with 1 additions and 1 deletions

View File

@ -415,7 +415,7 @@ class TTSDataset(Dataset):
language_ids = None
# get pre-computed d-vectors
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]
else:
d_vectors = None