fix d-vector

2026-09-09 11:16:00 +08:00 · 2021-12-09 13:36:48 +00:00
parent 868cf6424f
commit 390096fe0f
1 changed files with 1 additions and 1 deletions
@@ -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