From b63f42ec74dc03ec2d92beebbcb33b7b451fbee7 Mon Sep 17 00:00:00 2001 From: Alex Brooks Date: Wed, 20 May 2026 01:54:06 +0000 Subject: [PATCH] fix separate cfg flag helios 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 0b7ef723585..570c4f23840 100644 --- a/tests/diffusion/cache/test_cache_dit.py +++ b/tests/diffusion/cache/test_cache_dit.py @@ -16,6 +16,7 @@ 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 daf9b3939aa..3f82297220f 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,