minor fixes

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

View File

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

View File

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

View File

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

View File

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