From 23e243378f907ebf0ce0e52a694202d36219b9a1 Mon Sep 17 00:00:00 2001 From: Alex Brooks Date: Fri, 22 May 2026 05:56:41 -0600 Subject: [PATCH] [Bugfix] Set separate CFG flag in Helios for CacheDiT (#3756) Signed-off-by: Alex Brooks --- tests/diffusion/cache/test_cache_dit.py | 1 + vllm_omni/diffusion/cache/cache_dit_backend.py | 2 +- 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/diffusion/cache/test_cache_dit.py b/tests/diffusion/cache/test_cache_dit.py index 0b7ef7235..570c4f238 100644 --- a/tests/diffusion/cache/test_cache_dit.py +++ b/tests/diffusion/cache/test_cache_dit.py @@ -16,6 +16,7 @@ pytestmark = [pytest.mark.core_model, pytest.mark.cpu] SEPARATE_CFG_ENABLERS = [ cd_backend.enable_cache_for_ltx2, + cd_backend.enable_cache_for_helios, cd_backend.enable_cache_for_wan22, cd_backend.enable_cache_for_longcat_image, ] diff --git a/vllm_omni/diffusion/cache/cache_dit_backend.py b/vllm_omni/diffusion/cache/cache_dit_backend.py index daf9b3939..3f8229722 100644 --- a/vllm_omni/diffusion/cache/cache_dit_backend.py +++ b/vllm_omni/diffusion/cache/cache_dit_backend.py @@ -1413,7 +1413,7 @@ def enable_cache_for_helios(pipeline: Any, cache_config: Any) -> Callable[[int], params_modifiers=[ ParamsModifier(cache_config=db_cache_config, calibrator_config=calibrator_config), ], - has_separate_cfg=False, + has_separate_cfg=True, ), cache_config=db_cache_config, calibrator_config=calibrator_config,