aboutsummaryrefslogtreecommitdiffstats
path: root/modules/extra_networks.py
diff options
context:
space:
mode:
authorAUTOMATIC <16777216c@gmail.com>2023-05-21 14:37:09 +0000
committerAUTOMATIC <16777216c@gmail.com>2023-05-21 14:37:09 +0000
commit1f3182924ba8e70d0e0fc3ed270782f324376ba3 (patch)
tree27a9e5167e5b981dfe56f5084ea8e1e8743f3fc0 /modules/extra_networks.py
parent89f9faa63388756314e8a1d96cf86bf5e0663045 (diff)
parentfdaf0147b6d2a5f599464bb7c65817ef5832eff1 (diff)
downloadstable-diffusion-webui-gfx803-1f3182924ba8e70d0e0fc3ed270782f324376ba3.tar.gz
stable-diffusion-webui-gfx803-1f3182924ba8e70d0e0fc3ed270782f324376ba3.tar.bz2
stable-diffusion-webui-gfx803-1f3182924ba8e70d0e0fc3ed270782f324376ba3.zip
Merge branch 'dev' into release_candidate
Diffstat (limited to 'modules/extra_networks.py')
-rw-r--r--modules/extra_networks.py16
1 files changed, 15 insertions, 1 deletions
diff --git a/modules/extra_networks.py b/modules/extra_networks.py
index 1978673d..34a3ba63 100644
--- a/modules/extra_networks.py
+++ b/modules/extra_networks.py
@@ -14,9 +14,23 @@ def register_extra_network(extra_network):
extra_network_registry[extra_network.name] = extra_network
+def register_default_extra_networks():
+ from modules.extra_networks_hypernet import ExtraNetworkHypernet
+ register_extra_network(ExtraNetworkHypernet())
+
+
class ExtraNetworkParams:
def __init__(self, items=None):
self.items = items or []
+ self.positional = []
+ self.named = {}
+
+ for item in self.items:
+ parts = item.split('=', 2)
+ if len(parts) == 2:
+ self.named[parts[0]] = parts[1]
+ else:
+ self.positional.append(item)
class ExtraNetwork:
@@ -91,7 +105,7 @@ def deactivate(p, extra_network_data):
"""call deactivate for extra networks in extra_network_data in specified order, then call
deactivate for all remaining registered networks"""
- for extra_network_name, extra_network_args in extra_network_data.items():
+ for extra_network_name in extra_network_data:
extra_network = extra_network_registry.get(extra_network_name, None)
if extra_network is None:
continue