Merge branch 'master' into master

1 files changed, 5 insertions, 2 deletions

diff --git a/webui.py b/webui.py
index e1565a8d..0d0b8364 100644
--- a/webui.py
+++ b/webui.py
@@ -12,10 +12,9 @@ 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

-from modules.paths import script_path

 

 import torch

 

@@ -120,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())

@@ -228,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)

@@ -255,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())