diff --git a/src/diffusers/hooks/group_offloading.py b/src/diffusers/hooks/group_offloading.py index fd2e16c094f8..d88114436dc1 100644 --- a/src/diffusers/hooks/group_offloading.py +++ b/src/diffusers/hooks/group_offloading.py @@ -499,7 +499,10 @@ def _apply_group_offloading_block_level( the CPU memory is a bottleneck but may counteract the benefits of using streams. """ if stream is not None and num_blocks_per_group != 1: - raise ValueError(f"Using streams is only supported for num_blocks_per_group=1. Got {num_blocks_per_group=}.") + logger.warning( + f"Using streams is only supported for num_blocks_per_group=1. Got {num_blocks_per_group=}. Setting it to 1." + ) + num_blocks_per_group = 1 # Create module groups for ModuleList and Sequential blocks modules_with_group_offloading = set()