Fixed bad key naming on lora fuse I just pushed

NBSTpeterhill · Aug 18, 2024 · 13a965a · 13a965a
1 parent 77ee709
commit 13a965a
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/toolkit/stable_diffusion_model.py b/toolkit/stable_diffusion_model.py
@@ -545,11 +545,9 @@ def load_model(self):
                     double_block_key = "transformer.transformer_blocks."
                     for key, value in lora_state_dict.items():
                         if single_block_key in key:
-                            new_key = key.replace(single_block_key, "")
-                            single_transformer_lora[new_key] = value
+                            single_transformer_lora[key] = value
                         elif double_block_key in key:
-                            new_key = key.replace(double_block_key, "")
-                            double_transformer_lora[new_key] = value
+                            double_transformer_lora[key] = value
                         else:
                             raise ValueError(f"Unknown lora key: {key}. Cannot load this lora in low vram mode")