change `list` to `List` in config

This commit is contained in:
Eren Gölge 2021-05-18 14:50:29 +02:00
parent 4df31f7fbd
commit 218af1d9a2
2 changed files with 3 additions and 2 deletions

View File

@ -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

View File

@ -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