fix: normalize NumPy image outputs

Signed-off-by: Hongsheng Liu <liuhongsheng4@huawei.com>
This commit is contained in:
Hongsheng Liu committed 2026-08-11 12:14:23 +08:00
1 parent 53afbf3a0c
commit 5e3aa88e10
3 files changed
+77 -12

No files matched your search

@@ -604,18 +604,7 @@ def main():
raise ValueError("No output generated from omni.generate()")
logger.info(f"Outputs: {outputs}")
images = None
for output in outputs:
images = getattr(output, "images", None)
if images:
break
req_out = getattr(output, "request_output", None)
images = getattr(req_out, "images", None) if req_out is not None else None
if images:
break
if not images:
images = extract_images_from_outputs(outputs)
images = extract_images_from_outputs(outputs)
if not images:
raise ValueError("No images found in request_output")