Skip to content

Commit 5e90cb4

Browse files
committed
fix auto offload: don't reset if we are adding an existing component id
1 parent ea3a2ed commit 5e90cb4

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

src/diffusers/modular_pipelines/components_manager.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -386,6 +386,7 @@ def add(self, name: str, component: Any, collection: Optional[str] = None):
386386
id(component) is Python's built-in unique identifier for the object
387387
"""
388388
component_id = f"{name}_{id(component)}"
389+
is_new_component = True
389390

390391
# check for duplicated components
391392
for comp_id, comp in self.components.items():
@@ -394,6 +395,7 @@ def add(self, name: str, component: Any, collection: Optional[str] = None):
394395
if comp_name == name:
395396
logger.warning(f"ComponentsManager: component '{name}' already exists as '{comp_id}'")
396397
component_id = comp_id
398+
is_new_component = False
397399
break
398400
else:
399401
logger.warning(
@@ -436,7 +438,7 @@ def add(self, name: str, component: Any, collection: Optional[str] = None):
436438
else:
437439
logger.info(f"ComponentsManager: added component '{name}' as '{component_id}'")
438440

439-
if self._auto_offload_enabled:
441+
if self._auto_offload_enabled and is_new_component:
440442
self.enable_auto_cpu_offload(self._auto_offload_device)
441443

442444
return component_id

0 commit comments

Comments
 (0)