swappy f12d161d67 Fix broken group offloading with block_level for models with standalone layers (#12692) ...

* fix: group offloading to support standalone computational layers in block-level offloading

* test: for models with standalone and deeply nested layers in block-level offloading

* feat: support for block-level offloading in group offloading config

* fix: group offload block modules to AutoencoderKL and AutoencoderKLWan

* fix: update group offloading tests to use AutoencoderKL and adjust input dimensions

* refactor: streamline block offloading logic

* Apply style fixes

* update tests

* update

* fix for failing tests

* clean up

* revert to use skip_keys

* clean up

---------

Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
Co-authored-by: Dhruv Nair <dhruv.nair@gmail.com>

2025-12-05 18:54:05 +05:30

..

__init__.py

[Refactor] Move testing utils out of src (#12238)

2025-08-28 19:53:02 +05:30

test_group_offloading.py

Fix broken group offloading with block_level for models with standalone layers (#12692)

2025-12-05 18:54:05 +05:30

test_hooks.py

[Refactor] Move testing utils out of src (#12238)

2025-08-28 19:53:02 +05:30