aboutsummaryrefslogtreecommitdiff
path: root/extensions-builtin/Lora/scripts
diff options
context:
space:
mode:
authorAUTOMATIC1111 <16777216c@gmail.com>2023-08-13 15:07:37 +0300
committerAUTOMATIC1111 <16777216c@gmail.com>2023-08-13 15:07:37 +0300
commitd8419762c1454ba51baa710d9ce8e762efc056ef (patch)
treefa02a021336d1e19f64118d1f0554ba89bc3ff39 /extensions-builtin/Lora/scripts
parentda80d649fd6a6083be02aca5695367bd25abf0d5 (diff)
Lora: output warnings in UI rather than fail for unfitting loras; switch to logging for error output in console
Diffstat (limited to 'extensions-builtin/Lora/scripts')
-rw-r--r--extensions-builtin/Lora/scripts/lora_script.py6
1 files changed, 3 insertions, 3 deletions
diff --git a/extensions-builtin/Lora/scripts/lora_script.py b/extensions-builtin/Lora/scripts/lora_script.py
index dc307f8c..4c6e774a 100644
--- a/extensions-builtin/Lora/scripts/lora_script.py
+++ b/extensions-builtin/Lora/scripts/lora_script.py
@@ -23,9 +23,9 @@ def unload():
def before_ui():
ui_extra_networks.register_page(ui_extra_networks_lora.ExtraNetworksPageLora())
- extra_network = extra_networks_lora.ExtraNetworkLora()
- extra_networks.register_extra_network(extra_network)
- extra_networks.register_extra_network_alias(extra_network, "lyco")
+ networks.extra_network_lora = extra_networks_lora.ExtraNetworkLora()
+ extra_networks.register_extra_network(networks.extra_network_lora)
+ extra_networks.register_extra_network_alias(networks.extra_network_lora, "lyco")
if not hasattr(torch.nn, 'Linear_forward_before_network'):