From 10f841bb9840dae3cb70b68bcdace7cadd52513b Mon Sep 17 00:00:00 2001
From: ApsarasX <apsarax@outlook.com>
Date: Fri, 6 Feb 2026 21:44:51 +0800
Subject: [PATCH] [Bugfix] Fix '--max-generated-image-size' cli args type
 (#1249)

Signed-off-by: ApsarasX <apsarax@outlook.com>
---
 vllm_omni/entrypoints/cli/serve.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/vllm_omni/entrypoints/cli/serve.py b/vllm_omni/entrypoints/cli/serve.py
index acefc081b..ac03de143 100644
--- a/vllm_omni/entrypoints/cli/serve.py
+++ b/vllm_omni/entrypoints/cli/serve.py
@@ -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