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

cache_dit/cache_factory/cache_interface.py

@@ -1,9 +1,11 @@
+import torch
 from typing import Any, Tuple, List
 from diffusers import DiffusionPipeline
 from cache_dit.cache_factory.cache_types import CacheType
 from cache_dit.cache_factory.block_adapters import BlockAdapter
 from cache_dit.cache_factory.block_adapters import BlockAdapterRegistry
 from cache_dit.cache_factory.cache_adapters import CachedAdapter
+from cache_dit.cache_factory.cache_contexts import CachedContextManager
 
 from cache_dit.logger import init_logger
 
@@ -12,7 +14,7 @@ logger = init_logger(__name__)
 
 def enable_cache(
     # DiffusionPipeline or BlockAdapter
-    pipe_or_adapter: DiffusionPipeline | BlockAdapter | Any,
+    pipe_or_adapter: DiffusionPipeline | BlockAdapter,
     # Cache context kwargs
     Fn_compute_blocks: int = 8,
     Bn_compute_blocks: int = 0,
@@ -30,7 +32,7 @@ def enable_cache(
     taylorseer_cache_type: str = "residual",
     taylorseer_order: int = 2,
     **other_cache_context_kwargs,
-) -> DiffusionPipeline | Any:
+) -> BlockAdapter:
     r"""
     Unified Cache API for  almost Any Diffusion Transformers (with Transformer Blocks
     that match the specific Input and Output patterns).
@@ -129,16 +131,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 +144,81 @@ def enable_cache(
         )
 
 
+def disable_cache(
+    # DiffusionPipeline or BlockAdapter
+    pipe_or_adapter: DiffusionPipeline | BlockAdapter,
+):
+    from cache_dit.cache_factory.cache_blocks.utils import (
+        remove_cached_stats,
+    )
+
+    def _disable_blocks(blocks: torch.nn.ModuleList):
+        if blocks is None:
+            return
+        if hasattr(blocks, "_forward_pattern"):
+            del blocks._forward_pattern
+        if hasattr(blocks, "_cache_context_kwargs"):
+            del blocks._cache_context_kwargs
+        remove_cached_stats(blocks)
+
+    def _disable_transformer(transformer: torch.nn.Module):
+        if transformer is None or not BlockAdapter.is_cached(transformer):
+            return
+        if original_forward := getattr(transformer, "_original_forward"):
+            transformer.forward = original_forward.__get__(transformer)
+            del transformer._original_forward
+        if hasattr(transformer, "_is_cached"):
+            del transformer._is_cached
+        if hasattr(transformer, "_forward_pattern"):
+            del transformer._forward_pattern
+        if hasattr(transformer, "_has_separate_cfg"):
+            del transformer._has_separate_cfg
+        if hasattr(transformer, "_cache_context_kwargs"):
+            del transformer._cache_context_kwargs
+        remove_cached_stats(transformer)
+        for blocks in BlockAdapter.find_blocks(transformer):
+            _disable_blocks(blocks)
+
+    def _disable_pipe(pipe: DiffusionPipeline):
+        if pipe is None or not BlockAdapter.is_cached(pipe):
+            return
+        if original_call := getattr(pipe, "_original_call"):
+            pipe.__class__.__call__ = original_call
+            del pipe.__class__._original_call
+        if cache_manager := getattr(pipe, "_cache_manager"):
+            assert isinstance(cache_manager, CachedContextManager)
+            cache_manager.clear_contexts()
+            del pipe._cache_manager
+        if hasattr(pipe, "_is_cached"):
+            del pipe.__class__._is_cached
+        if hasattr(pipe, "_cache_context_kwargs"):
+            del pipe._cache_context_kwargs
+        remove_cached_stats(pipe)
+
+    if isinstance(pipe_or_adapter, DiffusionPipeline):
+        pipe = pipe_or_adapter
+        _disable_pipe(pipe)
+        if hasattr(pipe, "transformer"):
+            _disable_transformer(pipe.transformer)
+        if hasattr(pipe, "transformer_2"):  # Wan 2.2
+            _disable_transformer(pipe.transformer_2)
+        pipe_cls_name = pipe.__class__.__name__
+        logger.warning(f"Cache Acceleration is disabled for: {pipe_cls_name}")
+    elif isinstance(pipe_or_adapter, BlockAdapter):
+        # BlockAdapter
+        adapter = pipe_or_adapter
+        BlockAdapter.assert_normalized(adapter)
+        _disable_pipe(adapter.pipe)
+        for transformer in BlockAdapter.flatten(adapter.transformer):
+            _disable_transformer(transformer)
+        for blocks in BlockAdapter.flatten(adapter.blocks):
+            _disable_blocks(blocks)
+        pipe_cls_name = adapter.pipe.__class__.__name__
+        logger.warning(f"Cache Acceleration is disabled for: {pipe_cls_name}")
+    else:
+        pass  # do nothing
+
+
 def supported_pipelines(
     **kwargs,
 ) -> Tuple[int, List[str]]:

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:

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:

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,168 @@ 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
+
+        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 +211,8 @@ def summary(
         cache_stats.cache_options = cache_options
         if logging:
             print(f"\n🤗Cache Options: {cls_name}\n\n{cache_options}")
+    else:
+        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 +303,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.29.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cache_dit
-Version: 0.2.28
+Version: 0.2.29
 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>
@@ -63,13 +63,25 @@ Dynamic: requires-python
   🔥<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>  
@@ -78,7 +90,6 @@ Dynamic: requires-python
 - [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>
 
@@ -200,6 +211,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 +240,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 +261,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 +466,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
 

{cache_dit-0.2.28.dist-info → cache_dit-0.2.29.dist-info}/RECORD

@@ -1,30 +1,30 @@
-cache_dit/__init__.py,sha256=V4jCkTic4XvWojCUqYcjlvxiNM2DjGQbOLk6R-tAx2A,1191
-cache_dit/_version.py,sha256=hCl1MKhh249NDbigjeJY-mrKYvjPFbJ7oklAepBQrto,706
+cache_dit/__init__.py,sha256=kX9V-FegZG4c8LMwI4PTmMqH794MEW0pzDArdhC0cJw,1241
+cache_dit/_version.py,sha256=4_NDrwSRsA8gshfOOEHYB4RwOrbBlY3Re7Srt7YQl4M,706
 cache_dit/logger.py,sha256=0zsu42hN-3-rgGC_C29ms1IvVpV4_b4_SwJCKSenxBE,4304
-cache_dit/utils.py,sha256=pb5298XKmaZDoHwyteYRhixAG_0DGWrvvaObeShIhOM,7146
+cache_dit/utils.py,sha256=bMeZw377_mACEj3nV1tn5DTqypBsbUVvZWJYjNQQxPg,10399
 cache_dit/cache_factory/.gitignore,sha256=5Cb-qT9wsTUoMJ7vACDF7ZcLpAXhi5v-xdcWSRit988,23
-cache_dit/cache_factory/__init__.py,sha256=M8q9furJOq2AZcLHRuCXZCjR9fNSELoEYdsCofIjMAo,1037
-cache_dit/cache_factory/cache_adapters.py,sha256=q7MxY44qw90h449Gr8W5iJjSwXPJR-YIyRmu_KJnQo0,13284
-cache_dit/cache_factory/cache_interface.py,sha256=2jcuTZ4D_P0M5pSw0z3BMPalobYen3YO1yKvRjaQjdQ,8332
+cache_dit/cache_factory/__init__.py,sha256=Iw6-iJLFbdzCsIDZXXOw371L-HPmoeZO_P9a3sDjP5s,1103
+cache_dit/cache_factory/cache_adapters.py,sha256=knNzV4BbCQmyiwsybFGl3LpTDEFtenykLb9-y_bbDpA,13905
+cache_dit/cache_factory/cache_interface.py,sha256=wy9QNNNNH4ncdyGepuvyJSJLbaTRCqqlvxtO6Os20yA,11317
 cache_dit/cache_factory/cache_types.py,sha256=FIFa6ZBfvvSMMHyBBhvarvgg2Y2wbRgITcG_uGylGe0,991
 cache_dit/cache_factory/forward_pattern.py,sha256=FumlCuZ-TSmSYH0hGBHctSJ-oGLCftdZjLygqhsmdR4,2258
 cache_dit/cache_factory/utils.py,sha256=XkVM9AXcB9zYq8-S8QKAsGz80r3tA6U3lBNGDGeHOe4,1871
-cache_dit/cache_factory/block_adapters/__init__.py,sha256=UFuHxNR7Y0RZoCl97wn0u2d_2rj8PzNsWfzgda5AoKM,17395
-cache_dit/cache_factory/block_adapters/block_adapters.py,sha256=WsqGUDSDU_5-pIXwDqAK_k4a-4jgbFGoLCoF6kAjLt8,19198
-cache_dit/cache_factory/block_adapters/block_registers.py,sha256=99ouWioxldlZJYQWhcUkOu94f8vO9V9QGzVNhKWtyO4,2005
+cache_dit/cache_factory/block_adapters/__init__.py,sha256=mtYPmsAYz4MGsMmanf6xZLaxZEkgE8gwB5mYhrM4nw4,15862
+cache_dit/cache_factory/block_adapters/block_adapters.py,sha256=sRTAfhDxdj3VAm7ejyklGwM7HMimQSEgVroVkLx7CR8,20997
+cache_dit/cache_factory/block_adapters/block_registers.py,sha256=79HpWTX7PO2ynY8I-KnF6pa-ETV4Dlpbxn5wvp_iyvw,2387
 cache_dit/cache_factory/cache_blocks/__init__.py,sha256=OWjnpJxA8EJVoRzuyb5miuiRphUFj831-bbtWsTDjnM,2750
 cache_dit/cache_factory/cache_blocks/pattern_0_1_2.py,sha256=ElMps6_7uI74tSF9GDR_dEI0bZEhdzcepM29xFWnYo8,428
 cache_dit/cache_factory/cache_blocks/pattern_3_4_5.py,sha256=CtBr6nvtAW8SAeEwPwiwWtPgrmwyb5ukb-j3IwFULJU,9953
 cache_dit/cache_factory/cache_blocks/pattern_base.py,sha256=XSDy3hsaKbAZPGZY92YgGA0qLgjQyIX8irQkb2R5T2c,20331
-cache_dit/cache_factory/cache_blocks/utils.py,sha256=wfreGvtK22hDnXuw0z0hUw-9ywu91FnExfPkP8ZzlkA,891
+cache_dit/cache_factory/cache_blocks/utils.py,sha256=dGOC1tMMOvcbvEgx44eTESKn_jsv-0RZ3tRHPa3wmQ4,1315
 cache_dit/cache_factory/cache_contexts/__init__.py,sha256=rqnJ5__zqnpVHK5A1OqWILpNh5Ss-0ZDTGgtxZMKGGo,250
 cache_dit/cache_factory/cache_contexts/cache_context.py,sha256=N88WLdd4KE9DuMWmpX8URcF55E2zWNwcKMxgVYkxMJY,13691
-cache_dit/cache_factory/cache_contexts/cache_manager.py,sha256=wSghuX93KmCxE4pFEVKuyrO0Jt5STu_x4CxypS2EdxI,34276
+cache_dit/cache_factory/cache_contexts/cache_manager.py,sha256=Tgk2-VFEhUp-oe-TFHzXay_YgbU8v90_Nx2G17ZlTds,34937
 cache_dit/cache_factory/cache_contexts/taylorseer.py,sha256=etSUIZzDvqW3ScKCbccTPcFaSmxV1T-xAXdk-p3e3wk,3802
 cache_dit/cache_factory/patch_functors/__init__.py,sha256=yK05iONMGILsTZ83ynrUUJtiJKJ_FDjxmVIzRLy416s,252
 cache_dit/cache_factory/patch_functors/functor_base.py,sha256=Ahk0fTfrHgNdEl-9JSkACvfyyv9G-Ei5OSz7XBIlX5o,357
-cache_dit/cache_factory/patch_functors/functor_chroma.py,sha256=IFCuFU8HCnohM9Qpij7oU_UG1T8Ok8uTI3B9Nw7BHfY,10029
-cache_dit/cache_factory/patch_functors/functor_flux.py,sha256=3Q8x-PEeDiUtuv-FnQ2qEzo4qbpsYOPB9uf7YridE88,9538
+cache_dit/cache_factory/patch_functors/functor_chroma.py,sha256=Z0kPAib0TkXGzJIP9FRK559UlBVuGQSZIVFir6tHzJM,10022
+cache_dit/cache_factory/patch_functors/functor_flux.py,sha256=w_QaUwv7l7ypvFxWHzjHjAafLr1PxQcgv5N7VFjr6N8,9531
 cache_dit/compile/__init__.py,sha256=FcTVzCeyypl-mxlc59_ehHL3lBNiDAFsXuRoJ-5Cfi0,56
 cache_dit/compile/utils.py,sha256=nN2OIrSdwRR5zGxJinKDqb07pXpvTNTF3g_OgLkeeBU,3858
 cache_dit/custom_ops/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -39,9 +39,9 @@ cache_dit/quantize/__init__.py,sha256=kWYoMAyZgBXu9BJlZjTQ0dRffW9GqeeY9_iTkXrb70
 cache_dit/quantize/quantize_ao.py,sha256=mGspqYgQtenl3QnKPtsSYsSD7LbVX93f1M940bhXKLU,6066
 cache_dit/quantize/quantize_interface.py,sha256=2s_R7xPSKuJeFpEGeLwRxnq_CqJcBG3a3lzyW5wh-UM,1241
 cache_dit/quantize/quantize_svdq.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-cache_dit-0.2.28.dist-info/licenses/LICENSE,sha256=Dqb07Ik2dV41s9nIdMUbiRWEfDqo7-dQeRiY7kPO8PE,3769
-cache_dit-0.2.28.dist-info/METADATA,sha256=03FPh4nIDfjSFMfkDz-sWr2g3l30UsQek8VjQ6TPn8g,23204
-cache_dit-0.2.28.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-cache_dit-0.2.28.dist-info/entry_points.txt,sha256=FX2gysXaZx6NeK1iCLMcIdP8Q4_qikkIHtEmi3oWn8o,65
-cache_dit-0.2.28.dist-info/top_level.txt,sha256=ZJDydonLEhujzz0FOkVbO-BqfzO9d_VqRHmZU-3MOZo,10
-cache_dit-0.2.28.dist-info/RECORD,,
+cache_dit-0.2.29.dist-info/licenses/LICENSE,sha256=Dqb07Ik2dV41s9nIdMUbiRWEfDqo7-dQeRiY7kPO8PE,3769
+cache_dit-0.2.29.dist-info/METADATA,sha256=wagdLaiAIX7fs5Rsw89DMsj5KQl1t3zGsTosKJe2AlQ,24540
+cache_dit-0.2.29.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+cache_dit-0.2.29.dist-info/entry_points.txt,sha256=FX2gysXaZx6NeK1iCLMcIdP8Q4_qikkIHtEmi3oWn8o,65
+cache_dit-0.2.29.dist-info/top_level.txt,sha256=ZJDydonLEhujzz0FOkVbO-BqfzO9d_VqRHmZU-3MOZo,10
+cache_dit-0.2.29.dist-info/RECORD,,