From 14fad9835d6a5a8f1c01bc620ec5917bfef8bd53 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=B1=AA=E5=BF=97=E9=B9=8F?= Date: Mon, 6 Jul 2026 01:26:09 +0800 Subject: [PATCH] [BugFix]: fix #3236 bug (#4877) Signed-off-by: princepride --- tests/e2e/offline_inference/test_magi_human_expansion.py | 7 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) diff --git a/tests/e2e/offline_inference/test_magi_human_expansion.py b/tests/e2e/offline_inference/test_magi_human_expansion.py index 5ce016988..e6b3c6a42 100644 --- a/tests/e2e/offline_inference/test_magi_human_expansion.py +++ b/tests/e2e/offline_inference/test_magi_human_expansion.py @@ -22,6 +22,7 @@ _OMNI_RUNNER_PARAM = ( { "init_timeout": 1200, "tensor_parallel_size": 2, + "vae_use_tiling": True, }, ) @@ -32,6 +33,7 @@ _OMNI_RUNNER_PARAM_OFFLOAD = ( "init_timeout": 1200, "tensor_parallel_size": 2, "enable_cpu_offload": True, + "vae_use_tiling": True, }, ) @@ -45,10 +47,6 @@ pytestmark = [ ), ] -_SKIP_ISSUE_3236 = pytest.mark.skip( - reason="https://github.com/vllm-project/vllm-omni/issues/3236", -) - def _validate_mp4(video_bytes: bytes, min_frames: int = 10) -> None: """Validate that the MP4 contains meaningful video and audio tracks.""" @@ -74,7 +72,6 @@ def _validate_mp4(video_bytes: bytes, min_frames: int = 10) -> None: container.close() -@_SKIP_ISSUE_3236 @hardware_test(res={"cuda": "H100"}, num_cards=2) def test_magi_human_e2e(omni_runner: OmniRunner): """End-to-end test for MagiHuman generating video and audio."""