4 Commits

Author	SHA1	Message	Date
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
Dhruv Nair	66e6a0215f	[CI] Remove unittest dependency from testing_utils.py (#12621) ... * update * Update tests/testing_utils.py Co-authored-by: Sayak Paul <spsayakpaul@gmail.com> * Update tests/testing_utils.py Co-authored-by: Sayak Paul <spsayakpaul@gmail.com> * Apply style fixes --------- Co-authored-by: Sayak Paul <spsayakpaul@gmail.com> Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>	2025-11-11 16:40:39 +05:30
Sayak Paul	55d49d4379	[ci] don't run sana layerwise casting tests in CI. (#12551) ... * don't run sana layerwise casting tests in CI. * up	2025-10-28 13:29:51 +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