From 76722980753776698afa2151e7a11e596c78c061 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Zeyu=20Huang=20=7C=20=E9=BB=83=E6=BE=A4=E5=AE=87?= <11222265+fhfuih@users.noreply.github.com> Date: Wed, 4 Feb 2026 09:36:57 +0800 Subject: [PATCH] bugfix: /images/edits endpoint fails pipeline data format check (#1141) Signed-off-by: Huang, Zeyu <11222265+fhfuih@users.noreply.github.com> Co-authored-by: Hongsheng Liu --- .../longcat_image/pipeline_longcat_image_edit.py | 14 +++++++++----- .../models/qwen_image/pipeline_qwen_image_edit.py | 14 +++++++++----- .../qwen_image/pipeline_qwen_image_layered.py | 14 +++++++++----- 3 files changed, 27 insertions(+), 15 deletions(-) diff --git a/vllm_omni/diffusion/models/longcat_image/pipeline_longcat_image_edit.py b/vllm_omni/diffusion/models/longcat_image/pipeline_longcat_image_edit.py index a34a2cca3..3ba5e3448 100644 --- a/vllm_omni/diffusion/models/longcat_image/pipeline_longcat_image_edit.py +++ b/vllm_omni/diffusion/models/longcat_image/pipeline_longcat_image_edit.py @@ -72,11 +72,15 @@ def get_longcat_image_edit_pre_process_func( if "additional_information" not in prompt: prompt["additional_information"] = {} - if raw_image is None or isinstance(raw_image, list): - raise ValueError( - """Received no image or a list of image. Only a single image is supported by this model.""" - """Please correctly set `"multi_modal_data": {"image": , …}`""" - ) + if not raw_image: # None or empty list + raise ValueError("""Received no input image. This model requires one input image to run.""") + elif isinstance(raw_image, list): + if len(raw_image) > 1: + raise ValueError( + """Received multiple input images. Only a single image is supported by this model.""" + ) + else: + raw_image = raw_image[0] if isinstance(raw_image, str): image = PIL.Image.open(raw_image) diff --git a/vllm_omni/diffusion/models/qwen_image/pipeline_qwen_image_edit.py b/vllm_omni/diffusion/models/qwen_image/pipeline_qwen_image_edit.py index b9abb7549..082c9e754 100644 --- a/vllm_omni/diffusion/models/qwen_image/pipeline_qwen_image_edit.py +++ b/vllm_omni/diffusion/models/qwen_image/pipeline_qwen_image_edit.py @@ -75,11 +75,15 @@ def get_qwen_image_edit_pre_process_func( prompt["additional_information"] = {} # Only handles single image - if raw_image is None or isinstance(raw_image, list): - raise ValueError( - """Received no image or a list of image. Only a single image is supported by this model.""" - """Please correctly set `"multi_modal_data": {"image": , …}`""" - ) + if not raw_image: # None or empty list + raise ValueError("""Received no input image. This model requires one input image to run.""") + elif isinstance(raw_image, list): + if len(raw_image) > 1: + raise ValueError( + """Received multiple input images. Only a single image is supported by this model.""" + ) + else: + raw_image = raw_image[0] if isinstance(raw_image, str): image = PIL.Image.open(raw_image) diff --git a/vllm_omni/diffusion/models/qwen_image/pipeline_qwen_image_layered.py b/vllm_omni/diffusion/models/qwen_image/pipeline_qwen_image_layered.py index d200764eb..dbe0bfe4f 100644 --- a/vllm_omni/diffusion/models/qwen_image/pipeline_qwen_image_layered.py +++ b/vllm_omni/diffusion/models/qwen_image/pipeline_qwen_image_layered.py @@ -79,11 +79,15 @@ def get_qwen_image_layered_pre_process_func( if "additional_information" not in prompt: prompt["additional_information"] = {} - if raw_image is None or isinstance(raw_image, list): - raise ValueError( - """Received no image or a list of image. Only a single image is supported by this model.""" - """Please correctly set `"multi_modal_data": {"image": , …}`""" - ) + if not raw_image: # None or empty list + raise ValueError("""Received no input image. This model requires one input image to run.""") + elif isinstance(raw_image, list): + if len(raw_image) > 1: + raise ValueError( + """Received multiple input images. Only a single image is supported by this model.""" + ) + else: + raw_image = raw_image[0] if isinstance(raw_image, str): image = PIL.Image.open(raw_image)