diff --git a/tests/e2e/accuracy/sana_video/conftest.py b/tests/e2e/accuracy/sana_video/conftest.py new file mode 100644 index 000000000..427dae468 --- /dev/null +++ b/tests/e2e/accuracy/sana_video/conftest.py @@ -0,0 +1,26 @@ +# SPDX-License-Identifier: Apache-2.0 +# SPDX-FileCopyrightText: Copyright contributors to the vLLM-Omni project + +import pytest + + +@pytest.fixture +def sana_model_parallel(tmp_path): + """Construct native parallel layers with real single-rank GPU groups.""" + from vllm_omni.diffusion.distributed.parallel_state import ( + destroy_distributed_env, + init_distributed_environment, + initialize_model_parallel, + ) + + init_distributed_environment( + world_size=1, + rank=0, + local_rank=0, + distributed_init_method=f"file://{tmp_path / 'sana-distributed'}", + ) + try: + initialize_model_parallel() + yield + finally: + destroy_distributed_env() diff --git a/tests/e2e/accuracy/sana_video/test_sana_video_golden.py b/tests/e2e/accuracy/sana_video/test_sana_video_golden.py index cb6ffe6c9..87c91d093 100644 --- a/tests/e2e/accuracy/sana_video/test_sana_video_golden.py +++ b/tests/e2e/accuracy/sana_video/test_sana_video_golden.py @@ -202,6 +202,7 @@ def test_sana_video_transformer_matches_frozen_golden( variant: str, _hardware, accuracy_artifact_root: Path, + sana_model_parallel, ) -> None: del _hardware from vllm.config import LoadConfig diff --git a/tests/e2e/accuracy/sana_video/test_sana_video_pipeline_alignment.py b/tests/e2e/accuracy/sana_video/test_sana_video_pipeline_alignment.py index aabe166d2..787114529 100644 --- a/tests/e2e/accuracy/sana_video/test_sana_video_pipeline_alignment.py +++ b/tests/e2e/accuracy/sana_video/test_sana_video_pipeline_alignment.py @@ -189,6 +189,7 @@ def _cfg_prediction( def test_sana_video_quick_pipeline_stages_match_diffusers( variant: str, _hardware, + sana_model_parallel, ) -> None: """Compare the first denoising step without replacing native attention.