From 7c7ec07cdd7166d6b7293fa07d41bb51d70cf2dc Mon Sep 17 00:00:00 2001 From: "Huang, Zeyu" <11222265+fhfuih@users.noreply.github.com> Date: Tue, 17 Mar 2026 14:43:31 +0800 Subject: [PATCH] [Bugfix] NextStep cannot launch with offline example text_to_image.py Signed-off-by: Huang, Zeyu <11222265+fhfuih@users.noreply.github.com> --- vllm_omni/entrypoints/omni_diffusion.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/vllm_omni/entrypoints/omni_diffusion.py b/vllm_omni/entrypoints/omni_diffusion.py index d70aa5c5203..673897204d3 100644 --- a/vllm_omni/entrypoints/omni_diffusion.py +++ b/vllm_omni/entrypoints/omni_diffusion.py @@ -93,6 +93,8 @@ def __init__(self, od_config: OmniDiffusionConfig | None = None, **kwargs): elif model_type == "nextstep": if od_config.model_class_name is None: pipeline_class = "NextStep11Pipeline" + else: + pipeline_class = od_config.model_class_name elif model_type == "glm-image" or "GlmImageForConditionalGeneration" in architectures: pipeline_class = "GlmImagePipeline" elif architectures and len(architectures) == 1: