cache-dit 0.1.6__tar.gz → 0.1.7__tar.gz

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 (77) hide show
  1. {cache_dit-0.1.6 → cache_dit-0.1.7}/PKG-INFO +3 -9
  2. {cache_dit-0.1.6 → cache_dit-0.1.7}/README.md +2 -8
  3. {cache_dit-0.1.6 → cache_dit-0.1.7}/src/cache_dit/_version.py +2 -2
  4. {cache_dit-0.1.6 → cache_dit-0.1.7}/src/cache_dit.egg-info/PKG-INFO +3 -9
  5. {cache_dit-0.1.6 → cache_dit-0.1.7}/.github/workflows/issue.yml +0 -0
  6. {cache_dit-0.1.6 → cache_dit-0.1.7}/.gitignore +0 -0
  7. {cache_dit-0.1.6 → cache_dit-0.1.7}/.pre-commit-config.yaml +0 -0
  8. {cache_dit-0.1.6 → cache_dit-0.1.7}/CONTRIBUTE.md +0 -0
  9. {cache_dit-0.1.6 → cache_dit-0.1.7}/LICENSE +0 -0
  10. {cache_dit-0.1.6 → cache_dit-0.1.7}/MANIFEST.in +0 -0
  11. {cache_dit-0.1.6 → cache_dit-0.1.7}/assets/DBCACHE_F12B12S4_R0.2_S16.png +0 -0
  12. {cache_dit-0.1.6 → cache_dit-0.1.7}/assets/DBCACHE_F12B16S4_R0.08_S6.png +0 -0
  13. {cache_dit-0.1.6 → cache_dit-0.1.7}/assets/DBCACHE_F16B16S2_R0.2_S14.png +0 -0
  14. {cache_dit-0.1.6 → cache_dit-0.1.7}/assets/DBCACHE_F16B16S4_R0.2_S13.png +0 -0
  15. {cache_dit-0.1.6 → cache_dit-0.1.7}/assets/DBCACHE_F1B0S1_R0.08_S11.png +0 -0
  16. {cache_dit-0.1.6 → cache_dit-0.1.7}/assets/DBCACHE_F1B0S1_R0.2_S19.png +0 -0
  17. {cache_dit-0.1.6 → cache_dit-0.1.7}/assets/DBCACHE_F8B0S2_R0.12_S12.png +0 -0
  18. {cache_dit-0.1.6 → cache_dit-0.1.7}/assets/DBCACHE_F8B16S1_R0.2_S18.png +0 -0
  19. {cache_dit-0.1.6 → cache_dit-0.1.7}/assets/DBCACHE_F8B8S1_R0.08_S9.png +0 -0
  20. {cache_dit-0.1.6 → cache_dit-0.1.7}/assets/DBCACHE_F8B8S1_R0.12_S12.png +0 -0
  21. {cache_dit-0.1.6 → cache_dit-0.1.7}/assets/DBCACHE_F8B8S1_R0.15_S15.png +0 -0
  22. {cache_dit-0.1.6 → cache_dit-0.1.7}/assets/DBCache.png +0 -0
  23. {cache_dit-0.1.6 → cache_dit-0.1.7}/assets/DBPRUNE_F1B0_R0.03_P24.0_T19.43s.png +0 -0
  24. {cache_dit-0.1.6 → cache_dit-0.1.7}/assets/DBPRUNE_F1B0_R0.04_P34.6_T16.82s.png +0 -0
  25. {cache_dit-0.1.6 → cache_dit-0.1.7}/assets/DBPRUNE_F1B0_R0.05_P38.3_T15.95s.png +0 -0
  26. {cache_dit-0.1.6 → cache_dit-0.1.7}/assets/DBPRUNE_F1B0_R0.06_P45.2_T14.24s.png +0 -0
  27. {cache_dit-0.1.6 → cache_dit-0.1.7}/assets/DBPRUNE_F1B0_R0.07_P52.3_T12.53s.png +0 -0
  28. {cache_dit-0.1.6 → cache_dit-0.1.7}/assets/DBPRUNE_F1B0_R0.08_P52.4_T12.52s.png +0 -0
  29. {cache_dit-0.1.6 → cache_dit-0.1.7}/assets/DBPRUNE_F1B0_R0.09_P59.2_T10.81s.png +0 -0
  30. {cache_dit-0.1.6 → cache_dit-0.1.7}/assets/DBPRUNE_F1B0_R0.12_P59.5_T10.76s.png +0 -0
  31. {cache_dit-0.1.6 → cache_dit-0.1.7}/assets/DBPRUNE_F1B0_R0.12_P63.0_T9.90s.png +0 -0
  32. {cache_dit-0.1.6 → cache_dit-0.1.7}/assets/DBPRUNE_F1B0_R0.1_P62.8_T9.95s.png +0 -0
  33. {cache_dit-0.1.6 → cache_dit-0.1.7}/assets/DBPRUNE_F1B0_R0.2_P59.5_T10.66s.png +0 -0
  34. {cache_dit-0.1.6 → cache_dit-0.1.7}/assets/DBPRUNE_F1B0_R0.3_P63.1_T9.79s.png +0 -0
  35. {cache_dit-0.1.6 → cache_dit-0.1.7}/assets/NONE_R0.08_S0.png +0 -0
  36. {cache_dit-0.1.6 → cache_dit-0.1.7}/assets/cache-dit.png +0 -0
  37. {cache_dit-0.1.6 → cache_dit-0.1.7}/bench/.gitignore +0 -0
  38. {cache_dit-0.1.6 → cache_dit-0.1.7}/bench/bench.py +0 -0
  39. {cache_dit-0.1.6 → cache_dit-0.1.7}/docs/.gitignore +0 -0
  40. {cache_dit-0.1.6 → cache_dit-0.1.7}/examples/.gitignore +0 -0
  41. {cache_dit-0.1.6 → cache_dit-0.1.7}/examples/run_cogvideox.py +0 -0
  42. {cache_dit-0.1.6 → cache_dit-0.1.7}/examples/run_flux.py +0 -0
  43. {cache_dit-0.1.6 → cache_dit-0.1.7}/examples/run_mochi.py +0 -0
  44. {cache_dit-0.1.6 → cache_dit-0.1.7}/pyproject.toml +0 -0
  45. {cache_dit-0.1.6 → cache_dit-0.1.7}/pytest.ini +0 -0
  46. {cache_dit-0.1.6 → cache_dit-0.1.7}/requirements.txt +0 -0
  47. {cache_dit-0.1.6 → cache_dit-0.1.7}/setup.cfg +0 -0
  48. {cache_dit-0.1.6 → cache_dit-0.1.7}/setup.py +0 -0
  49. {cache_dit-0.1.6 → cache_dit-0.1.7}/src/cache_dit/__init__.py +0 -0
  50. {cache_dit-0.1.6 → cache_dit-0.1.7}/src/cache_dit/cache_factory/__init__.py +0 -0
  51. {cache_dit-0.1.6 → cache_dit-0.1.7}/src/cache_dit/cache_factory/dual_block_cache/__init__.py +0 -0
  52. {cache_dit-0.1.6 → cache_dit-0.1.7}/src/cache_dit/cache_factory/dual_block_cache/cache_context.py +0 -0
  53. {cache_dit-0.1.6 → cache_dit-0.1.7}/src/cache_dit/cache_factory/dual_block_cache/diffusers_adapters/__init__.py +0 -0
  54. {cache_dit-0.1.6 → cache_dit-0.1.7}/src/cache_dit/cache_factory/dual_block_cache/diffusers_adapters/cogvideox.py +0 -0
  55. {cache_dit-0.1.6 → cache_dit-0.1.7}/src/cache_dit/cache_factory/dual_block_cache/diffusers_adapters/flux.py +0 -0
  56. {cache_dit-0.1.6 → cache_dit-0.1.7}/src/cache_dit/cache_factory/dual_block_cache/diffusers_adapters/mochi.py +0 -0
  57. {cache_dit-0.1.6 → cache_dit-0.1.7}/src/cache_dit/cache_factory/dynamic_block_prune/__init__.py +0 -0
  58. {cache_dit-0.1.6 → cache_dit-0.1.7}/src/cache_dit/cache_factory/dynamic_block_prune/diffusers_adapters/__init__.py +0 -0
  59. {cache_dit-0.1.6 → cache_dit-0.1.7}/src/cache_dit/cache_factory/dynamic_block_prune/diffusers_adapters/cogvideox.py +0 -0
  60. {cache_dit-0.1.6 → cache_dit-0.1.7}/src/cache_dit/cache_factory/dynamic_block_prune/diffusers_adapters/flux.py +0 -0
  61. {cache_dit-0.1.6 → cache_dit-0.1.7}/src/cache_dit/cache_factory/dynamic_block_prune/diffusers_adapters/mochi.py +0 -0
  62. {cache_dit-0.1.6 → cache_dit-0.1.7}/src/cache_dit/cache_factory/dynamic_block_prune/prune_context.py +0 -0
  63. {cache_dit-0.1.6 → cache_dit-0.1.7}/src/cache_dit/cache_factory/first_block_cache/__init__.py +0 -0
  64. {cache_dit-0.1.6 → cache_dit-0.1.7}/src/cache_dit/cache_factory/first_block_cache/cache_context.py +0 -0
  65. {cache_dit-0.1.6 → cache_dit-0.1.7}/src/cache_dit/cache_factory/first_block_cache/diffusers_adapters/__init__.py +0 -0
  66. {cache_dit-0.1.6 → cache_dit-0.1.7}/src/cache_dit/cache_factory/first_block_cache/diffusers_adapters/cogvideox.py +0 -0
  67. {cache_dit-0.1.6 → cache_dit-0.1.7}/src/cache_dit/cache_factory/first_block_cache/diffusers_adapters/flux.py +0 -0
  68. {cache_dit-0.1.6 → cache_dit-0.1.7}/src/cache_dit/cache_factory/first_block_cache/diffusers_adapters/mochi.py +0 -0
  69. {cache_dit-0.1.6 → cache_dit-0.1.7}/src/cache_dit/cache_factory/first_block_cache/diffusers_adapters/wan.py +0 -0
  70. {cache_dit-0.1.6 → cache_dit-0.1.7}/src/cache_dit/cache_factory/taylorseer.py +0 -0
  71. {cache_dit-0.1.6 → cache_dit-0.1.7}/src/cache_dit/cache_factory/utils.py +0 -0
  72. {cache_dit-0.1.6 → cache_dit-0.1.7}/src/cache_dit/logger.py +0 -0
  73. {cache_dit-0.1.6 → cache_dit-0.1.7}/src/cache_dit/primitives.py +0 -0
  74. {cache_dit-0.1.6 → cache_dit-0.1.7}/src/cache_dit.egg-info/SOURCES.txt +0 -0
  75. {cache_dit-0.1.6 → cache_dit-0.1.7}/src/cache_dit.egg-info/dependency_links.txt +0 -0
  76. {cache_dit-0.1.6 → cache_dit-0.1.7}/src/cache_dit.egg-info/requires.txt +0 -0
  77. {cache_dit-0.1.6 → cache_dit-0.1.7}/src/cache_dit.egg-info/top_level.txt +0 -0
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: cache_dit
3
- Version: 0.1.6
3
+ Version: 0.1.7
4
4
  Summary: 🤗 CacheDiT: A Training-free and Easy-to-use Cache Acceleration Toolbox for Diffusion Transformers
5
5
  Author: DefTruth, vipshop.com, etc.
6
6
  Maintainer: DefTruth, vipshop.com, etc
@@ -44,7 +44,7 @@ Dynamic: requires-python
44
44
  <img src=https://img.shields.io/badge/PyPI-pass-brightgreen.svg >
45
45
  <img src=https://static.pepy.tech/badge/cache-dit >
46
46
  <img src=https://img.shields.io/badge/Python-3.10|3.11|3.12-9cf.svg >
47
- <img src=https://img.shields.io/badge/Release-v0.1.6-brightgreen.svg >
47
+ <img src=https://img.shields.io/badge/Release-v0.1.7-brightgreen.svg >
48
48
  </div>
49
49
  <p align="center">
50
50
  DeepCache is for UNet not DiT. Most DiT cache speedups are complex and not training-free. CacheDiT <br>offers a set of training-free cache accelerators for DiT: 🔥DBCache, DBPrune, FBCache, etc🔥
@@ -402,13 +402,7 @@ However, users intending to use **CacheDiT** for DiT with **dynamic input shapes
402
402
  torch._dynamo.config.recompile_limit = 96 # default is 8
403
403
  torch._dynamo.config.accumulated_recompile_limit = 2048 # default is 256
404
404
  ```
405
- Otherwise, the recompile_limit error may be triggered, causing the module to fall back to eager mode. Here is the case of **DBPrune + torch.compile**.
406
-
407
- <div align="center">
408
- <p align="center">
409
- DBPrune + compile, Steps: 28, "A cat holding a sign that says hello world with complex background"
410
- </p>
411
- </div>
405
+ Otherwise, the recompile_limit error may be triggered, causing the module to fall back to eager mode.
412
406
 
413
407
  ## 🎉Supported Models
414
408
 
@@ -9,7 +9,7 @@
9
9
  <img src=https://img.shields.io/badge/PyPI-pass-brightgreen.svg >
10
10
  <img src=https://static.pepy.tech/badge/cache-dit >
11
11
  <img src=https://img.shields.io/badge/Python-3.10|3.11|3.12-9cf.svg >
12
- <img src=https://img.shields.io/badge/Release-v0.1.6-brightgreen.svg >
12
+ <img src=https://img.shields.io/badge/Release-v0.1.7-brightgreen.svg >
13
13
  </div>
14
14
  <p align="center">
15
15
  DeepCache is for UNet not DiT. Most DiT cache speedups are complex and not training-free. CacheDiT <br>offers a set of training-free cache accelerators for DiT: 🔥DBCache, DBPrune, FBCache, etc🔥
@@ -367,13 +367,7 @@ However, users intending to use **CacheDiT** for DiT with **dynamic input shapes
367
367
  torch._dynamo.config.recompile_limit = 96 # default is 8
368
368
  torch._dynamo.config.accumulated_recompile_limit = 2048 # default is 256
369
369
  ```
370
- Otherwise, the recompile_limit error may be triggered, causing the module to fall back to eager mode. Here is the case of **DBPrune + torch.compile**.
371
-
372
- <div align="center">
373
- <p align="center">
374
- DBPrune + compile, Steps: 28, "A cat holding a sign that says hello world with complex background"
375
- </p>
376
- </div>
370
+ Otherwise, the recompile_limit error may be triggered, causing the module to fall back to eager mode.
377
371
 
378
372
  ## 🎉Supported Models
379
373
 
@@ -17,5 +17,5 @@ __version__: str
17
17
  __version_tuple__: VERSION_TUPLE
18
18
  version_tuple: VERSION_TUPLE
19
19
 
20
- __version__ = version = '0.1.6'
21
- __version_tuple__ = version_tuple = (0, 1, 6)
20
+ __version__ = version = '0.1.7'
21
+ __version_tuple__ = version_tuple = (0, 1, 7)
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: cache_dit
3
- Version: 0.1.6
3
+ Version: 0.1.7
4
4
  Summary: 🤗 CacheDiT: A Training-free and Easy-to-use Cache Acceleration Toolbox for Diffusion Transformers
5
5
  Author: DefTruth, vipshop.com, etc.
6
6
  Maintainer: DefTruth, vipshop.com, etc
@@ -44,7 +44,7 @@ Dynamic: requires-python
44
44
  <img src=https://img.shields.io/badge/PyPI-pass-brightgreen.svg >
45
45
  <img src=https://static.pepy.tech/badge/cache-dit >
46
46
  <img src=https://img.shields.io/badge/Python-3.10|3.11|3.12-9cf.svg >
47
- <img src=https://img.shields.io/badge/Release-v0.1.6-brightgreen.svg >
47
+ <img src=https://img.shields.io/badge/Release-v0.1.7-brightgreen.svg >
48
48
  </div>
49
49
  <p align="center">
50
50
  DeepCache is for UNet not DiT. Most DiT cache speedups are complex and not training-free. CacheDiT <br>offers a set of training-free cache accelerators for DiT: 🔥DBCache, DBPrune, FBCache, etc🔥
@@ -402,13 +402,7 @@ However, users intending to use **CacheDiT** for DiT with **dynamic input shapes
402
402
  torch._dynamo.config.recompile_limit = 96 # default is 8
403
403
  torch._dynamo.config.accumulated_recompile_limit = 2048 # default is 256
404
404
  ```
405
- Otherwise, the recompile_limit error may be triggered, causing the module to fall back to eager mode. Here is the case of **DBPrune + torch.compile**.
406
-
407
- <div align="center">
408
- <p align="center">
409
- DBPrune + compile, Steps: 28, "A cat holding a sign that says hello world with complex background"
410
- </p>
411
- </div>
405
+ Otherwise, the recompile_limit error may be triggered, causing the module to fall back to eager mode.
412
406
 
413
407
  ## 🎉Supported Models
414
408
 
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes