We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 99d5ad5 commit 4f081dcCopy full SHA for 4f081dc
src/diffusers/hooks/group_offloading.py
@@ -611,7 +611,7 @@ def _apply_group_offloading_block_level(
611
current_modules = submodule[i : i + num_blocks_per_group]
612
start_idx = i
613
end_idx = i + len(current_modules) - 1
614
- group_id = f"{name}.{start_idx}_to_{end_idx}"
+ group_id = f"{name}_{i}_{i+len(current_modules)-1}"
615
group = ModuleGroup(
616
modules=current_modules,
617
offload_device=offload_device,
0 commit comments