From ccd9414ff1a0c319f073cae63610ec58b412dd1e Mon Sep 17 00:00:00 2001 From: Pierre Chapuis Date: Tue, 5 Mar 2024 17:27:41 +0100 Subject: [PATCH] fix debug map when attaching two LoRAs (in that case return the path of the LoraAdapter) --- src/refiners/fluxion/adapters/lora.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/src/refiners/fluxion/adapters/lora.py b/src/refiners/fluxion/adapters/lora.py index 043a973..19f6d00 100644 --- a/src/refiners/fluxion/adapters/lora.py +++ b/src/refiners/fluxion/adapters/lora.py @@ -460,6 +460,9 @@ def _auto_attach_loras( adapter, parent = attached if parent is None: # `adapter` is already attached and `lora` has been added to it + if debug_map is not None: + path = adapter.get_path() + debug_map.append((key, path)) continue if debug_map is not None: path = adapter.target.get_path(parent)