diff options
author | Aarni Koskela <akx@iki.fi> | 2023-05-19 15:35:16 +0300 |
---|---|---|
committer | Aarni Koskela <akx@iki.fi> | 2023-05-19 15:35:16 +0300 |
commit | 21ee46eea791d83b3b49cedd2306c7f0f1807250 (patch) | |
tree | e844f92191b50f36b7a2f543dc8afba170864fc8 /modules | |
parent | de3abc29ae91d002789c49c836df9c8d8b35cada (diff) |
Deduplicate default extra network registration
Diffstat (limited to 'modules')
-rw-r--r-- | modules/extra_networks.py | 5 | ||||
-rw-r--r-- | modules/ui_extra_networks.py | 9 |
2 files changed, 14 insertions, 0 deletions
diff --git a/modules/extra_networks.py b/modules/extra_networks.py index f9db41bc..94347275 100644 --- a/modules/extra_networks.py +++ b/modules/extra_networks.py @@ -14,6 +14,11 @@ 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 []
diff --git a/modules/ui_extra_networks.py b/modules/ui_extra_networks.py index 24eeef0e..19fbaae5 100644 --- a/modules/ui_extra_networks.py +++ b/modules/ui_extra_networks.py @@ -236,6 +236,15 @@ def initialize(): extra_pages.clear()
+def register_default_pages():
+ from modules.ui_extra_networks_textual_inversion import ExtraNetworksPageTextualInversion
+ from modules.ui_extra_networks_hypernets import ExtraNetworksPageHypernetworks
+ from modules.ui_extra_networks_checkpoints import ExtraNetworksPageCheckpoints
+ register_page(ExtraNetworksPageTextualInversion())
+ register_page(ExtraNetworksPageHypernetworks())
+ register_page(ExtraNetworksPageCheckpoints())
+
+
class ExtraNetworksUi:
def __init__(self):
self.pages = None
|