Skip to content

Commit 15e5e12

Browse files
committed
Revert "fix(cache-dit): add step guard to prevent sao scm crash during warmup"
This reverts commit e4c5a1f.
1 parent e4c5a1f commit 15e5e12

File tree

1 file changed

+1
-5
lines changed

1 file changed

+1
-5
lines changed

vllm_omni/diffusion/cache/cache_dit_backend.py

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -395,11 +395,7 @@ def refresh_cache_context(pipeline: Any, num_inference_steps: int, verbose: bool
395395
num_inference_steps: New number of inference steps.
396396
verbose: Whether to log refresh operations.
397397
"""
398-
scm_policy = cache_config.get("scm_steps_mask_policy")
399-
is_supported_scm_step = num_inference_steps >= 8 or num_inference_steps in (4, 6)
400-
use_scm = scm_policy is not None and is_supported_scm_step
401-
402-
if not use_scm:
398+
if cache_config.scm_steps_mask_policy is None:
403399
cache_dit.refresh_context(pipeline.transformer, num_inference_steps=num_inference_steps, verbose=verbose)
404400
else:
405401
cache_dit.refresh_context(

0 commit comments

Comments
 (0)