mirror of https://github.com/coqui-ai/TTS.git
fix collate_fn
This commit is contained in:
parent
78c2d12a91
commit
2a2b5767c2
|
@ -402,8 +402,6 @@ class TTSDataset(Dataset):
|
||||||
# convert list of dicts to dict of lists
|
# convert list of dicts to dict of lists
|
||||||
batch = {k: [dic[k] for dic in batch] for k in batch[0]}
|
batch = {k: [dic[k] for dic in batch] for k in batch[0]}
|
||||||
|
|
||||||
speaker_names = [batch[idx]["speaker_name"] for idx in ids_sorted_decreasing]
|
|
||||||
|
|
||||||
# get language ids from language names
|
# get language ids from language names
|
||||||
if self.language_id_mapping is not None:
|
if self.language_id_mapping is not None:
|
||||||
language_names = [batch[idx]["language_name"] for idx in ids_sorted_decreasing]
|
language_names = [batch[idx]["language_name"] for idx in ids_sorted_decreasing]
|
||||||
|
|
Loading…
Reference in New Issue