diff --git a/tts/vocodes_common.py b/tts/vocodes_common.py index 6aaa36c..a7b4cc6 100644 --- a/tts/vocodes_common.py +++ b/tts/vocodes_common.py @@ -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)))