cache-dit 0.2.24__py3-none-any.whl → 0.2.26__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.
- cache_dit/__init__.py +5 -0
- cache_dit/_version.py +2 -2
- cache_dit/cache_factory/__init__.py +2 -0
- cache_dit/cache_factory/cache_adapters.py +375 -26
- cache_dit/cache_factory/cache_blocks/__init__.py +20 -0
- cache_dit/cache_factory/cache_blocks/pattern_0_1_2.py +16 -0
- cache_dit/cache_factory/cache_blocks/pattern_3_4_5.py +270 -0
- cache_dit/cache_factory/{cache_blocks.py → cache_blocks/pattern_base.py} +17 -18
- cache_dit/cache_factory/cache_blocks/utils.py +19 -0
- cache_dit/cache_factory/cache_context.py +32 -25
- cache_dit/cache_factory/cache_interface.py +8 -3
- cache_dit/cache_factory/forward_pattern.py +45 -24
- cache_dit/cache_factory/patch_functors/__init__.py +5 -0
- cache_dit/cache_factory/patch_functors/functor_base.py +18 -0
- cache_dit/cache_factory/patch_functors/functor_chroma.py +273 -0
- cache_dit/cache_factory/{patch/flux.py → patch_functors/functor_flux.py} +45 -31
- cache_dit/compile/utils.py +1 -1
- cache_dit/quantize/__init__.py +1 -0
- cache_dit/quantize/quantize_ao.py +196 -0
- cache_dit/quantize/quantize_interface.py +46 -0
- cache_dit/utils.py +49 -17
- {cache_dit-0.2.24.dist-info → cache_dit-0.2.26.dist-info}/METADATA +43 -18
- cache_dit-0.2.26.dist-info/RECORD +42 -0
- cache_dit-0.2.24.dist-info/RECORD +0 -32
- /cache_dit/{cache_factory/patch/__init__.py → quantize/quantize_svdq.py} +0 -0
- {cache_dit-0.2.24.dist-info → cache_dit-0.2.26.dist-info}/WHEEL +0 -0
- {cache_dit-0.2.24.dist-info → cache_dit-0.2.26.dist-info}/entry_points.txt +0 -0
- {cache_dit-0.2.24.dist-info → cache_dit-0.2.26.dist-info}/licenses/LICENSE +0 -0
- {cache_dit-0.2.24.dist-info → cache_dit-0.2.26.dist-info}/top_level.txt +0 -0
cache_dit/utils.py
CHANGED
|
@@ -5,7 +5,10 @@ import numpy as np
|
|
|
5
5
|
from pprint import pprint
|
|
6
6
|
from diffusers import DiffusionPipeline
|
|
7
7
|
|
|
8
|
+
from typing import Dict, Any
|
|
8
9
|
from cache_dit.logger import init_logger
|
|
10
|
+
from cache_dit.cache_factory import CacheType
|
|
11
|
+
|
|
9
12
|
|
|
10
13
|
logger = init_logger(__name__)
|
|
11
14
|
|
|
@@ -27,13 +30,14 @@ class CacheStats:
|
|
|
27
30
|
|
|
28
31
|
|
|
29
32
|
def summary(
|
|
30
|
-
pipe_or_transformer: DiffusionPipeline | torch.nn.Module,
|
|
33
|
+
pipe_or_transformer: DiffusionPipeline | torch.nn.Module | Any,
|
|
31
34
|
details: bool = False,
|
|
32
35
|
logging: bool = True,
|
|
33
36
|
) -> CacheStats:
|
|
34
37
|
cache_stats = CacheStats()
|
|
35
38
|
cls_name = pipe_or_transformer.__class__.__name__
|
|
36
|
-
if isinstance(pipe_or_transformer,
|
|
39
|
+
if not isinstance(pipe_or_transformer, torch.nn.Module):
|
|
40
|
+
assert hasattr(pipe_or_transformer, "transformer")
|
|
37
41
|
transformer = pipe_or_transformer.transformer
|
|
38
42
|
else:
|
|
39
43
|
transformer = pipe_or_transformer
|
|
@@ -137,28 +141,56 @@ def summary(
|
|
|
137
141
|
return cache_stats
|
|
138
142
|
|
|
139
143
|
|
|
140
|
-
def strify(
|
|
141
|
-
|
|
144
|
+
def strify(
|
|
145
|
+
pipe_or_stats: DiffusionPipeline | CacheStats | Dict[str, Any],
|
|
146
|
+
) -> str:
|
|
147
|
+
if isinstance(pipe_or_stats, DiffusionPipeline):
|
|
142
148
|
stats = summary(pipe_or_stats, logging=False)
|
|
143
|
-
|
|
149
|
+
cache_options = stats.cache_options
|
|
150
|
+
cached_steps = len(stats.cached_steps)
|
|
151
|
+
elif isinstance(pipe_or_stats, CacheStats):
|
|
144
152
|
stats = pipe_or_stats
|
|
145
|
-
|
|
146
|
-
|
|
147
|
-
|
|
153
|
+
cache_options = stats.cache_options
|
|
154
|
+
cached_steps = len(stats.cached_steps)
|
|
155
|
+
elif isinstance(pipe_or_stats, dict):
|
|
156
|
+
# Assume cache_context_kwargs
|
|
157
|
+
cache_options = pipe_or_stats
|
|
158
|
+
cached_steps = None
|
|
159
|
+
else:
|
|
160
|
+
raise ValueError(
|
|
161
|
+
"Please set pipe_or_stats param as one of: "
|
|
162
|
+
"DiffusionPipeline | CacheStats | Dict[str, Any]"
|
|
163
|
+
)
|
|
148
164
|
|
|
149
165
|
if not cache_options:
|
|
150
166
|
return "NONE"
|
|
151
167
|
|
|
168
|
+
if cache_options.get("cache_type", None) != CacheType.DBCache:
|
|
169
|
+
return "NONE"
|
|
170
|
+
|
|
171
|
+
def get_taylorseer_order():
|
|
172
|
+
taylorseer_order = 0
|
|
173
|
+
if "taylorseer_kwargs" in cache_options:
|
|
174
|
+
if "n_derivatives" in cache_options["taylorseer_kwargs"]:
|
|
175
|
+
taylorseer_order = cache_options["taylorseer_kwargs"][
|
|
176
|
+
"n_derivatives"
|
|
177
|
+
]
|
|
178
|
+
elif "taylorseer_order" in cache_options:
|
|
179
|
+
taylorseer_order = cache_options["taylorseer_order"]
|
|
180
|
+
return taylorseer_order
|
|
181
|
+
|
|
152
182
|
cache_type_str = (
|
|
153
|
-
f"DBCACHE_F{cache_options
|
|
154
|
-
f"B{cache_options
|
|
155
|
-
f"W{cache_options
|
|
156
|
-
f"M{max(0, cache_options
|
|
157
|
-
f"MC{max(0, cache_options
|
|
158
|
-
f"T{int(cache_options
|
|
159
|
-
f"O{
|
|
160
|
-
f"R{cache_options
|
|
161
|
-
f"S{cached_steps}" # skiped steps
|
|
183
|
+
f"DBCACHE_F{cache_options.get('Fn_compute_blocks', 1)}"
|
|
184
|
+
f"B{cache_options.get('Bn_compute_blocks', 0)}_"
|
|
185
|
+
f"W{cache_options.get('max_warmup_steps', 0)}"
|
|
186
|
+
f"M{max(0, cache_options.get('max_cached_steps', -1))}"
|
|
187
|
+
f"MC{max(0, cache_options.get('max_continuous_cached_steps', -1))}_"
|
|
188
|
+
f"T{int(cache_options.get('enable_taylorseer', False))}"
|
|
189
|
+
f"O{get_taylorseer_order()}_"
|
|
190
|
+
f"R{cache_options.get('residual_diff_threshold', 0.08)}"
|
|
162
191
|
)
|
|
163
192
|
|
|
193
|
+
if cached_steps:
|
|
194
|
+
cache_type_str += f"_S{cached_steps}"
|
|
195
|
+
|
|
164
196
|
return cache_type_str
|
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
Metadata-Version: 2.4
|
|
2
2
|
Name: cache_dit
|
|
3
|
-
Version: 0.2.
|
|
3
|
+
Version: 0.2.26
|
|
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
|
|
@@ -11,12 +11,13 @@ Description-Content-Type: text/markdown
|
|
|
11
11
|
License-File: LICENSE
|
|
12
12
|
Requires-Dist: packaging
|
|
13
13
|
Requires-Dist: pyyaml
|
|
14
|
-
Requires-Dist: torch>=2.
|
|
15
|
-
Requires-Dist: transformers>=4.
|
|
16
|
-
Requires-Dist: diffusers>=0.
|
|
14
|
+
Requires-Dist: torch>=2.7.1
|
|
15
|
+
Requires-Dist: transformers>=4.55.2
|
|
16
|
+
Requires-Dist: diffusers>=0.35.1
|
|
17
17
|
Requires-Dist: scikit-image
|
|
18
18
|
Requires-Dist: scipy
|
|
19
19
|
Requires-Dist: lpips==0.1.4
|
|
20
|
+
Requires-Dist: torchao>=0.12.0
|
|
20
21
|
Provides-Extra: all
|
|
21
22
|
Provides-Extra: dev
|
|
22
23
|
Requires-Dist: pre-commit; extra == "dev"
|
|
@@ -58,16 +59,17 @@ Dynamic: requires-python
|
|
|
58
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>🔥
|
|
59
60
|
</p>
|
|
60
61
|
<p align="center">
|
|
61
|
-
🎉Now, <b>cache-dit</b> covers <b>
|
|
62
|
+
🎉Now, <b>cache-dit</b> covers <b>All</b> mainstream <b>DiT-based</b> Diffusers' Pipelines</b>🎉<br>
|
|
62
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>🔥
|
|
63
64
|
</p>
|
|
64
65
|
</div>
|
|
65
66
|
|
|
66
67
|
## 🔥News
|
|
67
68
|
|
|
68
|
-
- [2025-08-
|
|
69
|
-
- [2025-08-
|
|
70
|
-
- [2025-08-
|
|
69
|
+
- [2025-08-29] 🔥</b>Covers <b>All</b> Diffusers' <b>DiT-based</b> Pipelines via **[BlockAdapter](#unified) + [Pattern Matching](#unified).**
|
|
70
|
+
- [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).
|
|
71
|
+
- [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).
|
|
72
|
+
- [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).
|
|
71
73
|
- [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.
|
|
72
74
|
|
|
73
75
|
<details>
|
|
@@ -78,7 +80,7 @@ Dynamic: requires-python
|
|
|
78
80
|
- [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)**.
|
|
79
81
|
|
|
80
82
|
</details>
|
|
81
|
-
|
|
83
|
+
|
|
82
84
|
## 📖Contents
|
|
83
85
|
|
|
84
86
|
<div id="contents"></div>
|
|
@@ -111,8 +113,19 @@ pip3 install git+https://github.com/vipshop/cache-dit.git
|
|
|
111
113
|
|
|
112
114
|
<div id="supported"></div>
|
|
113
115
|
|
|
116
|
+
```python
|
|
117
|
+
>>> import cache_dit
|
|
118
|
+
>>> cache_dit.supported_pipelines()
|
|
119
|
+
(31, ['Flux*', 'Mochi*', 'CogVideoX*', 'Wan*', 'HunyuanVideo*', 'QwenImage*', 'LTXVideo*',
|
|
120
|
+
'Allegro*', 'CogView3Plus*', 'CogView4*', 'Cosmos*', 'EasyAnimate*', 'SkyReelsV2*', 'SD3*',
|
|
121
|
+
'ConsisID*', 'DiT*', 'Amused*', 'Bria*', 'HunyuanDiT*', 'HunyuanDiTPAG*', 'Lumina*', 'Lumina2*',
|
|
122
|
+
'OmniGen*', 'PixArt*', 'Sana*', 'ShapE*', 'StableAudio*', 'VisualCloze*', 'AuraFlow*',
|
|
123
|
+
'Chroma*', 'HiDream*'])
|
|
124
|
+
```
|
|
125
|
+
|
|
114
126
|
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:
|
|
115
127
|
|
|
128
|
+
|
|
116
129
|
- [🚀Qwen-Image-Edit](https://github.com/vipshop/cache-dit/raw/main/examples)
|
|
117
130
|
- [🚀Qwen-Image](https://github.com/vipshop/cache-dit/raw/main/examples)
|
|
118
131
|
- [🚀FLUX.1-dev](https://github.com/vipshop/cache-dit/raw/main/examples)
|
|
@@ -124,6 +137,7 @@ Currently, **cache-dit** library supports almost **Any** Diffusion Transformers
|
|
|
124
137
|
- [🚀Wan2.1-T2V](https://github.com/vipshop/cache-dit/raw/main/examples)
|
|
125
138
|
- [🚀Wan2.1-FLF2V](https://github.com/vipshop/cache-dit/raw/main/examples)
|
|
126
139
|
- [🚀HunyuanVideo](https://github.com/vipshop/cache-dit/raw/main/examples)
|
|
140
|
+
- [🚀HunyuanDiT](https://github.com/vipshop/cache-dit/raw/main/examples)
|
|
127
141
|
|
|
128
142
|
<details>
|
|
129
143
|
<summary> More Pipelines </summary>
|
|
@@ -137,6 +151,20 @@ Currently, **cache-dit** library supports almost **Any** Diffusion Transformers
|
|
|
137
151
|
- [🚀EasyAnimate](https://github.com/vipshop/cache-dit/raw/main/examples)
|
|
138
152
|
- [🚀SkyReelsV2](https://github.com/vipshop/cache-dit/raw/main/examples)
|
|
139
153
|
- [🚀SD3](https://github.com/vipshop/cache-dit/raw/main/examples)
|
|
154
|
+
- [🚀ConsisID](https://github.com/vipshop/cache-dit/raw/main/examples)
|
|
155
|
+
- [🚀DiT](https://github.com/vipshop/cache-dit/raw/main/examples)
|
|
156
|
+
- [🚀Amused](https://github.com/vipshop/cache-dit/raw/main/examples)
|
|
157
|
+
- [🚀HunyuanDiTPAG](https://github.com/vipshop/cache-dit/raw/main/examples)
|
|
158
|
+
- [🚀Lumina](https://github.com/vipshop/cache-dit/raw/main/examples)
|
|
159
|
+
- [🚀Lumina2](https://github.com/vipshop/cache-dit/raw/main/examples)
|
|
160
|
+
- [🚀OmniGen](https://github.com/vipshop/cache-dit/raw/main/examples)
|
|
161
|
+
- [🚀PixArt](https://github.com/vipshop/cache-dit/raw/main/examples)
|
|
162
|
+
- [🚀Sana](https://github.com/vipshop/cache-dit/raw/main/examples)
|
|
163
|
+
- [🚀StableAudio](https://github.com/vipshop/cache-dit/raw/main/examples)
|
|
164
|
+
- [🚀VisualCloze](https://github.com/vipshop/cache-dit/raw/main/examples)
|
|
165
|
+
- [🚀AuraFlow](https://github.com/vipshop/cache-dit/raw/main/examples)
|
|
166
|
+
- [🚀Chroma](https://github.com/vipshop/cache-dit/raw/main/examples)
|
|
167
|
+
- [🚀HiDream](https://github.com/vipshop/cache-dit/raw/main/examples)
|
|
140
168
|
|
|
141
169
|
</details>
|
|
142
170
|
|
|
@@ -294,14 +322,11 @@ cache_dit.enable_cache(
|
|
|
294
322
|
enable_encoder_taylorseer=True,
|
|
295
323
|
# Taylorseer cache type cache be hidden_states or residual.
|
|
296
324
|
taylorseer_cache_type="residual",
|
|
297
|
-
# Higher values of
|
|
298
|
-
#
|
|
299
|
-
|
|
300
|
-
"n_derivatives": 2, # default is 2.
|
|
301
|
-
},
|
|
302
|
-
max_warmup_steps=3, # prefer: >= n_derivatives + 1
|
|
325
|
+
# Higher values of order will lead to longer computation time
|
|
326
|
+
taylorseer_order=2, # default is 2.
|
|
327
|
+
max_warmup_steps=3, # prefer: >= order + 1
|
|
303
328
|
residual_diff_threshold=0.12
|
|
304
|
-
)
|
|
329
|
+
)s
|
|
305
330
|
```
|
|
306
331
|
|
|
307
332
|
> [!Important]
|
|
@@ -332,8 +357,8 @@ cache_dit.enable_cache(
|
|
|
332
357
|
# For model that fused CFG and non-CFG into single forward step,
|
|
333
358
|
# should set do_separate_cfg as False. For example, set it as True
|
|
334
359
|
# for Wan 2.1/Qwen-Image and set it as False for FLUX.1, HunyuanVideo,
|
|
335
|
-
# CogVideoX, Mochi, etc.
|
|
336
|
-
do_separate_cfg=True, # Wan 2.1, Qwen-Image
|
|
360
|
+
# CogVideoX, Mochi, LTXVideo, Allegro, CogView3Plus, EasyAnimate, SD3, etc.
|
|
361
|
+
do_separate_cfg=True, # Wan 2.1, Qwen-Image, CogView4, Cosmos, SkyReelsV2, etc.
|
|
337
362
|
# Compute cfg forward first or not, default False, namely,
|
|
338
363
|
# 0, 2, 4, ..., -> non-CFG step; 1, 3, 5, ... -> CFG step.
|
|
339
364
|
cfg_compute_first=False,
|
|
@@ -0,0 +1,42 @@
|
|
|
1
|
+
cache_dit/__init__.py,sha256=6_DrKjVU0N7BpQLz4A5-qLLmpc4plflHbBAss_4FmC8,1140
|
|
2
|
+
cache_dit/_version.py,sha256=qaUIn8np9pb6UE7Q3omOIYbBHqbmKse_sogKIKw72sA,706
|
|
3
|
+
cache_dit/logger.py,sha256=0zsu42hN-3-rgGC_C29ms1IvVpV4_b4_SwJCKSenxBE,4304
|
|
4
|
+
cache_dit/utils.py,sha256=H4YqlkvenlBxh2-ilOflbVDFqhI1UtnFniDgQac-D6k,7303
|
|
5
|
+
cache_dit/cache_factory/.gitignore,sha256=5Cb-qT9wsTUoMJ7vACDF7ZcLpAXhi5v-xdcWSRit988,23
|
|
6
|
+
cache_dit/cache_factory/__init__.py,sha256=sHGxoYnUWy4CqWTTnrqF2JdleYGdtN7T3erz__zztzE,635
|
|
7
|
+
cache_dit/cache_factory/cache_adapters.py,sha256=NAM7Zo7WrdSb7WOUG2WFRkFXmSx9sXm2oTMx4PhtlZk,39302
|
|
8
|
+
cache_dit/cache_factory/cache_context.py,sha256=krIPLYExwRbwZBj4-eVLV-v5QSEQoVqoLFMZBFWIIT0,41874
|
|
9
|
+
cache_dit/cache_factory/cache_interface.py,sha256=1lCoN1Co1J6lqRI3mDikgqbscWnZShmEw53uatTHJdc,8588
|
|
10
|
+
cache_dit/cache_factory/cache_types.py,sha256=FIFa6ZBfvvSMMHyBBhvarvgg2Y2wbRgITcG_uGylGe0,991
|
|
11
|
+
cache_dit/cache_factory/forward_pattern.py,sha256=FumlCuZ-TSmSYH0hGBHctSJ-oGLCftdZjLygqhsmdR4,2258
|
|
12
|
+
cache_dit/cache_factory/taylorseer.py,sha256=etSUIZzDvqW3ScKCbccTPcFaSmxV1T-xAXdk-p3e3wk,3802
|
|
13
|
+
cache_dit/cache_factory/utils.py,sha256=XkVM9AXcB9zYq8-S8QKAsGz80r3tA6U3lBNGDGeHOe4,1871
|
|
14
|
+
cache_dit/cache_factory/cache_blocks/__init__.py,sha256=jxO8v6o-Ke30HGfnDfZNZ6XknP3sabA2tlHiBW2BZTo,815
|
|
15
|
+
cache_dit/cache_factory/cache_blocks/pattern_0_1_2.py,sha256=dSVcjHPkjlAqLaXxCyvcx8jdFFq6UfIhZk0geziQCVE,434
|
|
16
|
+
cache_dit/cache_factory/cache_blocks/pattern_3_4_5.py,sha256=IMfm3HPzCbQC_SHbn74pfri4zwmPKBgpnT_NhdLxRZs,9598
|
|
17
|
+
cache_dit/cache_factory/cache_blocks/pattern_base.py,sha256=_B1PPICBkyLUoHYWo0XmtIDv944bGBWFSN9y3X9uKTM,18739
|
|
18
|
+
cache_dit/cache_factory/cache_blocks/utils.py,sha256=3VeCcqsYhD719uakrKJNSIFUa0-Qqgw08uu0LCKFa_A,648
|
|
19
|
+
cache_dit/cache_factory/patch_functors/__init__.py,sha256=yK05iONMGILsTZ83ynrUUJtiJKJ_FDjxmVIzRLy416s,252
|
|
20
|
+
cache_dit/cache_factory/patch_functors/functor_base.py,sha256=Ahk0fTfrHgNdEl-9JSkACvfyyv9G-Ei5OSz7XBIlX5o,357
|
|
21
|
+
cache_dit/cache_factory/patch_functors/functor_chroma.py,sha256=RcN6AmpDp19ILY37tjOANlwXHrcaNMHlbv9XWF8hBwA,9942
|
|
22
|
+
cache_dit/cache_factory/patch_functors/functor_flux.py,sha256=ycAypjJ34Uh7hmvbRbHadswRQj_fpxU24YfX1vtBL6c,9450
|
|
23
|
+
cache_dit/compile/__init__.py,sha256=FcTVzCeyypl-mxlc59_ehHL3lBNiDAFsXuRoJ-5Cfi0,56
|
|
24
|
+
cache_dit/compile/utils.py,sha256=nN2OIrSdwRR5zGxJinKDqb07pXpvTNTF3g_OgLkeeBU,3858
|
|
25
|
+
cache_dit/custom_ops/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
26
|
+
cache_dit/custom_ops/triton_taylorseer.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
27
|
+
cache_dit/metrics/__init__.py,sha256=RaUhl5dieF40RqnizGzR30qoJJ9dyMUEADwgwMaMQrE,575
|
|
28
|
+
cache_dit/metrics/config.py,sha256=ieOgD9ayz722RjVzk24bSIqS2D6o7TZjGk8KeXV-OLQ,551
|
|
29
|
+
cache_dit/metrics/fid.py,sha256=9Ivtazl6mW0Bon2VXa-Ia5Xj2ewxRD3V1Qkd69zYM3Y,17066
|
|
30
|
+
cache_dit/metrics/inception.py,sha256=pBVe2X6ylLPIXTG4-GWDM9DWnCviMJbJ45R3ulhktR0,12759
|
|
31
|
+
cache_dit/metrics/lpips.py,sha256=I2qCNi6qJh5TRsaIsdxO0WoRX1DN7U_H3zS0oCSahYM,1032
|
|
32
|
+
cache_dit/metrics/metrics.py,sha256=8jvM1sF-nDxUuwCRy44QEoo4dYVLCQVh1QyAMs4eaQY,27840
|
|
33
|
+
cache_dit/quantize/__init__.py,sha256=kWYoMAyZgBXu9BJlZjTQ0dRffW9GqeeY9_iTkXrb70A,59
|
|
34
|
+
cache_dit/quantize/quantize_ao.py,sha256=x9zm7AX9JjNhh7mqMkjHDGz2rDl4PzBwwU-CP1e_AVA,6012
|
|
35
|
+
cache_dit/quantize/quantize_interface.py,sha256=2s_R7xPSKuJeFpEGeLwRxnq_CqJcBG3a3lzyW5wh-UM,1241
|
|
36
|
+
cache_dit/quantize/quantize_svdq.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
37
|
+
cache_dit-0.2.26.dist-info/licenses/LICENSE,sha256=Dqb07Ik2dV41s9nIdMUbiRWEfDqo7-dQeRiY7kPO8PE,3769
|
|
38
|
+
cache_dit-0.2.26.dist-info/METADATA,sha256=xBljGjnEV9OxSWLRmKx-FLWKUw2DGIv3pNJHg106uOo,21722
|
|
39
|
+
cache_dit-0.2.26.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
|
|
40
|
+
cache_dit-0.2.26.dist-info/entry_points.txt,sha256=FX2gysXaZx6NeK1iCLMcIdP8Q4_qikkIHtEmi3oWn8o,65
|
|
41
|
+
cache_dit-0.2.26.dist-info/top_level.txt,sha256=ZJDydonLEhujzz0FOkVbO-BqfzO9d_VqRHmZU-3MOZo,10
|
|
42
|
+
cache_dit-0.2.26.dist-info/RECORD,,
|
|
@@ -1,32 +0,0 @@
|
|
|
1
|
-
cache_dit/__init__.py,sha256=KwhX9NfYkWSvDFuuUVeVjcuiZiGS_22y386l8j4afMo,905
|
|
2
|
-
cache_dit/_version.py,sha256=AZPr2DJJAwMsYN7GLT_kjMvP33B8Rgy4O_7h4o_T_88,706
|
|
3
|
-
cache_dit/logger.py,sha256=0zsu42hN-3-rgGC_C29ms1IvVpV4_b4_SwJCKSenxBE,4304
|
|
4
|
-
cache_dit/utils.py,sha256=kzwF98nzfzIFHSLtCx7Vq4a9aTW42lY-Bth7Oi4jAhg,6083
|
|
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=Cexr1_uwEkX7v8gB7DSyhCX0SI2dqS_e_ccTR16G2es,41738
|
|
10
|
-
cache_dit/cache_factory/cache_interface.py,sha256=ri8wAxmHOsDW8c6qYP6VquOJQaTSXuOchWXG3PdcYQM,8434
|
|
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=53KPsMWHyGmHGtw0T4oP0VY4O60cVXOFwFGWTlZrUqI,3857
|
|
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-0.2.24.dist-info/licenses/LICENSE,sha256=Dqb07Ik2dV41s9nIdMUbiRWEfDqo7-dQeRiY7kPO8PE,3769
|
|
28
|
-
cache_dit-0.2.24.dist-info/METADATA,sha256=zq_bGjQ_X--m1njAbOob--MwOpTDlUlAzZ3u_MiNiFM,19977
|
|
29
|
-
cache_dit-0.2.24.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
|
|
30
|
-
cache_dit-0.2.24.dist-info/entry_points.txt,sha256=FX2gysXaZx6NeK1iCLMcIdP8Q4_qikkIHtEmi3oWn8o,65
|
|
31
|
-
cache_dit-0.2.24.dist-info/top_level.txt,sha256=ZJDydonLEhujzz0FOkVbO-BqfzO9d_VqRHmZU-3MOZo,10
|
|
32
|
-
cache_dit-0.2.24.dist-info/RECORD,,
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|