Skip to content

Commit 221726b

Browse files
committed
update
1 parent d1b1519 commit 221726b

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

src/diffusers/loaders/lora_base.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -659,6 +659,7 @@ def unfuse_lora(self, components: List[str] = [], **kwargs):
659659
if len(components) == 0:
660660
raise ValueError("`components` cannot be an empty list.")
661661

662+
merged_adapters = set()
662663
for fuse_component in components:
663664
if fuse_component not in self._lora_loadable_modules:
664665
raise ValueError(f"{fuse_component} is not found in {self._lora_loadable_modules=}.")
@@ -669,8 +670,9 @@ def unfuse_lora(self, components: List[str] = [], **kwargs):
669670
for module in model.modules():
670671
if isinstance(module, BaseTunerLayer):
671672
module.unmerge()
673+
merged_adapters.update(set(module.merged_adapters))
672674

673-
self.num_fused_loras = 0
675+
self.num_fused_loras = len(merged_adapters)
674676

675677
def set_adapters(
676678
self,

0 commit comments

Comments
 (0)