diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2023-07-13 09:59:31 +0000 |
|---|---|---|
| committer | AUTOMATIC1111 <16777216c@gmail.com> | 2023-07-13 09:59:31 +0000 |
| commit | 76ebb175ca996e93c063e7109c9f478a268952b6 (patch) | |
| tree | 79406506dabaa93617875ac4b0d526e8f626eb3a /extensions-builtin/Lora/lora.py | |
| parent | 594c8e7b263d9b37f4b18b56b159aeb6d1bba1b4 (diff) | |
| download | stable-diffusion-webui-gfx803-76ebb175ca996e93c063e7109c9f478a268952b6.tar.gz stable-diffusion-webui-gfx803-76ebb175ca996e93c063e7109c9f478a268952b6.tar.bz2 stable-diffusion-webui-gfx803-76ebb175ca996e93c063e7109c9f478a268952b6.zip | |
lora support
Diffstat (limited to 'extensions-builtin/Lora/lora.py')
| -rw-r--r-- | extensions-builtin/Lora/lora.py | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/extensions-builtin/Lora/lora.py b/extensions-builtin/Lora/lora.py index cd46e6c7..03f1ef85 100644 --- a/extensions-builtin/Lora/lora.py +++ b/extensions-builtin/Lora/lora.py @@ -179,6 +179,11 @@ def load_lora(name, lora_on_disk): if m:
sd_module = shared.sd_model.lora_layer_mapping.get(m.group(1), None)
+ # SDXL loras seem to already have correct compvis keys, so only need to replace "lora_unet" with "diffusion_model"
+ if sd_module is None and "lora_unet" in key_diffusers_without_lora_parts:
+ key = key_diffusers_without_lora_parts.replace("lora_unet", "diffusion_model")
+ sd_module = shared.sd_model.lora_layer_mapping.get(key, None)
+
if sd_module is None:
keys_failed_to_match[key_diffusers] = key
continue
|