diff options
Diffstat (limited to 'modules/extra_networks.py')
| -rw-r--r-- | modules/extra_networks.py | 14 |
1 files changed, 14 insertions, 0 deletions
diff --git a/modules/extra_networks.py b/modules/extra_networks.py index f9db41bc..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:
|