[Refactor][1/N]diffusion/cache: cache-dit (#5226)

Signed-off-by: yangjianjuan <66548561+yangjianjuan@users.noreply.github.com>

tests/diffusion/models/cosmos3/test_cosmos3_pipeline.py

@@ -465,7 +465,7 @@ def test_edge_forward_rejects_unsupported_modes(
 
 
 def test_pipeline_registered_and_exported() -> None:
-    from vllm_omni.diffusion.cache.cache_dit_backend import CUSTOM_DIT_ENABLERS
+    from vllm_omni.diffusion.cache.cachedit import CUSTOM_DIT_ENABLERS
     from vllm_omni.diffusion.models import cosmos3
     from vllm_omni.diffusion.models.cosmos3.pipeline_cosmos3 import Cosmos3OmniDiffusersPipeline
     from vllm_omni.diffusion.models.progress_bar import ProgressBarMixin

tests/diffusion/models/ltx2/test_ltx2_pipeline.py

@@ -731,7 +731,7 @@ class TestRegistryIntegration:
 
     def test_cache_dit_for_ltx2_does_not_have_custom_enablers_registered(self):
         """Pipeline variants are *not* registered in CUSTOM_DIT_ENABLERS."""
-        from vllm_omni.diffusion.cache.cache_dit_backend import CUSTOM_DIT_ENABLERS
+        from vllm_omni.diffusion.cache.cachedit import CUSTOM_DIT_ENABLERS
 
         # NOTE: We used to have custom enablers for this model, but refactored to handle
         # it more generically. Now we only need to ensure it has git cache adapter config.

tests/diffusion/models/ltx2/test_ltx2_transformer.py

@@ -4,7 +4,7 @@
 import pytest
 from torch import nn
 
-from vllm_omni.diffusion.cache.cache_dit_backend import CacheDiTAdapterConfig
+from vllm_omni.diffusion.cache.cachedit import CacheDiTAdapterConfig
 from vllm_omni.diffusion.models.ltx2.ltx2_transformer import LTX2VideoTransformer3DModel, _make_rms_norm
 
 pytestmark = [pytest.mark.core_model, pytest.mark.cpu]