cache-dit 0.2.28__py3-none-any.whl → 0.2.30__py3-none-any.whl

cache_dit/cache_factory/cache_contexts/cache_manager.py

@@ -63,6 +63,20 @@ class CachedContextManager:
             _context = self.new_context(*args, **kwargs)
         return _context
 
+    def remove_context(self, cached_context: CachedContext | str):
+        if isinstance(cached_context, CachedContext):
+            cached_context.clear_buffers()
+            if cached_context.name in self._cached_context_manager:
+                del self._cached_context_manager[cached_context.name]
+        else:
+            if cached_context in self._cached_context_manager:
+                self._cached_context_manager[cached_context].clear_buffers()
+                del self._cached_context_manager[cached_context]
+
+    def clear_contexts(self):
+        for cached_context in self._cached_context_manager:
+            self.remove_context(cached_context)
+
     @contextlib.contextmanager
     def enter_context(self, cached_context: CachedContext | str):
         old_cached_context = self._current_context
@@ -719,17 +733,15 @@ class CachedContextManager:
                     encoder_prefix
                 )
 
-            assert (
-                encoder_hidden_states_prev is not None
-            ), f"{prefix}_encoder_buffer must be set before"
+            if encoder_hidden_states_prev is not None:
 
-            if self.is_encoder_cache_residual():
-                encoder_hidden_states = (
-                    encoder_hidden_states_prev + encoder_hidden_states
-                )
-            else:
-                # If encoder cache is not residual, we use the encoder hidden states directly
-                encoder_hidden_states = encoder_hidden_states_prev
+                if self.is_encoder_cache_residual():
+                    encoder_hidden_states = (
+                        encoder_hidden_states_prev + encoder_hidden_states
+                    )
+                else:
+                    # If encoder cache is not residual, we use the encoder hidden states directly
+                    encoder_hidden_states = encoder_hidden_states_prev
 
             encoder_hidden_states = encoder_hidden_states.contiguous()
 

cache_dit/cache_factory/cache_interface.py

@@ -1,4 +1,4 @@
-from typing import Any, Tuple, List
+from typing import Any, Tuple, List, Union
 from diffusers import DiffusionPipeline
 from cache_dit.cache_factory.cache_types import CacheType
 from cache_dit.cache_factory.block_adapters import BlockAdapter
@@ -12,7 +12,10 @@ logger = init_logger(__name__)
 
 def enable_cache(
     # DiffusionPipeline or BlockAdapter
-    pipe_or_adapter: DiffusionPipeline | BlockAdapter | Any,
+    pipe_or_adapter: Union[
+        DiffusionPipeline,
+        BlockAdapter,
+    ],
     # Cache context kwargs
     Fn_compute_blocks: int = 8,
     Bn_compute_blocks: int = 0,
@@ -30,7 +33,10 @@ def enable_cache(
     taylorseer_cache_type: str = "residual",
     taylorseer_order: int = 2,
     **other_cache_context_kwargs,
-) -> DiffusionPipeline | Any:
+) -> Union[
+    DiffusionPipeline,
+    BlockAdapter,
+]:
     r"""
     Unified Cache API for  almost Any Diffusion Transformers (with Transformer Blocks
     that match the specific Input and Output patterns).
@@ -100,11 +106,11 @@ def enable_cache(
     >>> cache_dit.enable_cache(pipe) # One-line code with default cache options.
     >>> output = pipe(...) # Just call the pipe as normal.
     >>> stats = cache_dit.summary(pipe) # Then, get the summary of cache acceleration stats.
+    >>> cache_dit.disable_cache(pipe) # Disable cache and run original pipe.
     """
-
     # Collect cache context kwargs
     cache_context_kwargs = other_cache_context_kwargs.copy()
-    if cache_type := cache_context_kwargs.get("cache_type", None):
+    if (cache_type := cache_context_kwargs.get("cache_type", None)) is not None:
         if cache_type == CacheType.NONE:
             return pipe_or_adapter
 
@@ -129,16 +135,9 @@ def enable_cache(
     cache_context_kwargs["taylorseer_cache_type"] = taylorseer_cache_type
     cache_context_kwargs["taylorseer_order"] = taylorseer_order
 
-    if isinstance(pipe_or_adapter, BlockAdapter):
-        return CachedAdapter.apply(
-            pipe=None,
-            block_adapter=pipe_or_adapter,
-            **cache_context_kwargs,
-        )
-    elif isinstance(pipe_or_adapter, DiffusionPipeline):
+    if isinstance(pipe_or_adapter, (DiffusionPipeline, BlockAdapter)):
         return CachedAdapter.apply(
-            pipe=pipe_or_adapter,
-            block_adapter=None,
+            pipe_or_adapter,
             **cache_context_kwargs,
         )
     else:
@@ -149,6 +148,19 @@ def enable_cache(
         )
 
 
+def disable_cache(
+    pipe_or_adapter: Union[
+        DiffusionPipeline,
+        BlockAdapter,
+    ],
+):
+    CachedAdapter.maybe_release_hooks(pipe_or_adapter)
+    logger.warning(
+        f"Cache Acceleration is disabled for: "
+        f"{pipe_or_adapter.__class__.__name__}."
+    )
+
+
 def supported_pipelines(
     **kwargs,
 ) -> Tuple[int, List[str]]:

cache_dit/cache_factory/cache_types.py

@@ -22,11 +22,11 @@ def cache_type(type_hint: "CacheType | str") -> "CacheType":
     if isinstance(type_hint, CacheType):
         return type_hint
 
-    elif type_hint.lower() in (
-        "dual_block_cache",
-        "db_cache",
-        "dbcache",
-        "db",
+    elif type_hint.upper() in (
+        "DUAL_BLOCK_CACHE",
+        "DB_CACHE",
+        "DBCACHE",
+        "DB",
     ):
         return CacheType.DBCache
     return CacheType.NONE

cache_dit/cache_factory/patch_functors/functor_chroma.py

@@ -30,7 +30,7 @@ class ChromaPatchFunctor(PatchFunctor):
         blocks: torch.nn.ModuleList = None,
         **kwargs,
     ) -> ChromaTransformer2DModel:
-        if getattr(transformer, "_is_patched", False):
+        if hasattr(transformer, "_is_patched"):
             return transformer
 
         if blocks is None:
@@ -56,7 +56,8 @@ class ChromaPatchFunctor(PatchFunctor):
             transformer.forward = __patch_transformer_forward__.__get__(
                 transformer
             )
-            transformer._is_patched = True
+
+        transformer._is_patched = is_patched  # True or False
 
         cls_name = transformer.__class__.__name__
         logger.info(

cache_dit/cache_factory/patch_functors/functor_flux.py

@@ -31,7 +31,7 @@ class FluxPatchFunctor(PatchFunctor):
         **kwargs,
     ) -> FluxTransformer2DModel:
 
-        if getattr(transformer, "_is_patched", False):
+        if hasattr(transformer, "_is_patched"):
             return transformer
 
         if blocks is None:
@@ -57,7 +57,8 @@ class FluxPatchFunctor(PatchFunctor):
             transformer.forward = __patch_transformer_forward__.__get__(
                 transformer
             )
-            transformer._is_patched = True
+
+        transformer._is_patched = is_patched  # True or False
 
         cls_name = transformer.__class__.__name__
         logger.info(

cache_dit/utils.py

@@ -5,7 +5,8 @@ import numpy as np
 from pprint import pprint
 from diffusers import DiffusionPipeline
 
-from typing import Dict, Any
+from typing import Dict, Any, List, Union
+from cache_dit.cache_factory import BlockAdapter
 from cache_dit.logger import init_logger
 
 
@@ -29,9 +30,171 @@ class CacheStats:
 
 
 def summary(
-    pipe_or_module: DiffusionPipeline | torch.nn.Module | Any,
+    adapter_or_others: Union[
+        BlockAdapter,
+        DiffusionPipeline,
+        torch.nn.Module,
+    ],
     details: bool = False,
     logging: bool = True,
+    **kwargs,
+) -> List[CacheStats]:
+    if adapter_or_others is None:
+        return [CacheStats()]
+
+    if not isinstance(adapter_or_others, BlockAdapter):
+        if not isinstance(adapter_or_others, DiffusionPipeline):
+            transformer = adapter_or_others
+            transformer_2 = None
+        else:
+            transformer = adapter_or_others.transformer
+            transformer_2 = None
+            if hasattr(adapter_or_others, "transformer_2"):
+                transformer_2 = adapter_or_others.transformer_2
+
+        if not BlockAdapter.is_cached(transformer):
+            return [CacheStats()]
+
+        blocks_stats: List[CacheStats] = []
+        for blocks in BlockAdapter.find_blocks(transformer):
+            blocks_stats.append(
+                _summary(
+                    blocks,
+                    details=details,
+                    logging=logging,
+                    **kwargs,
+                )
+            )
+
+        if transformer_2 is not None:
+            for blocks in BlockAdapter.find_blocks(transformer_2):
+                blocks_stats.append(
+                    _summary(
+                        blocks,
+                        details=details,
+                        logging=logging,
+                        **kwargs,
+                    )
+                )
+
+        blocks_stats.append(
+            _summary(
+                transformer,
+                details=details,
+                logging=logging,
+                **kwargs,
+            )
+        )
+        if transformer_2 is not None:
+            blocks_stats.append(
+                _summary(
+                    transformer_2,
+                    details=details,
+                    logging=logging,
+                    **kwargs,
+                )
+            )
+
+        blocks_stats = [stats for stats in blocks_stats if stats.cache_options]
+
+        return blocks_stats if len(blocks_stats) else [CacheStats()]
+
+    adapter = adapter_or_others
+    if not BlockAdapter.check_block_adapter(adapter):
+        return [CacheStats()]
+
+    blocks_stats = []
+    flatten_blocks = BlockAdapter.flatten(adapter.blocks)
+    for blocks in flatten_blocks:
+        blocks_stats.append(
+            _summary(
+                blocks,
+                details=details,
+                logging=logging,
+                **kwargs,
+            )
+        )
+
+    blocks_stats = [stats for stats in blocks_stats if stats.cache_options]
+
+    return blocks_stats if len(blocks_stats) else [CacheStats()]
+
+
+def strify(
+    adapter_or_others: Union[
+        BlockAdapter,
+        DiffusionPipeline,
+        CacheStats,
+        List[CacheStats],
+        Dict[str, Any],
+    ],
+) -> str:
+    if isinstance(adapter_or_others, BlockAdapter):
+        stats = summary(adapter_or_others, logging=False)[-1]
+        cache_options = stats.cache_options
+        cached_steps = len(stats.cached_steps)
+    elif isinstance(adapter_or_others, DiffusionPipeline):
+        stats = summary(adapter_or_others, logging=False)[-1]
+        cache_options = stats.cache_options
+        cached_steps = len(stats.cached_steps)
+    elif isinstance(adapter_or_others, CacheStats):
+        stats = adapter_or_others
+        cache_options = stats.cache_options
+        cached_steps = len(stats.cached_steps)
+    elif isinstance(adapter_or_others, list):
+        stats = adapter_or_others[0]
+        cache_options = stats.cache_options
+        cached_steps = len(stats.cached_steps)
+    elif isinstance(adapter_or_others, dict):
+        from cache_dit.cache_factory import CacheType
+
+        # Assume cache_context_kwargs
+        cache_options = adapter_or_others
+        cached_steps = None
+        cache_type = cache_options.get("cache_type", CacheType.NONE)
+
+        if cache_type == CacheType.NONE:
+            return "NONE"
+    else:
+        raise ValueError(
+            "Please set pipe_or_stats param as one of: "
+            "DiffusionPipeline | CacheStats | Dict[str, Any]"
+        )
+
+    if not cache_options:
+        return "NONE"
+
+    def get_taylorseer_order():
+        taylorseer_order = 0
+        if "taylorseer_order" in cache_options:
+            taylorseer_order = cache_options["taylorseer_order"]
+        return taylorseer_order
+
+    cache_type_str = (
+        f"DBCACHE_F{cache_options.get('Fn_compute_blocks', 1)}"
+        f"B{cache_options.get('Bn_compute_blocks', 0)}_"
+        f"W{cache_options.get('max_warmup_steps', 0)}"
+        f"M{max(0, cache_options.get('max_cached_steps', -1))}"
+        f"MC{max(0, cache_options.get('max_continuous_cached_steps', -1))}_"
+        f"T{int(cache_options.get('enable_taylorseer', False))}"
+        f"O{get_taylorseer_order()}_"
+        f"R{cache_options.get('residual_diff_threshold', 0.08)}"
+    )
+
+    if cached_steps:
+        cache_type_str += f"_S{cached_steps}"
+
+    return cache_type_str
+
+
+def _summary(
+    pipe_or_module: Union[
+        DiffusionPipeline,
+        torch.nn.Module,
+    ],
+    details: bool = False,
+    logging: bool = True,
+    **kwargs,
 ) -> CacheStats:
     cache_stats = CacheStats()
 
@@ -51,6 +214,9 @@ def summary(
         cache_stats.cache_options = cache_options
         if logging:
             print(f"\n🤗Cache Options: {cls_name}\n\n{cache_options}")
+    else:
+        if logging:
+            logger.warning(f"Can't find Cache Options for: {cls_name}")
 
     if hasattr(module, "_cached_steps"):
         cached_steps: list[int] = module._cached_steps
@@ -141,56 +307,3 @@ def summary(
                 )
 
     return cache_stats
-
-
-def strify(
-    pipe_or_stats: DiffusionPipeline | CacheStats | Dict[str, Any],
-) -> str:
-    if isinstance(pipe_or_stats, DiffusionPipeline):
-        stats = summary(pipe_or_stats, logging=False)
-        cache_options = stats.cache_options
-        cached_steps = len(stats.cached_steps)
-    elif isinstance(pipe_or_stats, CacheStats):
-        stats = pipe_or_stats
-        cache_options = stats.cache_options
-        cached_steps = len(stats.cached_steps)
-    elif isinstance(pipe_or_stats, dict):
-        from cache_dit.cache_factory import CacheType
-
-        # Assume cache_context_kwargs
-        cache_options = pipe_or_stats
-        cached_steps = None
-        cache_type = cache_options.get("cache_type", CacheType.NONE)
-
-        if cache_type == CacheType.NONE:
-            return "NONE"
-    else:
-        raise ValueError(
-            "Please set pipe_or_stats param as one of: "
-            "DiffusionPipeline | CacheStats | Dict[str, Any]"
-        )
-
-    if not cache_options:
-        return "NONE"
-
-    def get_taylorseer_order():
-        taylorseer_order = 0
-        if "taylorseer_order" in cache_options:
-            taylorseer_order = cache_options["taylorseer_order"]
-        return taylorseer_order
-
-    cache_type_str = (
-        f"DBCACHE_F{cache_options.get('Fn_compute_blocks', 1)}"
-        f"B{cache_options.get('Bn_compute_blocks', 0)}_"
-        f"W{cache_options.get('max_warmup_steps', 0)}"
-        f"M{max(0, cache_options.get('max_cached_steps', -1))}"
-        f"MC{max(0, cache_options.get('max_continuous_cached_steps', -1))}_"
-        f"T{int(cache_options.get('enable_taylorseer', False))}"
-        f"O{get_taylorseer_order()}_"
-        f"R{cache_options.get('residual_diff_threshold', 0.08)}"
-    )
-
-    if cached_steps:
-        cache_type_str += f"_S{cached_steps}"
-
-    return cache_type_str

{cache_dit-0.2.28.dist-info → cache_dit-0.2.30.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cache_dit
-Version: 0.2.28
+Version: 0.2.30
 Summary: 🤗 A Unified and Training-free Cache Acceleration Toolbox for Diffusion Transformers
 Author: DefTruth, vipshop.com, etc.
 Maintainer: DefTruth, vipshop.com, etc
@@ -43,7 +43,7 @@ Dynamic: requires-python
 <div align="center">
   <img src=https://github.com/vipshop/cache-dit/raw/main/assets/cache-dit-logo.png height="120">
 
-  <p align="center">
+<p align="center">
     A <b>Unified</b> and Training-free <b>Cache Acceleration</b> Toolbox for <b>Diffusion Transformers</b> <br>
     ♥️ <b>Cache Acceleration</b> with <b>One-line</b> Code ~ ♥️
   </p>
@@ -59,26 +59,36 @@ Dynamic: requires-python
   🔥<b><a href="#unified">Unified Cache APIs</a> | <a href="#dbcache">DBCache</a> | <a href="#taylorseer">Hybrid TaylorSeer</a> | <a href="#cfg">Hybrid Cache CFG</a></b>🔥
   </p>
   <p align="center">
-  🎉Now, <b>cache-dit</b> covers <b>100%</b> Diffusers' <b>DiT-based</b> Pipelines🎉<br>
+  🎉Now, <b>cache-dit</b> covers <b>mainstream</b> Diffusers' <b>DiT-based</b> Pipelines🎉<br>
   🔥<b><a href="#supported">Qwen-Image</a> | <a href="#supported">FLUX.1</a> | <a href="#supported">Wan 2.1/2.2</a> | <a href="#supported"> ... </a> | <a href="#supported">CogVideoX</a></b>🔥
   </p>
 </div>
+<div align='center'>
+  <img src=./assets/gifs/wan2.2.C0_Q0_NONE.gif width=160px>
+  <img src=./assets/gifs/wan2.2.C1_Q0_DBCACHE_F1B0_W2M8MC2_T1O2_R0.08.gif width=160px>
+  <img src=./assets/gifs/wan2.2.C1_Q1_fp8_w8a8_dq_DBCACHE_F1B0_W2M8MC2_T1O2_R0.08.gif width=160px>
+  <p><b>🔥Wan2.2 MoE</b> Baseline | <b><a href="https://github.com/vipshop/cache-dit">+cache-dit</a>:~2.0x↑🎉</b> | +FP8 DQ:<b>~2.4x↑🎉</b></p>
+  <img src=./assets/qwen-image.C0_Q0_NONE.png width=160px>
+  <img src=./assets/qwen-image.C1_Q0_DBCACHE_F8B0_W8M0MC0_T1O4_R0.12_S23.png width=160px>
+  <img src=./assets/qwen-image.C1_Q1_fp8_w8a8_dq_DBCACHE_F8B0_W8M0MC0_T1O4_R0.12_S18.png width=160px>
+  <p><b>🔥Qwen-Image</b> Baseline | <b><a href="https://github.com/vipshop/cache-dit">+cache-dit</a>:~1.8x↑🎉</b> | +FP8 DQ:<b>~2.2x↑🎉</b><br>♥️ Please consider to leave a <b>⭐️ Star</b> to support us ~ ♥️</p>
+  </p>
+</div>
 
 ## 🔥News  
 
-- [2025-09-03] 🎉[**Wan2.2-MoE**](https://github.com/Wan-Video) **2.4x⚡️** speedup! Please refer to [run_wan_2.2.py](./examples/pipeline/run_wan_2.2.py) as an example.
-- [2025-08-19] 🔥[**Qwen-Image-Edit**](https://github.com/QwenLM/Qwen-Image) **2x⚡️** speedup! Check the example: [run_qwen_image_edit.py](./examples/pipeline/run_qwen_image_edit.py).
+- [2025-09-03] 🎉[**Wan2.2-MoE**](https://github.com/Wan-Video) **2.4x↑🎉** speedup! Please refer to [run_wan_2.2.py](./examples/pipeline/run_wan_2.2.py) as an example.
+- [2025-08-19] 🔥[**Qwen-Image-Edit**](https://github.com/QwenLM/Qwen-Image) **2x↑🎉** speedup! Check the example: [run_qwen_image_edit.py](./examples/pipeline/run_qwen_image_edit.py).
 - [2025-08-12] 🎉First caching mechanism in [QwenLM/Qwen-Image](https://github.com/QwenLM/Qwen-Image) with **[cache-dit](https://github.com/vipshop/cache-dit)**, check this [PR](https://github.com/QwenLM/Qwen-Image/pull/61). 
-- [2025-08-11] 🔥[**Qwen-Image**](https://github.com/QwenLM/Qwen-Image) **1.8x⚡️** speedup! Please refer to [run_qwen_image.py](./examples/pipeline/run_qwen_image.py) as an example.
+- [2025-08-11] 🔥[**Qwen-Image**](https://github.com/QwenLM/Qwen-Image) **1.8x↑🎉** speedup! Please refer to [run_qwen_image.py](./examples/pipeline/run_qwen_image.py) as an example.
+- [2025-07-13] 🎉[**FLUX.1-Dev**](https://github.com/xlite-dev/flux-faster) **3.3x↑🎉** speedup! NVIDIA L20 with **[cache-dit](https://github.com/vipshop/cache-dit)** + `compile + FP8 DQ`.
 
 <details>
 <summary> Previous News </summary>  
 
 - [2025-09-01] 📚[**Hybird Forward Pattern**](#unified) is supported! Please check [FLUX.1-dev](./examples/run_flux_adapter.py) as an example.
-- [2025-08-29] 🔥</b>Covers <b>100%</b> Diffusers' <b>DiT-based</b> Pipelines: **[BlockAdapter](#unified) + [Pattern Matching](#unified).**  
 - [2025-08-10] 🔥[**FLUX.1-Kontext-dev**](https://huggingface.co/black-forest-labs/FLUX.1-Kontext-dev) is supported! Please refer [run_flux_kontext.py](./examples/pipeline/run_flux_kontext.py) as an example.
 - [2025-07-18] 🎉First caching mechanism in [🤗huggingface/flux-fast](https://github.com/huggingface/flux-fast) with **[cache-dit](https://github.com/vipshop/cache-dit)**, check the [PR](https://github.com/huggingface/flux-fast/pull/13). 
-- [2025-07-13] **[🤗flux-faster](https://github.com/xlite-dev/flux-faster)** is released! **3.3x** speedup for FLUX.1 on NVIDIA L20 with **[cache-dit](https://github.com/vipshop/cache-dit)**.
 
 </details>
 
@@ -119,19 +129,8 @@ pip3 install git+https://github.com/vipshop/cache-dit.git
 
 <div id="supported"></div>
 
-```python
->>> import cache_dit
->>> cache_dit.supported_pipelines()
-(31, ['Flux*', 'Mochi*', 'CogVideoX*', 'Wan*', 'HunyuanVideo*', 'QwenImage*', 'LTXVideo*', 
-'Allegro*', 'CogView3Plus*', 'CogView4*', 'Cosmos*', 'EasyAnimate*', 'SkyReelsV2*', 'SD3*', 
-'ConsisID*', 'DiT*', 'Amused*', 'Bria*', 'HunyuanDiT*', 'HunyuanDiTPAG*', 'Lumina*', 'Lumina2*',
-'OmniGen*', 'PixArt*', 'Sana*', 'ShapE*', 'StableAudio*', 'VisualCloze*', 'AuraFlow*', 
-'Chroma*', 'HiDream*'])
-```
-
 Currently, **cache-dit** library supports almost **Any** Diffusion Transformers (with **Transformer Blocks** that match the specific Input and Output **patterns**). Please check [🎉Unified Cache APIs](#unified) for more details. Here are just some of the tested models listed:
 
-
 - [🚀Qwen-Image-Edit](https://github.com/vipshop/cache-dit/raw/main/examples)  
 - [🚀Qwen-Image](https://github.com/vipshop/cache-dit/raw/main/examples)  
 - [🚀FLUX.1-dev](https://github.com/vipshop/cache-dit/raw/main/examples)  
@@ -143,35 +142,7 @@ Currently, **cache-dit** library supports almost **Any** Diffusion Transformers
 - [🚀Wan2.1-T2V](https://github.com/vipshop/cache-dit/raw/main/examples)
 - [🚀Wan2.1-FLF2V](https://github.com/vipshop/cache-dit/raw/main/examples)
 - [🚀HunyuanVideo](https://github.com/vipshop/cache-dit/raw/main/examples)
-- [🚀HunyuanDiT](https://github.com/vipshop/cache-dit/raw/main/examples)
 
-<details>
-<summary> More Pipelines </summary>  
-
-- [🚀mochi-1-preview](https://github.com/vipshop/cache-dit/raw/main/examples)
-- [🚀LTXVideo](https://github.com/vipshop/cache-dit/raw/main/examples)
-- [🚀Allegro](https://github.com/vipshop/cache-dit/raw/main/examples)
-- [🚀CogView3Plus](https://github.com/vipshop/cache-dit/raw/main/examples)
-- [🚀CogView4](https://github.com/vipshop/cache-dit/raw/main/examples)
-- [🚀Cosmos](https://github.com/vipshop/cache-dit/raw/main/examples)
-- [🚀EasyAnimate](https://github.com/vipshop/cache-dit/raw/main/examples)
-- [🚀SkyReelsV2](https://github.com/vipshop/cache-dit/raw/main/examples)
-- [🚀SD3](https://github.com/vipshop/cache-dit/raw/main/examples)
-- [🚀ConsisID](https://github.com/vipshop/cache-dit/raw/main/examples)
-- [🚀DiT](https://github.com/vipshop/cache-dit/raw/main/examples)
-- [🚀Amused](https://github.com/vipshop/cache-dit/raw/main/examples)
-- [🚀HunyuanDiTPAG](https://github.com/vipshop/cache-dit/raw/main/examples)
-- [🚀Lumina](https://github.com/vipshop/cache-dit/raw/main/examples)
-- [🚀Lumina2](https://github.com/vipshop/cache-dit/raw/main/examples)
-- [🚀OmniGen](https://github.com/vipshop/cache-dit/raw/main/examples)
-- [🚀PixArt](https://github.com/vipshop/cache-dit/raw/main/examples)
-- [🚀Sana](https://github.com/vipshop/cache-dit/raw/main/examples)
-- [🚀StableAudio](https://github.com/vipshop/cache-dit/raw/main/examples)
-- [🚀VisualCloze](https://github.com/vipshop/cache-dit/raw/main/examples)
-- [🚀AuraFlow](https://github.com/vipshop/cache-dit/raw/main/examples)
-- [🚀Chroma](https://github.com/vipshop/cache-dit/raw/main/examples)
-- [🚀HiDream](https://github.com/vipshop/cache-dit/raw/main/examples)
-  
 </details>
 
 ## 🎉Unified Cache APIs
@@ -200,6 +171,9 @@ cache_dit.enable_cache(pipe)
 
 # Just call the pipe as normal.
 output = pipe(...)
+
+# Disable cache and run original pipe.
+cache_dit.disable_cache(pipe)
 ```
 
 ### 🔥Automatic Block Adapter
@@ -226,7 +200,6 @@ cache_dit.enable_cache(
         pipe=pipe, # Qwen-Image, etc.
         transformer=pipe.transformer,
         blocks=pipe.transformer.transformer_blocks,
-        blocks_name="transformer_blocks",
         forward_pattern=ForwardPattern.Pattern_1,
     ), 
 )
@@ -248,10 +221,6 @@ cache_dit.enable_cache(
             pipe.transformer.transformer_blocks,
             pipe.transformer.single_transformer_blocks,
         ],
-        blocks_name=[
-            "transformer_blocks",
-            "single_transformer_blocks",
-        ],
         forward_pattern=[
             ForwardPattern.Pattern_1,
             ForwardPattern.Pattern_3,
@@ -457,11 +426,21 @@ cache-dit-metrics-cli all  -i1 true_dir -i2 test_dir  # image dir
 
 How to contribute? Star ⭐️ this repo to support us or check [CONTRIBUTE.md](./CONTRIBUTE.md).
 
-## ©️License   
+<div align='center'>
+<a href="https://star-history.com/#vipshop/cache-dit&Date">
+  <picture align='center'>
+    <source media="(prefers-color-scheme: dark)" srcset="https://api.star-history.com/svg?repos=vipshop/cache-dit&type=Date&theme=dark" />
+    <source media="(prefers-color-scheme: light)" srcset="https://api.star-history.com/svg?repos=vipshop/cache-dit&type=Date" />
+    <img alt="Star History Chart" src="https://api.star-history.com/svg?repos=vipshop/cache-dit&type=Date" width=400px />
+  </picture>
+</a>
+</div>
+
+## ©️Acknowledgements
 
-<div id="license"></div>
+<div id="Acknowledgements"></div>
 
-The **cache-dit** codebase is adapted from FBCache. Special thanks to their excellent work! We have followed the original License from FBCache, please check [LICENSE](./LICENSE) for more details.
+The **cache-dit** codebase is adapted from FBCache. Over time its codebase diverged a lot, and **cache-dit** API is no longer compatible with FBCache. 
 
 ## ©️Citations