Deduplicate default extra network registration

2023-05-19 15:35:16 +03:00
parent de3abc29ae
commit 21ee46eea7
3 changed files with 20 additions and 10 deletions
@@ -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