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

load ParallelWaveGAN from PYTHONPATH if pwgan_lib_path is unset

This commit is contained in:
Jörg Thalheim 2020-06-13 19:34:26 +01:00
parent 7e799d58d6
commit 3e5ce85bf7
No known key found for this signature in database
GPG Key ID: 003F2096411B5F92
1 changed files with 9 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
server/synthesizer.py
View File

@ -37,7 +37,7 @@ class Synthesizer(object):
if self.config.wavernn_lib_path: if self.config.wavernn_lib_path:
self.load_wavernn(self.config.wavernn_lib_path, self.config.wavernn_file, self.load_wavernn(self.config.wavernn_lib_path, self.config.wavernn_file,
self.config.wavernn_config, self.config.use_cuda) self.config.wavernn_config, self.config.use_cuda)
if self.config.pwgan_lib_path: if self.config.pwgan_file:
self.load_pwgan(self.config.pwgan_lib_path, self.config.pwgan_file, self.load_pwgan(self.config.pwgan_lib_path, self.config.pwgan_file,
self.config.pwgan_config, self.config.use_cuda) self.config.pwgan_config, self.config.use_cuda)
@ -113,9 +113,14 @@ 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 ParallelWaveGAN is not installed globally if lib_path:
#pylint: disable=import-outside-toplevel # set this if ParallelWaveGAN is not installed globally
from parallel_wavegan.models import ParallelWaveGANGenerator sys.path.append(lib_path)
try:
#pylint: disable=import-outside-toplevel
from parallel_wavegan.models import ParallelWaveGANGenerator
except ImportError as e:
raise RuntimeError(f"cannot import parallel-wavegan, either install it or set its directory using the --pwgan_lib_path command line argument: {e}")
print(" > Loading PWGAN model ...") print(" > Loading PWGAN model ...")
print(" | > model config: ", model_config) print(" | > model config: ", model_config)
print(" | > model file: ", model_file) print(" | > model file: ", model_file)