From 6b5ee00c71870e131dde453e58286e3c2a38cf3c Mon Sep 17 00:00:00 2001 From: Aryan Date: Sun, 27 Apr 2025 07:18:33 +0200 Subject: [PATCH] raise warning instead of error --- src/diffusers/hooks/group_offloading.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) 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()