kxngluca/DL-Art-School
1
0
Fork 0
forked from mrq/DL-Art-School
Code Pull Requests Activity

pairedvoice: error out if clip is too short

Browse Source
This commit is contained in:
James Betker 2022-02-21 19:11:10 -07:00
parent 38802a96c8
commit af50afe222
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
codes/data/audio/paired_voice_audio_dataset.py
View File

@ -140,6 +140,9 @@ class TextWavLoader(torch.utils.data.Dataset):
tseq, wav, text, path = self.get_wav_text_pair(self.audiopaths_and_text[index])
if text is None or len(text.strip()) == 0:
raise ValueError
if wav is None or wav.shape[-1] < (.1 * self.sample_rate):
# Ultra short clips are also useless (and can cause problems within some models).
raise ValueError
cond, cond_is_self = load_similar_clips(self.audiopaths_and_text[index][0], self.conditioning_length, self.sample_rate,
n=self.conditioning_candidates) if self.load_conditioning else (None, False)
except: