Medix-AI/coqui-tts
Medix-AI
/
coqui-tts
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Allow custom padding value

This commit is contained in:
Eren Gölge 2021-09-30 14:38:34 +00:00
parent 355dfee98d
commit 4157e99d2d
1 changed files with 4 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
TTS/tts/utils/data.py
View File

@ -1,15 +1,14 @@
import numpy as np
def _pad_data(x, length):
_pad = 0
def _pad_data(x, length, pad_val=0):
assert x.ndim == 1
return np.pad(x, (0, length - x.shape[0]), mode="constant", constant_values=_pad)
return np.pad(x, (0, length - x.shape[0]), mode="constant", constant_values=pad_val)
def prepare_data(inputs):
def prepare_data(inputs, pad_val=0):
max_len = max((len(x) for x in inputs))
return np.stack([_pad_data(x, max_len) for x in inputs])
return np.stack([_pad_data(x, max_len, pad_val) for x in inputs])
def _pad_tensor(x, length):