From 9af0331b21edcefa872727eee854b7318f597a52 Mon Sep 17 00:00:00 2001 From: tzhouam Date: Sat, 5 Sep 2026 18:47:27 +0800 Subject: [PATCH] rebase: align vllm-omni with vLLM d2906cc19586 Signed-off-by: tzhouam --- tests/e2e/accuracy/test_diffusers_backend_similarity.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/e2e/accuracy/test_diffusers_backend_similarity.py b/tests/e2e/accuracy/test_diffusers_backend_similarity.py index be3d7f84c..b12e9c1e7 100644 --- a/tests/e2e/accuracy/test_diffusers_backend_similarity.py +++ b/tests/e2e/accuracy/test_diffusers_backend_similarity.py @@ -164,6 +164,7 @@ def _run_diffusers_wan22_i2v(*, model: str, output_path: Path, conditioning_imag ) pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config, flow_shift=FLOW_SHIFT) pipe.to("cuda") + _set_matched_attention_backend(pipe) _diffusers_dummy_run(pipe)