mirror of https://github.com/coqui-ai/TTS.git
fix longer phoneme seqs
This commit is contained in:
parent
7b0a93d2f8
commit
a757b203bc
|
@ -138,6 +138,12 @@ class MyDataset(Dataset):
|
||||||
if "attn_file" in locals():
|
if "attn_file" in locals():
|
||||||
attn = np.load(attn_file)
|
attn = np.load(attn_file)
|
||||||
|
|
||||||
|
if len(text) > self.max_seq_len:
|
||||||
|
# return a different sample if the phonemized
|
||||||
|
# text is longer than the threshold
|
||||||
|
# TODO: find a better fix
|
||||||
|
return self.load_data(100)
|
||||||
|
|
||||||
sample = {
|
sample = {
|
||||||
'text': text,
|
'text': text,
|
||||||
'wav': wav,
|
'wav': wav,
|
||||||
|
|
Loading…
Reference in New Issue