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

pylint check

This commit is contained in:
erogol 2020-02-06 15:16:29 +01:00
parent c776526c45
commit 532cf8160c
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
server/synthesizer.py
View File

@ -68,12 +68,15 @@ class Synthesizer(object):
def load_wavernn(self, lib_path, model_file, model_config, use_cuda): def load_wavernn(self, lib_path, model_file, model_config, use_cuda):
# TODO: set a function in wavernn code base for model setup and call it here. # TODO: set a function in wavernn code base for model setup and call it here.
sys.path.append(lib_path) # set this if TTS is not installed globally sys.path.append(lib_path) # set this if WaveRNN is not installed globally
#pylint: disable=import-outside-toplevel
from WaveRNN.models.wavernn import Model from WaveRNN.models.wavernn import Model
print(" > Loading WaveRNN model ...") print(" > Loading WaveRNN model ...")
print(" | > model config: ", model_config) print(" | > model config: ", model_config)
print(" | > model file: ", model_file) print(" | > model file: ", model_file)
self.wavernn_config = load_config(model_config) self.wavernn_config = load_config(model_config)
# This is the default architecture we use for our models.
# You might need to update it
self.wavernn = Model( self.wavernn = Model(
rnn_dims=512, rnn_dims=512,
fc_dims=512, fc_dims=512,
@ -98,7 +101,8 @@ class Synthesizer(object):
self.wavernn.eval() self.wavernn.eval()
def load_pwgan(self, lib_path, model_file, model_config, use_cuda): def load_pwgan(self, lib_path, model_file, model_config, use_cuda):
sys.path.append(lib_path) # set this if TTS is not installed globally sys.path.append(lib_path) # set this if ParallelWaveGAN is not installed globally
#pylint: disable=import-outside-toplevel
from parallel_wavegan.models import ParallelWaveGANGenerator from parallel_wavegan.models import ParallelWaveGANGenerator
print(" > Loading PWGAN model ...") print(" > Loading PWGAN model ...")
print(" | > model config: ", model_config) print(" | > model config: ", model_config)