Skip to content
Merged
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
clean up
  • Loading branch information
DN6 committed Dec 5, 2025
commit 4bd3384b05da88b9b94866279276be095e425e70
2 changes: 1 addition & 1 deletion src/diffusers/hooks/group_offloading.py
Original file line number Diff line number Diff line change
Expand Up @@ -615,7 +615,7 @@ def _apply_group_offloading_block_level(module: torch.nn.Module, config: GroupOf
done at the top-level blocks and modules specified in block_modules.

When block_modules is provided, only those modules will be treated as blocks for offloading. For each specified
module, we either offload the entire submodule or recursively apply block offloading to it.
module, recursively apply block offloading to it.
"""
if config.stream is not None and config.num_blocks_per_group != 1:
logger.warning(
Expand Down
Loading