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

small fixes

pull/422/head
Eren Gölge 2021-04-05 15:51:36 +02:00
parent 8c9e1c9e58
commit d57f416957
2 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
TTS/utils/audio.py
View File

@ -387,10 +387,9 @@ class AudioProcessor(object):
x = self.sound_norm(x) x = self.sound_norm(x)
return x return x
def save_wav(self, wav, path, sample_rate=None): def save_wav(self, wav, path, sr=None):
sample_rate = self.sample_rate if sample_rate is None else sample_rate
wav_norm = wav * (32767 / max(0.01, np.max(np.abs(wav)))) wav_norm = wav * (32767 / max(0.01, np.max(np.abs(wav))))
scipy.io.wavfile.write(path, sample_rate, wav_norm.astype(np.int16)) scipy.io.wavfile.write(path, sr if sr else self.sample_rate, wav_norm.astype(np.int16))
@staticmethod @staticmethod
def mulaw_encode(wav, qc): def mulaw_encode(wav, qc):

5
TTS/vocoder/models/hifigan_generator.py
View File

@ -1,8 +1,8 @@
import torch import torch
import torch.nn.functional as F import torch.nn.functional as F
import torch.nn as nn import torch.nn as nn
from torch.nn import Conv1d, ConvTranspose1d, AvgPool1d, Conv2d from torch.nn import Conv1d, ConvTranspose1d
from torch.nn.utils import weight_norm, remove_weight_norm, spectral_norm from torch.nn.utils import weight_norm, remove_weight_norm
LRELU_SLOPE = 0.1 LRELU_SLOPE = 0.1
@ -115,6 +115,7 @@ class HifiganGenerator(torch.nn.Module):
resblock_kernel_sizes, upsample_kernel_sizes, resblock_kernel_sizes, upsample_kernel_sizes,
upsample_initial_channel, upsample_factors): upsample_initial_channel, upsample_factors):
super().__init__() super().__init__()
self.inference_padding = 5
self.num_kernels = len(resblock_kernel_sizes) self.num_kernels = len(resblock_kernel_sizes)
self.num_upsamples = len(upsample_factors) self.num_upsamples = len(upsample_factors)
self.conv_pre = weight_norm( self.conv_pre = weight_norm(