Suryansh Sijwali
c5469b7ceb
Invalidate HookRegistry child-registries cache on enable/disable cache ( #14093 )
...
* Invalidate HookRegistry child-registries cache on enable/disable cache
* Test the enable-cache-after-inference flow through a pipeline
* Make the FirstBlockCache regression test drive two explicit cache_context forwards
* Drop the pipeline-level enable-cache-after-inference test
---------
Co-authored-by: Sayak Paul <spsayakpaul@gmail.com >
2026-09-05 09:13:42 +05:30
Joseph Loftin
60092255da
Preserve hook forward signature ( #14285 )
...
* [hooks] preserve wrapped module's forward signature in HookRegistry
* [hooks] ensure exported runs
* [hooks] Remove extra
---------
Co-authored-by: Sayak Paul <spsayakpaul@gmail.com >
2026-07-29 10:14:42 +05:30
Sayak Paul
5f33428194
update licensing for others to 2026. ( #14128 )
...
Co-authored-by: dg845 <58458699+dg845@users.noreply.github.com >
2026-07-08 07:40:20 +05:30
Sayak Paul
79e408a705
[tests] migrate test_hooks.py to pytest ( #13242 )
...
* move test_hooks.py to pytest
* Create `TestHooks` generator on `torch_device` (#13871 )
Create TestHooks generator on device to avoid device mismatch errors
---------
Co-authored-by: dg845 <58458699+dg845@users.noreply.github.com >
2026-06-05 10:33:45 +05:30
Dhruv Nair
7aa6af1138
[Refactor] Move testing utils out of src ( #12238 )
...
* update
* update
* update
* update
* update
* merge main
* Revert "merge main"
This reverts commit 65efbcead5
2025-08-28 19:53:02 +05:30
Tianqi Tang
4b7fe044e3
Fix typos and inconsistencies ( #12204 )
...
Fix typos and test assertions
Co-authored-by: Sayak Paul <spsayakpaul@gmail.com >
2025-08-26 07:58:08 -07:00
Sayak Paul
62cce3045d
[chore] change to 2025 licensing for remaining ( #11741 )
...
change to 2025 licensing for remaining
2025-06-18 20:56:00 +05:30
Dhruv Nair
edc154da09
Update Ruff to latest Version ( #10919 )
...
* update
* update
* update
* update
2025-04-09 16:51:34 +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
