[LoRA] Fix dedup for post-replacement module aliases (#44413)

Signed-off-by: bk-201 <joy25810@foxmail.com>
2026-06-03 18:23:23 +00:00
parent 2b91012650
commit 271328e256
1 changed files with 1 additions and 0 deletions
@@ -451,6 +451,7 @@ class LoRAModelManager:
            )
            if isinstance(new_module, BaseLayerWithLoRA):
                wrapped_by_id[id(module)] = new_module
+                wrapped_by_id[id(new_module)] = new_module

            # (yard1): TODO make this more robust
            if "lm_head" in module_name: