cache-dit 0.2.25__py3-none-any.whl → 0.2.27__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.


This version of cache-dit might be problematic. Click here for more details.

Files changed (32) hide show
  1. cache_dit/__init__.py +9 -4
  2. cache_dit/_version.py +2 -2
  3. cache_dit/cache_factory/__init__.py +16 -3
  4. cache_dit/cache_factory/block_adapters/__init__.py +538 -0
  5. cache_dit/cache_factory/block_adapters/block_adapters.py +333 -0
  6. cache_dit/cache_factory/block_adapters/block_registers.py +77 -0
  7. cache_dit/cache_factory/cache_adapters.py +121 -563
  8. cache_dit/cache_factory/cache_blocks/__init__.py +18 -0
  9. cache_dit/cache_factory/cache_blocks/pattern_0_1_2.py +16 -0
  10. cache_dit/cache_factory/cache_blocks/pattern_3_4_5.py +275 -0
  11. cache_dit/cache_factory/{cache_blocks.py → cache_blocks/pattern_base.py} +100 -82
  12. cache_dit/cache_factory/cache_blocks/utils.py +23 -0
  13. cache_dit/cache_factory/cache_contexts/__init__.py +2 -0
  14. cache_dit/cache_factory/{cache_context.py → cache_contexts/cache_context.py} +94 -56
  15. cache_dit/cache_factory/cache_interface.py +24 -16
  16. cache_dit/cache_factory/forward_pattern.py +45 -24
  17. cache_dit/cache_factory/patch_functors/__init__.py +5 -0
  18. cache_dit/cache_factory/patch_functors/functor_base.py +18 -0
  19. cache_dit/cache_factory/patch_functors/functor_chroma.py +276 -0
  20. cache_dit/cache_factory/{patch/flux.py → patch_functors/functor_flux.py} +49 -31
  21. cache_dit/quantize/quantize_ao.py +19 -4
  22. cache_dit/quantize/quantize_interface.py +2 -2
  23. cache_dit/utils.py +19 -15
  24. {cache_dit-0.2.25.dist-info → cache_dit-0.2.27.dist-info}/METADATA +76 -19
  25. cache_dit-0.2.27.dist-info/RECORD +47 -0
  26. cache_dit-0.2.25.dist-info/RECORD +0 -36
  27. /cache_dit/cache_factory/{patch/__init__.py → cache_contexts/cache_manager.py} +0 -0
  28. /cache_dit/cache_factory/{taylorseer.py → cache_contexts/taylorseer.py} +0 -0
  29. {cache_dit-0.2.25.dist-info → cache_dit-0.2.27.dist-info}/WHEEL +0 -0
  30. {cache_dit-0.2.25.dist-info → cache_dit-0.2.27.dist-info}/entry_points.txt +0 -0
  31. {cache_dit-0.2.25.dist-info → cache_dit-0.2.27.dist-info}/licenses/LICENSE +0 -0
  32. {cache_dit-0.2.25.dist-info → cache_dit-0.2.27.dist-info}/top_level.txt +0 -0
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: cache_dit
3
- Version: 0.2.25
3
+ Version: 0.2.27
4
4
  Summary: 🤗 CacheDiT: An Unified and Training-free Cache Acceleration Toolbox for Diffusion Transformers
5
5
  Author: DefTruth, vipshop.com, etc.
6
6
  Maintainer: DefTruth, vipshop.com, etc
@@ -59,27 +59,33 @@ Dynamic: requires-python
59
59
  🔥<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>🔥
60
60
  </p>
61
61
  <p align="center">
62
- 🎉Now, <b>cache-dit</b> covers <b>Most</b> mainstream <b>Diffusers'</b> Pipelines</b>🎉<br>
62
+ 🎉Now, <b>cache-dit</b> covers <b>All</b> mainstream <b>DiT-based</b> Diffusers' Pipelines</b>🎉<br>
63
63
  🔥<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>🔥
64
64
  </p>
65
65
  </div>
66
66
 
67
67
  ## 🔥News
68
68
 
69
- - [2025-08-26] 🎉[**Wan2.2**](https://github.com/Wan-Video) **1.8x⚡️** speedup with `cache-dit + compile`! Check the [example](./examples/run_wan_2.2.py).
70
- - [2025-08-19] 🔥[**Qwen-Image-Edit**](https://github.com/QwenLM/Qwen-Image) **2x⚡️** speedup! Check example [run_qwen_image_edit.py](./examples/run_qwen_image_edit.py).
71
- - [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 the [PR](https://github.com/QwenLM/Qwen-Image/pull/61).
69
+ - [2025-09-01] 📚[**Hybird Forward Pattern**](#unified) is supported! Please check [FLUX.1-dev](./examples/run_flux_adapter.py) as an example.
70
+ - [2025-08-29] 🔥</b>Covers <b>All</b> Diffusers' <b>DiT-based</b> Pipelines via **[BlockAdapter](#unified) + [Pattern Matching](#unified).**
71
+ - [2025-08-26] 🎉[**Wan2.2**](https://github.com/Wan-Video) **1.8x⚡️** speedup with `cache-dit + compile`! Please check the [example](./examples/run_wan_2.2.py).
72
+ - [2025-08-19] 🔥[**Qwen-Image-Edit**](https://github.com/QwenLM/Qwen-Image) **2x⚡️** speedup! Check the example at [run_qwen_image_edit.py](./examples/run_qwen_image_edit.py).
73
+ - [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).
72
74
  - [2025-08-11] 🔥[**Qwen-Image**](https://github.com/QwenLM/Qwen-Image) **1.8x⚡️** speedup! Please refer [run_qwen_image.py](./examples/run_qwen_image.py) as an example.
75
+ - [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/run_flux_kontext.py) as an example.
76
+ - [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).
77
+ - [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)**.
73
78
 
79
+ <!--
74
80
  <details>
75
81
  <summary> Previous News </summary>
76
-
77
- - [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/run_flux_kontext.py) as an example.
82
+
78
83
  - [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).
79
84
  - [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)**.
80
85
 
81
86
  </details>
82
-
87
+ -->
88
+
83
89
  ## 📖Contents
84
90
 
85
91
  <div id="contents"></div>
@@ -112,8 +118,19 @@ pip3 install git+https://github.com/vipshop/cache-dit.git
112
118
 
113
119
  <div id="supported"></div>
114
120
 
121
+ ```python
122
+ >>> import cache_dit
123
+ >>> cache_dit.supported_pipelines()
124
+ (31, ['Flux*', 'Mochi*', 'CogVideoX*', 'Wan*', 'HunyuanVideo*', 'QwenImage*', 'LTXVideo*',
125
+ 'Allegro*', 'CogView3Plus*', 'CogView4*', 'Cosmos*', 'EasyAnimate*', 'SkyReelsV2*', 'SD3*',
126
+ 'ConsisID*', 'DiT*', 'Amused*', 'Bria*', 'HunyuanDiT*', 'HunyuanDiTPAG*', 'Lumina*', 'Lumina2*',
127
+ 'OmniGen*', 'PixArt*', 'Sana*', 'ShapE*', 'StableAudio*', 'VisualCloze*', 'AuraFlow*',
128
+ 'Chroma*', 'HiDream*'])
129
+ ```
130
+
115
131
  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:
116
132
 
133
+
117
134
  - [🚀Qwen-Image-Edit](https://github.com/vipshop/cache-dit/raw/main/examples)
118
135
  - [🚀Qwen-Image](https://github.com/vipshop/cache-dit/raw/main/examples)
119
136
  - [🚀FLUX.1-dev](https://github.com/vipshop/cache-dit/raw/main/examples)
@@ -125,6 +142,7 @@ Currently, **cache-dit** library supports almost **Any** Diffusion Transformers
125
142
  - [🚀Wan2.1-T2V](https://github.com/vipshop/cache-dit/raw/main/examples)
126
143
  - [🚀Wan2.1-FLF2V](https://github.com/vipshop/cache-dit/raw/main/examples)
127
144
  - [🚀HunyuanVideo](https://github.com/vipshop/cache-dit/raw/main/examples)
145
+ - [🚀HunyuanDiT](https://github.com/vipshop/cache-dit/raw/main/examples)
128
146
 
129
147
  <details>
130
148
  <summary> More Pipelines </summary>
@@ -138,6 +156,20 @@ Currently, **cache-dit** library supports almost **Any** Diffusion Transformers
138
156
  - [🚀EasyAnimate](https://github.com/vipshop/cache-dit/raw/main/examples)
139
157
  - [🚀SkyReelsV2](https://github.com/vipshop/cache-dit/raw/main/examples)
140
158
  - [🚀SD3](https://github.com/vipshop/cache-dit/raw/main/examples)
159
+ - [🚀ConsisID](https://github.com/vipshop/cache-dit/raw/main/examples)
160
+ - [🚀DiT](https://github.com/vipshop/cache-dit/raw/main/examples)
161
+ - [🚀Amused](https://github.com/vipshop/cache-dit/raw/main/examples)
162
+ - [🚀HunyuanDiTPAG](https://github.com/vipshop/cache-dit/raw/main/examples)
163
+ - [🚀Lumina](https://github.com/vipshop/cache-dit/raw/main/examples)
164
+ - [🚀Lumina2](https://github.com/vipshop/cache-dit/raw/main/examples)
165
+ - [🚀OmniGen](https://github.com/vipshop/cache-dit/raw/main/examples)
166
+ - [🚀PixArt](https://github.com/vipshop/cache-dit/raw/main/examples)
167
+ - [🚀Sana](https://github.com/vipshop/cache-dit/raw/main/examples)
168
+ - [🚀StableAudio](https://github.com/vipshop/cache-dit/raw/main/examples)
169
+ - [🚀VisualCloze](https://github.com/vipshop/cache-dit/raw/main/examples)
170
+ - [🚀AuraFlow](https://github.com/vipshop/cache-dit/raw/main/examples)
171
+ - [🚀Chroma](https://github.com/vipshop/cache-dit/raw/main/examples)
172
+ - [🚀HiDream](https://github.com/vipshop/cache-dit/raw/main/examples)
141
173
 
142
174
  </details>
143
175
 
@@ -149,7 +181,7 @@ Currently, **cache-dit** library supports almost **Any** Diffusion Transformers
149
181
 
150
182
  Currently, for any **Diffusion** models with **Transformer Blocks** that match the specific **Input/Output patterns**, we can use the **Unified Cache APIs** from **cache-dit**, namely, the `cache_dit.enable_cache(...)` API. The **Unified Cache APIs** are currently in the experimental phase; please stay tuned for updates. The supported patterns are listed as follows:
151
183
 
152
- ![](https://github.com/vipshop/cache-dit/raw/main/assets/patterns.png)
184
+ ![](https://github.com/vipshop/cache-dit/raw/main/assets/patterns-v1.png)
153
185
 
154
186
  ### ♥️Cache Acceleration with One-line Code
155
187
 
@@ -169,19 +201,22 @@ cache_dit.enable_cache(pipe)
169
201
  output = pipe(...)
170
202
  ```
171
203
 
172
- ### 🔥Automatic Block Adapter
204
+ ### 🔥Automatic Block Adapter + 📚Hybird Forward Pattern
173
205
 
174
- But in some cases, you may have a **modified** Diffusion Pipeline or Transformer that is not located in the diffusers library or not officially supported by **cache-dit** at this time. The **BlockAdapter** can help you solve this problems. Please refer to [Qwen-Image w/ BlockAdapter](./examples/run_qwen_image_adapter.py) as an example.
206
+ But in some cases, you may have a **modified** Diffusion Pipeline or Transformer that is not located in the diffusers library or not officially supported by **cache-dit** at this time. The **BlockAdapter** can help you solve this problems. Please refer to [🔥Qwen-Image w/ BlockAdapter](./examples/run_qwen_image_adapter.py) and [📚FLUX.1 w/ Hybird Forward Pattern](./examples/run_flux_adapter.py) for more details.
175
207
 
176
208
  ```python
177
209
  from cache_dit import ForwardPattern, BlockAdapter
178
210
 
179
- # Use BlockAdapter with `auto` mode.
211
+ # Use 🔥BlockAdapter with `auto` mode.
180
212
  cache_dit.enable_cache(
181
- BlockAdapter(pipe=pipe, auto=True), # Qwen-Image, etc.
182
- # Check `📚Forward Pattern Matching` documentation and hack the code of
183
- # of Qwen-Image, you will find that it has satisfied `FORWARD_PATTERN_1`.
184
- forward_pattern=ForwardPattern.Pattern_1,
213
+ BlockAdapter(
214
+ # Any DiffusionPipeline, Qwen-Image, etc.
215
+ pipe=pipe, auto=True,
216
+ # Check `📚Forward Pattern Matching` documentation and hack the code of
217
+ # of Qwen-Image, you will find that it has satisfied `FORWARD_PATTERN_1`.
218
+ forward_pattern=ForwardPattern.Pattern_1,
219
+ ),
185
220
  )
186
221
 
187
222
  # Or, manually setup transformer configurations.
@@ -191,8 +226,30 @@ cache_dit.enable_cache(
191
226
  transformer=pipe.transformer,
192
227
  blocks=pipe.transformer.transformer_blocks,
193
228
  blocks_name="transformer_blocks",
229
+ forward_pattern=ForwardPattern.Pattern_1,
194
230
  ),
195
- forward_pattern=ForwardPattern.Pattern_1,
231
+ )
232
+
233
+ # cache-dit supported 📚Hybird Forward Pattern, for example:
234
+ # For diffusers <= 0.34.0, FLUX.1 transformer_blocks and
235
+ # single_transformer_blocks has different forward pattern.
236
+ cache_dit.enable_cache(
237
+ BlockAdapter(
238
+ pipe=pipe, # FLUX.1, etc.
239
+ transformer=pipe.transformer,
240
+ blocks=[
241
+ pipe.transformer.transformer_blocks,
242
+ pipe.transformer.single_transformer_blocks,
243
+ ],
244
+ blocks_name=[
245
+ "transformer_blocks",
246
+ "single_transformer_blocks",
247
+ ],
248
+ forward_pattern=[
249
+ ForwardPattern.Pattern_1,
250
+ ForwardPattern.Pattern_3,
251
+ ],
252
+ ),
196
253
  )
197
254
  ```
198
255
  For such situations, **BlockAdapter** can help you quickly apply various cache acceleration features to your own Diffusion Pipelines and Transformers. Please check the [📚BlockAdapter.md](./docs/BlockAdapter.md) for more details.
@@ -330,8 +387,8 @@ cache_dit.enable_cache(
330
387
  # For model that fused CFG and non-CFG into single forward step,
331
388
  # should set do_separate_cfg as False. For example, set it as True
332
389
  # for Wan 2.1/Qwen-Image and set it as False for FLUX.1, HunyuanVideo,
333
- # CogVideoX, Mochi, etc.
334
- do_separate_cfg=True, # Wan 2.1, Qwen-Image
390
+ # CogVideoX, Mochi, LTXVideo, Allegro, CogView3Plus, EasyAnimate, SD3, etc.
391
+ do_separate_cfg=True, # Wan 2.1, Qwen-Image, CogView4, Cosmos, SkyReelsV2, etc.
335
392
  # Compute cfg forward first or not, default False, namely,
336
393
  # 0, 2, 4, ..., -> non-CFG step; 1, 3, 5, ... -> CFG step.
337
394
  cfg_compute_first=False,
@@ -0,0 +1,47 @@
1
+ cache_dit/__init__.py,sha256=-ENQmZ9rQfIlyiyyL1M_i_7dmxdp7EIXk3Q0ijccweE,1140
2
+ cache_dit/_version.py,sha256=sQt3vJQXfZVL852h4ZP8iWmYkhYM4zSob72-HinVTME,706
3
+ cache_dit/logger.py,sha256=0zsu42hN-3-rgGC_C29ms1IvVpV4_b4_SwJCKSenxBE,4304
4
+ cache_dit/utils.py,sha256=d3aE6dybxu8zVTskfPvZIDXqtA8f7WZQuvhVrE9iUq8,7336
5
+ cache_dit/cache_factory/.gitignore,sha256=5Cb-qT9wsTUoMJ7vACDF7ZcLpAXhi5v-xdcWSRit988,23
6
+ cache_dit/cache_factory/__init__.py,sha256=8IqCdKEzqJGGlAcmgn4ySXmpyZE0BzUA7xBrZHpYGb4,899
7
+ cache_dit/cache_factory/cache_adapters.py,sha256=wTLgJAxAJLn7iF6ylWvfaxWT9dhQ977AF5t1cdYHFOg,9857
8
+ cache_dit/cache_factory/cache_interface.py,sha256=-PI7Oph2TDUCE6sviudV75jQ36p8WymLSSlyhHazL4k,8369
9
+ cache_dit/cache_factory/cache_types.py,sha256=FIFa6ZBfvvSMMHyBBhvarvgg2Y2wbRgITcG_uGylGe0,991
10
+ cache_dit/cache_factory/forward_pattern.py,sha256=FumlCuZ-TSmSYH0hGBHctSJ-oGLCftdZjLygqhsmdR4,2258
11
+ cache_dit/cache_factory/utils.py,sha256=XkVM9AXcB9zYq8-S8QKAsGz80r3tA6U3lBNGDGeHOe4,1871
12
+ cache_dit/cache_factory/block_adapters/__init__.py,sha256=wtbQLqTCyegAzDjBn3qVmg80Q6SH7WE8IxSYh-a6Chw,16935
13
+ cache_dit/cache_factory/block_adapters/block_adapters.py,sha256=Q1BCk96ibWOuyCI_JO_qkQ-SCoCs0J4UtSFupaZ8jeU,11560
14
+ cache_dit/cache_factory/block_adapters/block_registers.py,sha256=99ouWioxldlZJYQWhcUkOu94f8vO9V9QGzVNhKWtyO4,2005
15
+ cache_dit/cache_factory/cache_blocks/__init__.py,sha256=yqZZLNM9NN8SHv9bb1jPAaRAfb_jQ77Km8nMBJpfhpI,777
16
+ cache_dit/cache_factory/cache_blocks/pattern_0_1_2.py,sha256=ElMps6_7uI74tSF9GDR_dEI0bZEhdzcepM29xFWnYo8,428
17
+ cache_dit/cache_factory/cache_blocks/pattern_3_4_5.py,sha256=BD-bJdFleN7B2JLrNc90s9f-wQon9Ud5JUV7fBaaLmg,9902
18
+ cache_dit/cache_factory/cache_blocks/pattern_base.py,sha256=2FPQBnX4yroH4i5-qb8yPyAN523hrGc-ozGsv7dENQU,19809
19
+ cache_dit/cache_factory/cache_blocks/utils.py,sha256=DM4YEtbRVV-_NsbFWjI-WZPBbSDc4Gl366AgFz06L_s,770
20
+ cache_dit/cache_factory/cache_contexts/__init__.py,sha256=KqD3XSAnUlaa2wzuOhG0q8A3Xf0m5rsOEihCNLjsdfU,153
21
+ cache_dit/cache_factory/cache_contexts/cache_context.py,sha256=6rtLjAPX5zVrmBXYPHOGQuyIR0fuDxrqsRiWV9A1h8M,43135
22
+ cache_dit/cache_factory/cache_contexts/cache_manager.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
23
+ cache_dit/cache_factory/cache_contexts/taylorseer.py,sha256=etSUIZzDvqW3ScKCbccTPcFaSmxV1T-xAXdk-p3e3wk,3802
24
+ cache_dit/cache_factory/patch_functors/__init__.py,sha256=yK05iONMGILsTZ83ynrUUJtiJKJ_FDjxmVIzRLy416s,252
25
+ cache_dit/cache_factory/patch_functors/functor_base.py,sha256=Ahk0fTfrHgNdEl-9JSkACvfyyv9G-Ei5OSz7XBIlX5o,357
26
+ cache_dit/cache_factory/patch_functors/functor_chroma.py,sha256=IFCuFU8HCnohM9Qpij7oU_UG1T8Ok8uTI3B9Nw7BHfY,10029
27
+ cache_dit/cache_factory/patch_functors/functor_flux.py,sha256=3Q8x-PEeDiUtuv-FnQ2qEzo4qbpsYOPB9uf7YridE88,9538
28
+ cache_dit/compile/__init__.py,sha256=FcTVzCeyypl-mxlc59_ehHL3lBNiDAFsXuRoJ-5Cfi0,56
29
+ cache_dit/compile/utils.py,sha256=nN2OIrSdwRR5zGxJinKDqb07pXpvTNTF3g_OgLkeeBU,3858
30
+ cache_dit/custom_ops/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
31
+ cache_dit/custom_ops/triton_taylorseer.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
32
+ cache_dit/metrics/__init__.py,sha256=RaUhl5dieF40RqnizGzR30qoJJ9dyMUEADwgwMaMQrE,575
33
+ cache_dit/metrics/config.py,sha256=ieOgD9ayz722RjVzk24bSIqS2D6o7TZjGk8KeXV-OLQ,551
34
+ cache_dit/metrics/fid.py,sha256=9Ivtazl6mW0Bon2VXa-Ia5Xj2ewxRD3V1Qkd69zYM3Y,17066
35
+ cache_dit/metrics/inception.py,sha256=pBVe2X6ylLPIXTG4-GWDM9DWnCviMJbJ45R3ulhktR0,12759
36
+ cache_dit/metrics/lpips.py,sha256=I2qCNi6qJh5TRsaIsdxO0WoRX1DN7U_H3zS0oCSahYM,1032
37
+ cache_dit/metrics/metrics.py,sha256=8jvM1sF-nDxUuwCRy44QEoo4dYVLCQVh1QyAMs4eaQY,27840
38
+ cache_dit/quantize/__init__.py,sha256=kWYoMAyZgBXu9BJlZjTQ0dRffW9GqeeY9_iTkXrb70A,59
39
+ cache_dit/quantize/quantize_ao.py,sha256=mGspqYgQtenl3QnKPtsSYsSD7LbVX93f1M940bhXKLU,6066
40
+ cache_dit/quantize/quantize_interface.py,sha256=2s_R7xPSKuJeFpEGeLwRxnq_CqJcBG3a3lzyW5wh-UM,1241
41
+ cache_dit/quantize/quantize_svdq.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
42
+ cache_dit-0.2.27.dist-info/licenses/LICENSE,sha256=Dqb07Ik2dV41s9nIdMUbiRWEfDqo7-dQeRiY7kPO8PE,3769
43
+ cache_dit-0.2.27.dist-info/METADATA,sha256=DN_iaPFoqnieAUKirJp-A0eLdYiiABAO8TY1EyCh8ss,23150
44
+ cache_dit-0.2.27.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
45
+ cache_dit-0.2.27.dist-info/entry_points.txt,sha256=FX2gysXaZx6NeK1iCLMcIdP8Q4_qikkIHtEmi3oWn8o,65
46
+ cache_dit-0.2.27.dist-info/top_level.txt,sha256=ZJDydonLEhujzz0FOkVbO-BqfzO9d_VqRHmZU-3MOZo,10
47
+ cache_dit-0.2.27.dist-info/RECORD,,
@@ -1,36 +0,0 @@
1
- cache_dit/__init__.py,sha256=VsT0f0R0COp8v6Sx9hGNsqxiElERaDpfG11a9MfK0is,945
2
- cache_dit/_version.py,sha256=t9iixyDuMWz1nP7KM0bgrLNIpwu8JK6uZApA8DoCQwM,706
3
- cache_dit/logger.py,sha256=0zsu42hN-3-rgGC_C29ms1IvVpV4_b4_SwJCKSenxBE,4304
4
- cache_dit/utils.py,sha256=1oWDMYs6E7FRsd8cidsVOPT-meIRKeuqbGbE6CrCUec,7236
5
- cache_dit/cache_factory/.gitignore,sha256=5Cb-qT9wsTUoMJ7vACDF7ZcLpAXhi5v-xdcWSRit988,23
6
- cache_dit/cache_factory/__init__.py,sha256=evWenCin1kuBGa6W5BCKMrDZc1C1R2uVPSg0BjXgdXE,499
7
- cache_dit/cache_factory/cache_adapters.py,sha256=Yugqljm9tm615srM2BGQlR_tA0QiZo3PbLPceObh4dQ,25988
8
- cache_dit/cache_factory/cache_blocks.py,sha256=ZeazBsYvLIjI5M_OnLL2xP2W7zMeM0rxVfBBwIVHBRs,18661
9
- cache_dit/cache_factory/cache_context.py,sha256=HhA5IMSdF-i-koSB1jqf5AMC_UyDV7VinpHm4Qee9Ig,41800
10
- cache_dit/cache_factory/cache_interface.py,sha256=HymagnKEDs48Ly_x3IM5jTMNJpLrIdJnppVlkr2xHaM,8433
11
- cache_dit/cache_factory/cache_types.py,sha256=FIFa6ZBfvvSMMHyBBhvarvgg2Y2wbRgITcG_uGylGe0,991
12
- cache_dit/cache_factory/forward_pattern.py,sha256=B2YeqV2t_zo2Ar8m7qimPBjwQgoXHGp2grPZmEAhi8s,1286
13
- cache_dit/cache_factory/taylorseer.py,sha256=etSUIZzDvqW3ScKCbccTPcFaSmxV1T-xAXdk-p3e3wk,3802
14
- cache_dit/cache_factory/utils.py,sha256=XkVM9AXcB9zYq8-S8QKAsGz80r3tA6U3lBNGDGeHOe4,1871
15
- cache_dit/cache_factory/patch/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
16
- cache_dit/cache_factory/patch/flux.py,sha256=iNQ-1RlOgXupZ4uPiEvJ__Ro6vKT_fOKja9JrpMrO78,8998
17
- cache_dit/compile/__init__.py,sha256=FcTVzCeyypl-mxlc59_ehHL3lBNiDAFsXuRoJ-5Cfi0,56
18
- cache_dit/compile/utils.py,sha256=nN2OIrSdwRR5zGxJinKDqb07pXpvTNTF3g_OgLkeeBU,3858
19
- cache_dit/custom_ops/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
20
- cache_dit/custom_ops/triton_taylorseer.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
21
- cache_dit/metrics/__init__.py,sha256=RaUhl5dieF40RqnizGzR30qoJJ9dyMUEADwgwMaMQrE,575
22
- cache_dit/metrics/config.py,sha256=ieOgD9ayz722RjVzk24bSIqS2D6o7TZjGk8KeXV-OLQ,551
23
- cache_dit/metrics/fid.py,sha256=9Ivtazl6mW0Bon2VXa-Ia5Xj2ewxRD3V1Qkd69zYM3Y,17066
24
- cache_dit/metrics/inception.py,sha256=pBVe2X6ylLPIXTG4-GWDM9DWnCviMJbJ45R3ulhktR0,12759
25
- cache_dit/metrics/lpips.py,sha256=I2qCNi6qJh5TRsaIsdxO0WoRX1DN7U_H3zS0oCSahYM,1032
26
- cache_dit/metrics/metrics.py,sha256=8jvM1sF-nDxUuwCRy44QEoo4dYVLCQVh1QyAMs4eaQY,27840
27
- cache_dit/quantize/__init__.py,sha256=kWYoMAyZgBXu9BJlZjTQ0dRffW9GqeeY9_iTkXrb70A,59
28
- cache_dit/quantize/quantize_ao.py,sha256=sKz_RmVtxLOpAPnUv_iOjzY_226pfaxgB_HMNrfyqB8,5465
29
- cache_dit/quantize/quantize_interface.py,sha256=NG4WP7s8CLW6KhVFb9e1aAjW30KWTCcM2aS5n8QuwsA,1241
30
- cache_dit/quantize/quantize_svdq.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
31
- cache_dit-0.2.25.dist-info/licenses/LICENSE,sha256=Dqb07Ik2dV41s9nIdMUbiRWEfDqo7-dQeRiY7kPO8PE,3769
32
- cache_dit-0.2.25.dist-info/METADATA,sha256=a5wbENMZ9BDjHbM3Ejb7Il7x4QzD8W7Lzmu4poo95Wo,19913
33
- cache_dit-0.2.25.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
34
- cache_dit-0.2.25.dist-info/entry_points.txt,sha256=FX2gysXaZx6NeK1iCLMcIdP8Q4_qikkIHtEmi3oWn8o,65
35
- cache_dit-0.2.25.dist-info/top_level.txt,sha256=ZJDydonLEhujzz0FOkVbO-BqfzO9d_VqRHmZU-3MOZo,10
36
- cache_dit-0.2.25.dist-info/RECORD,,