Tran Thanh Luan
|
6290fdfda4
|
[Feat] TaylorSeer Cache (#12648)
* init taylor_seer cache
* make compatible with any tuple size returned
* use logger for printing, add warmup feature
* still update in warmup steps
* refractor, add docs
* add configurable cache, skip compute module
* allow special cache ids only
* add stop_predicts (cooldown)
* update docs
* apply ruff
* update to handle multple calls per timestep
* refractor to use state manager
* fix format & doc
* chores: naming, remove redundancy
* add docs
* quality & style
* fix taylor precision
* Apply style fixes
* add tests
* Apply style fixes
* Remove TaylorSeerCacheTesterMixin from flux2 tests
* rename identifiers, use more expressive taylor predict loop
* torch compile compatible
* Apply style fixes
* Update src/diffusers/hooks/taylorseer_cache.py
Co-authored-by: Dhruv Nair <dhruv.nair@gmail.com>
* update docs
* make fix-copies
* fix example usage.
* remove tests on flux kontext
---------
Co-authored-by: toilaluan <toilaluan@github.com>
Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
Co-authored-by: Dhruv Nair <dhruv.nair@gmail.com>
Co-authored-by: Sayak Paul <spsayakpaul@gmail.com>
|
2025-12-06 05:39:54 +05:30 |
|
Aryan
|
0454fbb30b
|
First Block Cache (#11180)
* update
* modify flux single blocks to make compatible with cache techniques (without too much model-specific intrusion code)
* remove debug logs
* update
* cache context for different batches of data
* fix hs residual bug for single return outputs; support ltx
* fix controlnet flux
* support flux, ltx i2v, ltx condition
* update
* update
* Update docs/source/en/api/cache.md
* Update src/diffusers/hooks/hooks.py
Co-authored-by: Dhruv Nair <dhruv.nair@gmail.com>
* address review comments pt. 1
* address review comments pt. 2
* cache context refacotr; address review pt. 3
* address review comments
* metadata registration with decorators instead of centralized
* support cogvideox
* support mochi
* fix
* remove unused function
* remove central registry based on review
* update
---------
Co-authored-by: Dhruv Nair <dhruv.nair@gmail.com>
|
2025-07-09 03:27:15 +05:30 |
|
Aryan
|
a4df8dbc40
|
Update more licenses to 2025 (#11746)
update
|
2025-06-19 07:46:01 +05:30 |
|
Steven Liu
|
9f48394bf7
|
[docs] Caching methods (#11625)
* cache
* feedback
|
2025-06-02 10:58:47 -07:00 |
|
Aryan
|
844221ae4e
|
[core] FasterCache (#10163)
* init
* update
* update
* update
* make style
* update
* fix
* make it work with guidance distilled models
* update
* make fix-copies
* add tests
* update
* apply_faster_cache -> apply_fastercache
* fix
* reorder
* update
* refactor
* update docs
* add fastercache to CacheMixin
* update tests
* Apply suggestions from code review
* make style
* try to fix partial import error
* Apply style fixes
* raise warning
* update
---------
Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
|
2025-03-21 09:35:04 +05:30 |
|
Aryan
|
658e24e86c
|
[core] Pyramid Attention Broadcast (#9562)
* start pyramid attention broadcast
* add coauthor
Co-Authored-By: Xuanlei Zhao <43881818+oahzxl@users.noreply.github.com>
* update
* make style
* update
* make style
* add docs
* add tests
* update
* Update docs/source/en/api/pipelines/cogvideox.md
Co-authored-by: Steven Liu <59462357+stevhliu@users.noreply.github.com>
* Update docs/source/en/api/pipelines/cogvideox.md
Co-authored-by: Steven Liu <59462357+stevhliu@users.noreply.github.com>
* Pyramid Attention Broadcast rewrite + introduce hooks (#9826)
* rewrite implementation with hooks
* make style
* update
* merge pyramid-attention-rewrite-2
* make style
* remove changes from latte transformer
* revert docs changes
* better debug message
* add todos for future
* update tests
* make style
* cleanup
* fix
* improve log message; fix latte test
* refactor
* update
* update
* update
* revert changes to tests
* update docs
* update tests
* Apply suggestions from code review
Co-authored-by: Steven Liu <59462357+stevhliu@users.noreply.github.com>
* update
* fix flux test
* reorder
* refactor
* make fix-copies
* update docs
* fixes
* more fixes
* make style
* update tests
* update code example
* make fix-copies
* refactor based on reviews
* use maybe_free_model_hooks
* CacheMixin
* make style
* update
* add current_timestep property; update docs
* make fix-copies
* update
* improve tests
* try circular import fix
* apply suggestions from review
* address review comments
* Apply suggestions from code review
* refactor hook implementation
* add test suite for hooks
* PAB Refactor (#10667)
* update
* update
* update
---------
Co-authored-by: DN6 <dhruv.nair@gmail.com>
* update
* fix remove hook behaviour
---------
Co-authored-by: Xuanlei Zhao <43881818+oahzxl@users.noreply.github.com>
Co-authored-by: Steven Liu <59462357+stevhliu@users.noreply.github.com>
Co-authored-by: DN6 <dhruv.nair@gmail.com>
|
2025-01-28 05:09:04 +05:30 |
|