From 5188866bd5ba2e0ef09503879772d4dedd0b2249 Mon Sep 17 00:00:00 2001 From: James Betker Date: Tue, 24 May 2022 14:55:51 -0600 Subject: [PATCH] facepalm --- codes/data/audio/fast_paired_dataset.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/codes/data/audio/fast_paired_dataset.py b/codes/data/audio/fast_paired_dataset.py index 9aaaa6fc..dec869a4 100644 --- a/codes/data/audio/fast_paired_dataset.py +++ b/codes/data/audio/fast_paired_dataset.py @@ -213,8 +213,8 @@ class FastPairedVoiceDataset(torch.utils.data.Dataset): res['conditioning'] = cond res['conditioning_contains_self'] = cond_is_self if self.load_aligned_codes: - res['aligned_codes']: aligned_codes - res['aligned_codes_lengths']: orig_aligned_code_length + res['aligned_codes'] = aligned_codes + res['aligned_codes_lengths'] = orig_aligned_code_length if self.produce_ctc_metadata: res.update(self.get_ctc_metadata(raw_codes))