Make style

This commit is contained in:
Eren Gölge 2022-02-03 15:51:16 +01:00
parent 33b98e6cc3
commit 8622226f3f
27 changed files with 50 additions and 26 deletions

1
.gitignore vendored
View File

@ -165,3 +165,4 @@ internal/*
*_phoneme.npy *_phoneme.npy
wandb wandb
depot/* depot/*
coqui_recipes/*

View File

@ -1,7 +1,8 @@
import os import os
from TTS.config import load_config, register_config
from trainer import Trainer, TrainerArgs from trainer import Trainer, TrainerArgs
from TTS.config import load_config, register_config
from TTS.tts.datasets import load_tts_samples from TTS.tts.datasets import load_tts_samples
from TTS.tts.models import setup_model from TTS.tts.models import setup_model

View File

@ -1,7 +1,8 @@
import os import os
from TTS.config import load_config, register_config
from trainer import Trainer, TrainerArgs from trainer import Trainer, TrainerArgs
from TTS.config import load_config, register_config
from TTS.utils.audio import AudioProcessor from TTS.utils.audio import AudioProcessor
from TTS.vocoder.datasets.preprocess import load_wav_data, load_wav_feat_data from TTS.vocoder.datasets.preprocess import load_wav_data, load_wav_feat_data
from TTS.vocoder.models import setup_model from TTS.vocoder.models import setup_model

View File

@ -995,8 +995,8 @@ class Vits(BaseTTS):
return {"figures": test_figures, "audios": test_audios} return {"figures": test_figures, "audios": test_audios}
def test_log(self, outputs: dict, logger: "Logger", assets: dict, steps: int) -> None: def test_log(self, outputs: dict, logger: "Logger", assets: dict, steps: int) -> None:
logger.test_audios(steps, outputs['audios'], self.ap.sample_rate) logger.test_audios(steps, outputs["audios"], self.ap.sample_rate)
logger.test_figures(steps, outputs['figures']) logger.test_figures(steps, outputs["figures"])
def get_optimizer(self) -> List: def get_optimizer(self) -> List:
"""Initiate and return the GAN optimizers based on the config parameters. """Initiate and return the GAN optimizers based on the config parameters.

View File

@ -11,6 +11,7 @@ def is_tool(name):
return which(name) is not None return which(name) is not None
# priority: espeakng > espeak # priority: espeakng > espeak
if is_tool("espeak-ng"): if is_tool("espeak-ng"):
_DEF_ESPEAK_LIB = "espeak-ng" _DEF_ESPEAK_LIB = "espeak-ng"
@ -116,7 +117,6 @@ class ESpeak(BasePhonemizer):
# ^ # ^
self.num_skip_chars = 1 self.num_skip_chars = 1
def auto_set_espeak_lib(self) -> None: def auto_set_espeak_lib(self) -> None:
if is_tool("espeak-ng"): if is_tool("espeak-ng"):
self._ESPEAK_LIB = "espeak-ng" self._ESPEAK_LIB = "espeak-ng"

View File

@ -1,6 +1,7 @@
import os import os
from trainer import Trainer, TrainerArgs from trainer import Trainer, TrainerArgs
from TTS.tts.configs.align_tts_config import AlignTTSConfig from TTS.tts.configs.align_tts_config import AlignTTSConfig
from TTS.tts.configs.shared_configs import BaseDatasetConfig from TTS.tts.configs.shared_configs import BaseDatasetConfig
from TTS.tts.datasets import load_tts_samples from TTS.tts.datasets import load_tts_samples

View File

@ -1,7 +1,8 @@
import os import os
from TTS.config.shared_configs import BaseAudioConfig, BaseDatasetConfig
from trainer import Trainer, TrainerArgs from trainer import Trainer, TrainerArgs
from TTS.config.shared_configs import BaseAudioConfig, BaseDatasetConfig
from TTS.tts.configs.fast_pitch_config import FastPitchConfig from TTS.tts.configs.fast_pitch_config import FastPitchConfig
from TTS.tts.datasets import load_tts_samples from TTS.tts.datasets import load_tts_samples
from TTS.tts.models.forward_tts import ForwardTTS from TTS.tts.models.forward_tts import ForwardTTS

View File

@ -1,7 +1,8 @@
import os import os
from TTS.config import BaseAudioConfig, BaseDatasetConfig
from trainer import Trainer, TrainerArgs from trainer import Trainer, TrainerArgs
from TTS.config import BaseAudioConfig, BaseDatasetConfig
from TTS.tts.configs.fast_speech_config import FastSpeechConfig from TTS.tts.configs.fast_speech_config import FastSpeechConfig
from TTS.tts.datasets import load_tts_samples from TTS.tts.datasets import load_tts_samples
from TTS.tts.models.forward_tts import ForwardTTS from TTS.tts.models.forward_tts import ForwardTTS

View File

@ -1,6 +1,7 @@
import os import os
from trainer import Trainer, TrainerArgs from trainer import Trainer, TrainerArgs
from TTS.utils.audio import AudioProcessor from TTS.utils.audio import AudioProcessor
from TTS.vocoder.configs import HifiganConfig from TTS.vocoder.configs import HifiganConfig
from TTS.vocoder.datasets.preprocess import load_wav_data from TTS.vocoder.datasets.preprocess import load_wav_data

View File

@ -1,6 +1,7 @@
import os import os
from trainer import Trainer, TrainerArgs from trainer import Trainer, TrainerArgs
from TTS.utils.audio import AudioProcessor from TTS.utils.audio import AudioProcessor
from TTS.vocoder.configs import MultibandMelganConfig from TTS.vocoder.configs import MultibandMelganConfig
from TTS.vocoder.datasets.preprocess import load_wav_data from TTS.vocoder.datasets.preprocess import load_wav_data

View File

@ -1,7 +1,8 @@
import os import os
from TTS.config import BaseAudioConfig, BaseDatasetConfig
from trainer import Trainer, TrainerArgs from trainer import Trainer, TrainerArgs
from TTS.config import BaseAudioConfig, BaseDatasetConfig
from TTS.tts.configs.speedy_speech_config import SpeedySpeechConfig from TTS.tts.configs.speedy_speech_config import SpeedySpeechConfig
from TTS.tts.datasets import load_tts_samples from TTS.tts.datasets import load_tts_samples
from TTS.tts.models.forward_tts import ForwardTTS from TTS.tts.models.forward_tts import ForwardTTS

View File

@ -1,7 +1,8 @@
import os import os
from TTS.config.shared_configs import BaseAudioConfig
from trainer import Trainer, TrainerArgs from trainer import Trainer, TrainerArgs
from TTS.config.shared_configs import BaseAudioConfig
from TTS.tts.configs.shared_configs import BaseDatasetConfig from TTS.tts.configs.shared_configs import BaseDatasetConfig
from TTS.tts.configs.tacotron2_config import Tacotron2Config from TTS.tts.configs.tacotron2_config import Tacotron2Config
from TTS.tts.datasets import load_tts_samples from TTS.tts.datasets import load_tts_samples

View File

@ -1,7 +1,8 @@
import os import os
from TTS.config.shared_configs import BaseAudioConfig
from trainer import Trainer, TrainerArgs from trainer import Trainer, TrainerArgs
from TTS.config.shared_configs import BaseAudioConfig
from TTS.tts.configs.shared_configs import BaseDatasetConfig from TTS.tts.configs.shared_configs import BaseDatasetConfig
from TTS.tts.configs.tacotron2_config import Tacotron2Config from TTS.tts.configs.tacotron2_config import Tacotron2Config
from TTS.tts.datasets import load_tts_samples from TTS.tts.datasets import load_tts_samples

View File

@ -1,6 +1,7 @@
import os import os
from trainer import Trainer, TrainerArgs from trainer import Trainer, TrainerArgs
from TTS.utils.audio import AudioProcessor from TTS.utils.audio import AudioProcessor
from TTS.vocoder.configs import UnivnetConfig from TTS.vocoder.configs import UnivnetConfig
from TTS.vocoder.datasets.preprocess import load_wav_data from TTS.vocoder.datasets.preprocess import load_wav_data

View File

@ -1,7 +1,8 @@
import os import os
from TTS.config.shared_configs import BaseAudioConfig
from trainer import Trainer, TrainerArgs from trainer import Trainer, TrainerArgs
from TTS.config.shared_configs import BaseAudioConfig
from TTS.tts.configs.shared_configs import BaseDatasetConfig from TTS.tts.configs.shared_configs import BaseDatasetConfig
from TTS.tts.configs.vits_config import VitsConfig from TTS.tts.configs.vits_config import VitsConfig
from TTS.tts.datasets import load_tts_samples from TTS.tts.datasets import load_tts_samples

View File

@ -1,6 +1,7 @@
import os import os
from trainer import Trainer, TrainerArgs from trainer import Trainer, TrainerArgs
from TTS.utils.audio import AudioProcessor from TTS.utils.audio import AudioProcessor
from TTS.vocoder.configs import WavegradConfig from TTS.vocoder.configs import WavegradConfig
from TTS.vocoder.datasets.preprocess import load_wav_data from TTS.vocoder.datasets.preprocess import load_wav_data

View File

@ -1,6 +1,7 @@
import os import os
from trainer import Trainer, TrainerArgs from trainer import Trainer, TrainerArgs
from TTS.utils.audio import AudioProcessor from TTS.utils.audio import AudioProcessor
from TTS.vocoder.configs import WavernnConfig from TTS.vocoder.configs import WavernnConfig
from TTS.vocoder.datasets.preprocess import load_wav_data from TTS.vocoder.datasets.preprocess import load_wav_data

View File

@ -1,8 +1,9 @@
import os import os
from glob import glob from glob import glob
from TTS.config.shared_configs import BaseAudioConfig
from trainer import Trainer, TrainerArgs from trainer import Trainer, TrainerArgs
from TTS.config.shared_configs import BaseAudioConfig
from TTS.tts.configs.shared_configs import BaseDatasetConfig from TTS.tts.configs.shared_configs import BaseDatasetConfig
from TTS.tts.configs.vits_config import VitsConfig from TTS.tts.configs.vits_config import VitsConfig
from TTS.tts.datasets import load_tts_samples from TTS.tts.datasets import load_tts_samples

View File

@ -1,7 +1,8 @@
import os import os
from TTS.config import BaseAudioConfig, BaseDatasetConfig
from trainer import Trainer, TrainerArgs from trainer import Trainer, TrainerArgs
from TTS.config import BaseAudioConfig, BaseDatasetConfig
from TTS.tts.configs.fast_pitch_config import FastPitchConfig from TTS.tts.configs.fast_pitch_config import FastPitchConfig
from TTS.tts.datasets import load_tts_samples from TTS.tts.datasets import load_tts_samples
from TTS.tts.models.forward_tts import ForwardTTS from TTS.tts.models.forward_tts import ForwardTTS

View File

@ -1,7 +1,8 @@
import os import os
from TTS.config import BaseAudioConfig, BaseDatasetConfig
from trainer import Trainer, TrainerArgs from trainer import Trainer, TrainerArgs
from TTS.config import BaseAudioConfig, BaseDatasetConfig
from TTS.tts.configs.fast_speech_config import FastSpeechConfig from TTS.tts.configs.fast_speech_config import FastSpeechConfig
from TTS.tts.datasets import load_tts_samples from TTS.tts.datasets import load_tts_samples
from TTS.tts.models.forward_tts import ForwardTTS from TTS.tts.models.forward_tts import ForwardTTS

View File

@ -1,7 +1,8 @@
import os import os
from TTS.config.shared_configs import BaseAudioConfig
from trainer import Trainer, TrainerArgs from trainer import Trainer, TrainerArgs
from TTS.config.shared_configs import BaseAudioConfig
from TTS.tts.configs.glow_tts_config import GlowTTSConfig from TTS.tts.configs.glow_tts_config import GlowTTSConfig
from TTS.tts.configs.shared_configs import BaseDatasetConfig from TTS.tts.configs.shared_configs import BaseDatasetConfig
from TTS.tts.datasets import load_tts_samples from TTS.tts.datasets import load_tts_samples

View File

@ -1,7 +1,8 @@
import os import os
from TTS.config import BaseAudioConfig, BaseDatasetConfig
from trainer import Trainer, TrainerArgs from trainer import Trainer, TrainerArgs
from TTS.config import BaseAudioConfig, BaseDatasetConfig
from TTS.tts.configs.speedy_speech_config import SpeedySpeechConfig from TTS.tts.configs.speedy_speech_config import SpeedySpeechConfig
from TTS.tts.datasets import load_tts_samples from TTS.tts.datasets import load_tts_samples
from TTS.tts.models.forward_tts import ForwardTTS from TTS.tts.models.forward_tts import ForwardTTS

View File

@ -1,7 +1,8 @@
import os import os
from TTS.config.shared_configs import BaseAudioConfig
from trainer import Trainer, TrainerArgs from trainer import Trainer, TrainerArgs
from TTS.config.shared_configs import BaseAudioConfig
from TTS.tts.configs.shared_configs import BaseDatasetConfig from TTS.tts.configs.shared_configs import BaseDatasetConfig
from TTS.tts.configs.tacotron_config import TacotronConfig from TTS.tts.configs.tacotron_config import TacotronConfig
from TTS.tts.datasets import load_tts_samples from TTS.tts.datasets import load_tts_samples

View File

@ -1,7 +1,8 @@
import os import os
from TTS.config.shared_configs import BaseAudioConfig
from trainer import Trainer, TrainerArgs from trainer import Trainer, TrainerArgs
from TTS.config.shared_configs import BaseAudioConfig
from TTS.tts.configs.shared_configs import BaseDatasetConfig from TTS.tts.configs.shared_configs import BaseDatasetConfig
from TTS.tts.configs.tacotron2_config import Tacotron2Config from TTS.tts.configs.tacotron2_config import Tacotron2Config
from TTS.tts.datasets import load_tts_samples from TTS.tts.datasets import load_tts_samples

View File

@ -1,7 +1,8 @@
import os import os
from TTS.config.shared_configs import BaseAudioConfig
from trainer import Trainer, TrainerArgs from trainer import Trainer, TrainerArgs
from TTS.config.shared_configs import BaseAudioConfig
from TTS.tts.configs.shared_configs import BaseDatasetConfig from TTS.tts.configs.shared_configs import BaseDatasetConfig
from TTS.tts.configs.tacotron2_config import Tacotron2Config from TTS.tts.configs.tacotron2_config import Tacotron2Config
from TTS.tts.datasets import load_tts_samples from TTS.tts.datasets import load_tts_samples

View File

@ -1,7 +1,8 @@
import os import os
from TTS.config.shared_configs import BaseAudioConfig
from trainer import Trainer, TrainerArgs from trainer import Trainer, TrainerArgs
from TTS.config.shared_configs import BaseAudioConfig
from TTS.tts.configs.shared_configs import BaseDatasetConfig from TTS.tts.configs.shared_configs import BaseDatasetConfig
from TTS.tts.configs.vits_config import VitsConfig from TTS.tts.configs.vits_config import VitsConfig
from TTS.tts.datasets import load_tts_samples from TTS.tts.datasets import load_tts_samples