mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-27 02:34:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

minor fixes

Browse Source
This commit is contained in:
Fangjun Kuang 2024-03-12 21:59:47 +08:00
parent 7f9cbf1ce2
commit a92c6df76a
4 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
egs/ljspeech/TTS/vits/export-onnx.py
View File

@ -76,7 +76,8 @@ def get_parser():
parser.add_argument( parser.add_argument(
"--model-type", "--model-type",
type=str, type=str,
default="", default="high",
choices=["low", "medium", "high"],
help="""If not empty, valid values are: low, medium, high. help="""If not empty, valid values are: low, medium, high.
It controls the model size. low -> runs faster. It controls the model size. low -> runs faster.
""", """,

3
egs/ljspeech/TTS/vits/infer.py
View File

@ -75,7 +75,8 @@ def get_parser():
parser.add_argument( parser.add_argument(
"--model-type", "--model-type",
type=str, type=str,
default="", default="high",
choices=["low", "medium", "high"],
help="""If not empty, valid values are: low, medium, high. help="""If not empty, valid values are: low, medium, high.
It controls the model size. low -> runs faster. It controls the model size. low -> runs faster.
""", """,

3
egs/ljspeech/TTS/vits/train.py
View File

@ -156,7 +156,8 @@ def get_parser():
parser.add_argument( parser.add_argument(
"--model-type", "--model-type",
type=str, type=str,
default="", default="high",
choices=["low", "medium", "high"],
help="""If not empty, valid values are: low, medium, high. help="""If not empty, valid values are: low, medium, high.
It controls the model size. low -> runs faster. It controls the model size. low -> runs faster.
""", """,

2
egs/ljspeech/TTS/vits/vits.py
View File

@ -66,7 +66,7 @@ HIGH_CONFIG = {
"decoder_upsample_kernel_sizes": (16, 16, 4, 4), "decoder_upsample_kernel_sizes": (16, 16, 4, 4),
"decoder_resblock_kernel_sizes": (3, 7, 11), "decoder_resblock_kernel_sizes": (3, 7, 11),
"decoder_resblock_dilations": ((1, 3, 5), (1, 3, 5), (1, 3, 5)), "decoder_resblock_dilations": ((1, 3, 5), (1, 3, 5), (1, 3, 5)),
"text_encoder_cnn_module_kernel": 3, "text_encoder_cnn_module_kernel": 5,
} }