mrq/DL-Art-School
1
3
Fork 5
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix bug in code alignment

Browse Source
This commit is contained in:
James Betker 2022-01-20 11:28:12 -07:00
parent 8e2439f50d
commit 20312211e0
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
codes/data/audio/fast_paired_dataset.py
View File

@ -155,7 +155,7 @@ class FastPairedVoiceDataset(torch.utils.data.Dataset):
'real_text': text,
'padded_text': tseq,
'aligned_codes': aligned_codes,
'aligned_code_lengths': orig_aligned_code_length,
'aligned_codes_lengths': orig_aligned_code_length,
'text_lengths': torch.tensor(orig_text_len, dtype=torch.long),
'wav': wav,
'wav_lengths': torch.tensor(orig_output, dtype=torch.long),