From aaea77a180fcb9775b285f4440ec2f910bb5d738 Mon Sep 17 00:00:00 2001 From: Yueqian Lin <70319226+linyueqian@users.noreply.github.com> Date: Thu, 15 Jan 2026 06:14:34 -0700 Subject: [PATCH] [bugfix] Fix Wan2.2 I2V warmup failure by adding support_image_input attribute (#791) Signed-off-by: linyueqian --- vllm_omni/diffusion/models/wan2_2/pipeline_wan2_2_i2v.py | 3 ++- vllm_omni/diffusion/models/wan2_2/pipeline_wan2_2_ti2v.py | 3 ++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/vllm_omni/diffusion/models/wan2_2/pipeline_wan2_2_i2v.py b/vllm_omni/diffusion/models/wan2_2/pipeline_wan2_2_i2v.py index d72afc9ee..f3172e3d9 100644 --- a/vllm_omni/diffusion/models/wan2_2/pipeline_wan2_2_i2v.py +++ b/vllm_omni/diffusion/models/wan2_2/pipeline_wan2_2_i2v.py @@ -18,6 +18,7 @@ from vllm.model_executor.models.utils import AutoWeightsLoader from vllm_omni.diffusion.data import DiffusionOutput, OmniDiffusionConfig from vllm_omni.diffusion.distributed.utils import get_local_device from vllm_omni.diffusion.model_loader.diffusers_loader import DiffusersPipelineLoader +from vllm_omni.diffusion.models.interface import SupportImageInput from vllm_omni.diffusion.models.wan2_2.pipeline_wan2_2 import ( create_transformer_from_config, load_transformer_config, @@ -112,7 +113,7 @@ def get_wan22_i2v_pre_process_func( return pre_process_func -class Wan22I2VPipeline(nn.Module): +class Wan22I2VPipeline(nn.Module, SupportImageInput): """ Wan2.2 Image-to-Video Pipeline. diff --git a/vllm_omni/diffusion/models/wan2_2/pipeline_wan2_2_ti2v.py b/vllm_omni/diffusion/models/wan2_2/pipeline_wan2_2_ti2v.py index bee70a7a9..5351419ba 100644 --- a/vllm_omni/diffusion/models/wan2_2/pipeline_wan2_2_ti2v.py +++ b/vllm_omni/diffusion/models/wan2_2/pipeline_wan2_2_ti2v.py @@ -31,6 +31,7 @@ from vllm.model_executor.models.utils import AutoWeightsLoader from vllm_omni.diffusion.data import DiffusionOutput, OmniDiffusionConfig from vllm_omni.diffusion.distributed.utils import get_local_device from vllm_omni.diffusion.model_loader.diffusers_loader import DiffusersPipelineLoader +from vllm_omni.diffusion.models.interface import SupportImageInput from vllm_omni.diffusion.models.wan2_2.pipeline_wan2_2 import ( create_transformer_from_config, load_transformer_config, @@ -102,7 +103,7 @@ def get_wan22_ti2v_pre_process_func( return pre_process_func -class Wan22TI2VPipeline(nn.Module): +class Wan22TI2VPipeline(nn.Module, SupportImageInput): """ Wan2.2 Text-Image-to-Video (TI2V) Pipeline.