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

[cherry-pick][release/v.0.18.0.post1]remove duplicate video preprocess in Wan2.2 pipeline (#3005)

Browse Source 
Signed-off-by: bjf-frz <frz123db@gmail.com>
This commit is contained in:
bjf-frz authored and GitHub committed 2026-04-22 11:09:27 +08:00
1 parent caa26e1d04
commit 6d76e7d343
3 files changed
-21
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.py
-7
View File
@@ -148,9 +148,6 @@ def get_wan22_pre_process_func(
):
"""Pre-process function for Wan2.2: optionally load and resize input image for I2V mode."""
import numpy as np
from diffusers.video_processor import VideoProcessor
video_processor = VideoProcessor(vae_scale_factor=8)
def pre_process_func(request: OmniDiffusionRequest) -> OmniDiffusionRequest:
for i, prompt in enumerate(request.prompts):
@@ -194,10 +191,6 @@ def get_wan22_pre_process_func(
)
prompt["multi_modal_data"]["image"] = image # type: ignore # key existence already checked above
# Preprocess for VAE
prompt["additional_information"]["preprocessed_image"] = video_processor.preprocess(
image, height=request.sampling_params.height, width=request.sampling_params.width
)
request.prompts[i] = prompt
return request
vllm_omni/diffusion/models/wan2_2/pipeline_wan2_2_i2v.py
-7
View File
@@ -99,9 +99,6 @@ def get_wan22_i2v_pre_process_func(
od_config: OmniDiffusionConfig,
):
"""Pre-process function for I2V: load and resize input image."""
from diffusers.video_processor import VideoProcessor
video_processor = VideoProcessor(vae_scale_factor=8)
def pre_process_func(request: OmniDiffusionRequest) -> OmniDiffusionRequest:
for i, prompt in enumerate(request.prompts):
@@ -147,10 +144,6 @@ def get_wan22_i2v_pre_process_func(
)
prompt["multi_modal_data"]["image"] = image # type: ignore # key existence already checked above
# Preprocess for VAE
prompt["additional_information"]["preprocessed_image"] = video_processor.preprocess(
image, height=request.sampling_params.height, width=request.sampling_params.width
)
request.prompts[i] = prompt
return request
vllm_omni/diffusion/models/wan2_2/pipeline_wan2_2_ti2v.py
-7
View File
@@ -85,9 +85,6 @@ def get_wan22_ti2v_pre_process_func(
od_config: OmniDiffusionConfig,
):
"""Pre-process function for TI2V: optionally load and resize input image."""
from diffusers.video_processor import VideoProcessor
video_processor = VideoProcessor(vae_scale_factor=8)
def pre_process_func(request: OmniDiffusionRequest) -> OmniDiffusionRequest:
for i, prompt in enumerate(request.prompts):
@@ -133,10 +130,6 @@ def get_wan22_ti2v_pre_process_func(
)
prompt["multi_modal_data"]["image"] = image # type: ignore # key existence already checked above
# Preprocess for VAE
prompt["additional_information"]["preprocessed_image"] = video_processor.preprocess(
image, height=request.sampling_params.height, width=request.sampling_params.width
)
request.prompts[i] = prompt
return request