add checkpoints tab for extra networks UI

1 files changed, 5 insertions, 1 deletions

diff --git a/webui.py b/webui.py
index 41f32f5c..0d0b8364 100644
--- a/webui.py
+++ b/webui.py
@@ -12,7 +12,7 @@ from packaging import version
 import logging

 logging.getLogger("xformers").addFilter(lambda record: 'A matching Triton is not available' not in record.getMessage())

 

-from modules import import_hook, errors, extra_networks

+from modules import import_hook, errors, extra_networks, ui_extra_networks_checkpoints

 from modules import extra_networks_hypernet, ui_extra_networks_hypernets, ui_extra_networks_textual_inversion

 from modules.call_queue import wrap_queued_call, queue_lock, wrap_gradio_gpu_call

 

@@ -119,6 +119,7 @@ def initialize():
     ui_extra_networks.intialize()

     ui_extra_networks.register_page(ui_extra_networks_textual_inversion.ExtraNetworksPageTextualInversion())

     ui_extra_networks.register_page(ui_extra_networks_hypernets.ExtraNetworksPageHypernetworks())

+    ui_extra_networks.register_page(ui_extra_networks_checkpoints.ExtraNetworksPageCheckpoints())

 

     extra_networks.initialize()

     extra_networks.register_extra_network(extra_networks_hypernet.ExtraNetworkHypernet())

@@ -227,6 +228,8 @@ def webui():
         if launch_api:

             create_api(app)

 

+        ui_extra_networks.add_pages_to_demo(app)

+

         modules.script_callbacks.app_started_callback(shared.demo, app)

 

         wait_on_server(shared.demo)

@@ -254,6 +257,7 @@ def webui():
         ui_extra_networks.intialize()

         ui_extra_networks.register_page(ui_extra_networks_textual_inversion.ExtraNetworksPageTextualInversion())

         ui_extra_networks.register_page(ui_extra_networks_hypernets.ExtraNetworksPageHypernetworks())

+        ui_extra_networks.register_page(ui_extra_networks_checkpoints.ExtraNetworksPageCheckpoints())

 

         extra_networks.initialize()

         extra_networks.register_extra_network(extra_networks_hypernet.ExtraNetworkHypernet())