From 218af1d9a2a4b270a09a42af7502d473a833c172 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Eren=20G=C3=B6lge?= Date: Tue, 18 May 2021 14:50:29 +0200 Subject: [PATCH] change `list` to `List` in config --- TTS/config/shared_configs.py | 2 +- TTS/tts/configs/align_tts_config.py | 3 ++- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/TTS/config/shared_configs.py b/TTS/config/shared_configs.py index 3aa80c98..69f1ee31 100644 --- a/TTS/config/shared_configs.py +++ b/TTS/config/shared_configs.py @@ -1,5 +1,5 @@ from dataclasses import asdict, dataclass -from typing import List, Union +from typing import List from coqpit import MISSING, Coqpit, check_argument diff --git a/TTS/tts/configs/align_tts_config.py b/TTS/tts/configs/align_tts_config.py index 84e0ba13..2956d935 100644 --- a/TTS/tts/configs/align_tts_config.py +++ b/TTS/tts/configs/align_tts_config.py @@ -1,4 +1,5 @@ from dataclasses import dataclass, field +from typing import List from TTS.tts.configs.shared_configs import BaseTTSConfig @@ -78,7 +79,7 @@ class AlignTTSConfig(BaseTTSConfig): decoder_params: dict = field( default_factory=lambda: {"hidden_channels_ffn": 1024, "num_heads": 2, "num_layers": 6, "dropout_p": 0.1} ) - phase_start_steps: list = None + phase_start_steps: List[int] = None ssim_alpha: float = 1.0 spec_loss_alpha: float = 1.0