[Bugfix] Fix '--max-generated-image-size' cli args type (#1249)
Signed-off-by: ApsarasX <apsarax@outlook.com>
This commit is contained in:
1 parent
11a5a914ca
commit
10f841bb98
1 file changed
+1
-1
@@ -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
|
||||
|
||||
Reference in new issue
Block a user