Skip to content

Commit 4f081dc

Browse files
sayakpaulDN6
andauthored
Apply suggestions from code review
Co-authored-by: Dhruv Nair <[email protected]>
1 parent 99d5ad5 commit 4f081dc

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/diffusers/hooks/group_offloading.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -611,7 +611,7 @@ def _apply_group_offloading_block_level(
611611
current_modules = submodule[i : i + num_blocks_per_group]
612612
start_idx = i
613613
end_idx = i + len(current_modules) - 1
614-
group_id = f"{name}.{start_idx}_to_{end_idx}"
614+
group_id = f"{name}_{i}_{i+len(current_modules)-1}"
615615
group = ModuleGroup(
616616
modules=current_modules,
617617
offload_device=offload_device,

0 commit comments

Comments
 (0)