storytold/storyteller-ml
2
0
Fork 0
mirror of https://github.com/storytold/storyteller-ml.git synced 2026-10-17 00:23:14 +00:00
Code Issues Packages Projects Releases Wiki Activity

Remove logger.flush()

Browse Source
This commit is contained in:
Brandon Thomas
2022-07-14 20:29:56 -04:00
parent 0cc56be156
commit 4524389ced
1 changed files with 0 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tts/vocodes_common.py
-4
View File
@@ -218,7 +218,6 @@ class TacotronWaveglowPipeline:
return
LOGGER.info('Dumping previous Hifi-gan model from memory: {}'.format(self.hifigan_filename))
LOGGER.info('Loading Hifi-gan model into memory: {}'.format(checkpoint_path))
LOGGER.flush()
hifigan, h, denoiser = load_hifigan_model(checkpoint_path, False)
self.hifigan = hifigan
self.hifigan_denoiser = denoiser
@@ -231,7 +230,6 @@ class TacotronWaveglowPipeline:
return
LOGGER.info('Dumping previous Hifi-gan (SuperRes) model from memory: {}'.format(self.hifigan_super_resolution_filename))
LOGGER.info('Loading Hifi-gan (SuperRes) model into memory: {}'.format(checkpoint_path))
LOGGER.flush()
hifigan, h, denoiser = load_hifigan_model(checkpoint_path, True)
self.hifigan_super_resolution = hifigan
self.hifigan_super_resolution_denoiser = denoiser
@@ -244,7 +242,6 @@ class TacotronWaveglowPipeline:
return
LOGGER.info('Dumping previous waveglow model from memory: {}'.format(self.waveglow_filename))
LOGGER.info('Loading waveglow model into memory: {}'.format(checkpoint_path))
LOGGER.flush()
self.waveglow = load_waveglow_model(checkpoint_path)
self.waveglow_filename = checkpoint_path
@@ -253,7 +250,6 @@ class TacotronWaveglowPipeline:
LOGGER.info('Tacotron model already loaded into memory: {}'.format(checkpoint_path))
return
LOGGER.info('Loading tacotron model into memory: {}'.format(checkpoint_path))
LOGGER.flush()
model = load_tacotron_model(checkpoint_path)
self.tacotron_model_cache[checkpoint_path] = model
LOGGER.info('Now {} synthesizer models in memory'.format(len(self.tacotron_model_cache)))