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
wandb
depot/*
coqui_recipes/*

View File

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

View File

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

View File

@ -995,8 +995,8 @@ class Vits(BaseTTS):
return {"figures": test_figures, "audios": test_audios}
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_figures(steps, outputs['figures'])
logger.test_audios(steps, outputs["audios"], self.ap.sample_rate)
logger.test_figures(steps, outputs["figures"])
def get_optimizer(self) -> List:
"""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
# priority: espeakng > espeak
if is_tool("espeak-ng"):
_DEF_ESPEAK_LIB = "espeak-ng"
@ -116,7 +117,6 @@ class ESpeak(BasePhonemizer):
# ^
self.num_skip_chars = 1
def auto_set_espeak_lib(self) -> None:
if is_tool("espeak-ng"):
self._ESPEAK_LIB = "espeak-ng"

View File

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

View File

@ -1,7 +1,8 @@
import os
from TTS.config.shared_configs import BaseAudioConfig, BaseDatasetConfig
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.datasets import load_tts_samples
from TTS.tts.models.forward_tts import ForwardTTS

View File

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

View File

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

View File

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

View File

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

View File

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

View File

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

View File

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

View File

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

View File

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

View File

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

View File

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

View File

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

View File

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

View File

@ -1,7 +1,8 @@
import os
from TTS.config.shared_configs import BaseAudioConfig
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.shared_configs import BaseDatasetConfig
from TTS.tts.datasets import load_tts_samples

View File

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

View File

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

View File

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

View File

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

View File

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