Skip to content

Fix group-offloading bug#13211

Merged
sayakpaul merged 4 commits intohuggingface:mainfrom
SHYuanBest:main
Mar 4, 2026
Merged

Fix group-offloading bug#13211
sayakpaul merged 4 commits intohuggingface:mainfrom
SHYuanBest:main

Conversation

@SHYuanBest
Copy link
Contributor

Related to #13208 and #13210

cc @yiyixuxu
cc @sayakpaul
cc @dg845

Add fallback synchronous onload for conditionally-executed modules.
@SHYuanBest SHYuanBest changed the title Fix group-offload bug Fix group-offloading bug Mar 4, 2026
@sayakpaul sayakpaul requested a review from DN6 March 4, 2026 13:33
@sayakpaul
Copy link
Member

Could we also add a test for this code path in https://github.com/huggingface/diffusers/blob/main/tests/hooks/test_group_offloading.py?

@SHYuanBest
Copy link
Contributor Author

Could we also add a test for this code path in https://github.com/huggingface/diffusers/blob/main/tests/hooks/test_group_offloading.py?

done.

Copy link
Member

@sayakpaul sayakpaul left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks! A few small comments. I also ran the tests on my end and they are passing.

@sayakpaul sayakpaul merged commit 06ccde9 into huggingface:main Mar 4, 2026
10 of 11 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants