anodyine/vllm-omni
1
0 Fork 0
Code Issues 4 Pull Requests 2 Actions Packages Projects Releases Wiki Activity

[bugfix] Fix Wan2.2 I2V warmup failure by adding support_image_input attribute (#791)

Browse Source 
Signed-off-by: linyueqian <linyueqian@outlook.com>
This commit is contained in:
Yueqian Lin authored and GitHub committed 2026-01-15 13:14:34 +00:00
1 parent 2d5faf3634
commit aaea77a180
2 files changed
+4 -2
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

vllm_omni/diffusion/models/wan2_2/pipeline_wan2_2_i2v.py
+2 -1
View File
@@ -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.
vllm_omni/diffusion/models/wan2_2/pipeline_wan2_2_ti2v.py
+2 -1
View File
@@ -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.