From 04e496be3ee61ed500722a8bef30b56d8919291d Mon Sep 17 00:00:00 2001 From: Sy03 <1370724210@qq.com> Date: Mon, 21 Sep 2026 20:06:15 +0800 Subject: [PATCH] [Test] Pass explicit diffusion overrides in text encoder TP regression Signed-off-by: Sy03 <1370724210@qq.com> --- tests/engine/test_arg_utils.py | 11 +++++------ 1 file changed, 5 insertions(+), 6 deletions(-) diff --git a/tests/engine/test_arg_utils.py b/tests/engine/test_arg_utils.py index d40a9c226..5a3a1d655 100644 --- a/tests/engine/test_arg_utils.py +++ b/tests/engine/test_arg_utils.py @@ -563,19 +563,18 @@ def test_from_cli_args_preserves_text_encoder_tp_size(): def test_text_encoder_tp_size_reaches_default_diffusion_parallel_config(): """The preserved CLI value must land in DiffusionParallelConfig. - Mirrors the library flow: from_cli_args -> kwargs -> the generic - diffusion fallback, which resolves the field through - ``DiffusionParallelConfig.from_stage_overrides``. + Forward the explicitly supplied override after from_cli_args, without + injecting unrelated LLM defaults from the full EngineArgs dataclass into + the strict diffusion ingress. The generic diffusion fallback resolves + the field through ``DiffusionParallelConfig.from_stage_overrides``. """ - from dataclasses import asdict - from vllm_omni.config.config_factory import StageConfigFactory engine_args = OmniEngineArgs.from_cli_args( SimpleNamespace(text_encoder_tp_size=2), ) stage_cfg = StageConfigFactory.create_default_diffusion( - {k: v for k, v in asdict(engine_args).items() if v is not None}, + {"text_encoder_tp_size": engine_args.text_encoder_tp_size}, )[0] parallel_config = stage_cfg["engine_args"]["parallel_config"]