diff options
| author | v0xie <28695009+v0xie@users.noreply.github.com> | 2023-11-02 00:11:32 -0700 |
|---|---|---|
| committer | v0xie <28695009+v0xie@users.noreply.github.com> | 2023-11-02 00:11:32 -0700 |
| commit | 65ccd6305fcf72347d5ed68f03095dced865ef6e (patch) | |
| tree | 9c60b591d98a636132967a2871ed8273d1504173 /extensions-builtin | |
| parent | a2fad6ee055f3f4e98e46b6c2d912776fe608214 (diff) | |
detect diag_oft type
Diffstat (limited to 'extensions-builtin')
| -rw-r--r-- | extensions-builtin/Lora/networks.py | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/extensions-builtin/Lora/networks.py b/extensions-builtin/Lora/networks.py index 78a97033..7f814706 100644 --- a/extensions-builtin/Lora/networks.py +++ b/extensions-builtin/Lora/networks.py @@ -191,10 +191,17 @@ def load_network(name, network_on_disk): key = key_network_without_network_parts.replace("lora_te1_text_model", "transformer_text_model")
sd_module = shared.sd_model.network_layer_mapping.get(key, None)
+ # kohya_ss OFT module
elif sd_module is None and "oft_unet" in key_network_without_network_parts:
key = key_network_without_network_parts.replace("oft_unet", "diffusion_model")
sd_module = shared.sd_model.network_layer_mapping.get(key, None)
+ # KohakuBlueLeaf OFT module
+ if sd_module is None and "oft_diag" in key:
+ key = key_network_without_network_parts.replace("lora_unet", "diffusion_model")
+ key = key_network_without_network_parts.replace("lora_te1_text_model", "0_transformer_text_model")
+ sd_module = shared.sd_model.network_layer_mapping.get(key, None)
+
if sd_module is None:
keys_failed_to_match[key_network] = key
continue
|