[Bugfix] Fix '--max-generated-image-size' cli args type (#1249)

Signed-off-by: ApsarasX <apsarax@outlook.com>
This commit is contained in:
ApsarasX authored and GitHub committed 2026-02-06 21:44:51 +08:00
1 parent 11a5a914ca
commit 10f841bb98
1 file changed
+1 -1
+1 -1
View File
@@ -245,7 +245,7 @@ class OmniServeCommand(CLISubcommand):
# Diffusion model mixed precision
omni_config_group.add_argument(
"--max-generated-image-size",
type=float,
type=int,
help="The max size of generate image (height * width).",
)
return serve_parser