mirror of https://github.com/coqui-ai/TTS.git
change `list` to `List` in config
This commit is contained in:
parent
4df31f7fbd
commit
218af1d9a2
|
@ -1,5 +1,5 @@
|
||||||
from dataclasses import asdict, dataclass
|
from dataclasses import asdict, dataclass
|
||||||
from typing import List, Union
|
from typing import List
|
||||||
|
|
||||||
from coqpit import MISSING, Coqpit, check_argument
|
from coqpit import MISSING, Coqpit, check_argument
|
||||||
|
|
||||||
|
|
|
@ -1,4 +1,5 @@
|
||||||
from dataclasses import dataclass, field
|
from dataclasses import dataclass, field
|
||||||
|
from typing import List
|
||||||
|
|
||||||
from TTS.tts.configs.shared_configs import BaseTTSConfig
|
from TTS.tts.configs.shared_configs import BaseTTSConfig
|
||||||
|
|
||||||
|
@ -78,7 +79,7 @@ class AlignTTSConfig(BaseTTSConfig):
|
||||||
decoder_params: dict = field(
|
decoder_params: dict = field(
|
||||||
default_factory=lambda: {"hidden_channels_ffn": 1024, "num_heads": 2, "num_layers": 6, "dropout_p": 0.1}
|
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
|
ssim_alpha: float = 1.0
|
||||||
spec_loss_alpha: float = 1.0
|
spec_loss_alpha: float = 1.0
|
||||||
|
|
Loading…
Reference in New Issue