Skip to content

Commit 2cd3665

Browse files
Reviewer feedback: check _lora_loadable_modules
1 parent 4b01401 commit 2cd3665

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/diffusers/loaders/lora_base.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -909,6 +909,6 @@ def enable_lora_hotswap(self, **kwargs) -> None:
909909
target_rank (`int`):
910910
The highest rank among all the adapters that will be loaded.
911911
"""
912-
for component in self.components.values():
913-
if hasattr(component, "enable_lora_hotswap"):
912+
for key, component in self.components.items():
913+
if hasattr(component, "enable_lora_hotswap") and (key in self._lora_loadable_modules):
914914
component.enable_lora_hotswap(**kwargs)

0 commit comments

Comments
 (0)