Frontend] Support --dtype in qwen3_omni offline e2e script (#2246)

Signed-off-by: reidliu41 <reid201711@gmail.com>
This commit is contained in:
Reid authored and GitHub committed 2026-03-27 03:12:11 +00:00
1 parent 4be54ccec4
commit 6f87a9382f
1 file changed
+7
@@ -296,6 +296,7 @@ def main(args):
omni = Omni(
model=model_name,
dtype=args.dtype,
stage_configs_path=args.stage_configs_path,
log_stats=args.log_stats,
stage_init_timeout=args.stage_init_timeout,
@@ -555,6 +556,12 @@ def parse_args():
default=None,
help="List of stage IDs to profile. If not set, profiles all stages.",
)
parser.add_argument(
"--dtype",
type=str,
default="auto",
help="Model dtype (auto, half, float16, bfloat16, float, float32).",
)
return parser.parse_args()