Commit cc6c0fc7 authored by AUTOMATIC1111's avatar AUTOMATIC1111 Committed by GitHub

Merge pull request #10557 from akx/dedupe-webui-boot

Refactor & deduplicate web UI boot code
parents db1ce5aa 71f4a4af
...@@ -14,6 +14,11 @@ def register_extra_network(extra_network): ...@@ -14,6 +14,11 @@ def register_extra_network(extra_network):
extra_network_registry[extra_network.name] = 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: class ExtraNetworkParams:
def __init__(self, items=None): def __init__(self, items=None):
self.items = items or [] self.items = items or []
......
...@@ -271,6 +271,12 @@ def load_scripts(): ...@@ -271,6 +271,12 @@ def load_scripts():
sys.path = syspath sys.path = syspath
current_basedir = paths.script_path current_basedir = paths.script_path
global scripts_txt2img, scripts_img2img, scripts_postproc
scripts_txt2img = ScriptRunner()
scripts_img2img = ScriptRunner()
scripts_postproc = scripts_postprocessing.ScriptPostprocessingRunner()
def wrap_call(func, filename, funcname, *args, default=None, **kwargs): def wrap_call(func, filename, funcname, *args, default=None, **kwargs):
try: try:
...@@ -527,9 +533,9 @@ class ScriptRunner: ...@@ -527,9 +533,9 @@ class ScriptRunner:
self.scripts[si].args_to = args_to self.scripts[si].args_to = args_to
scripts_txt2img = ScriptRunner() scripts_txt2img: ScriptRunner = None
scripts_img2img = ScriptRunner() scripts_img2img: ScriptRunner = None
scripts_postproc = scripts_postprocessing.ScriptPostprocessingRunner() scripts_postproc: scripts_postprocessing.ScriptPostprocessingRunner = None
scripts_current: ScriptRunner = None scripts_current: ScriptRunner = None
...@@ -539,14 +545,7 @@ def reload_script_body_only(): ...@@ -539,14 +545,7 @@ def reload_script_body_only():
scripts_img2img.reload_sources(cache) scripts_img2img.reload_sources(cache)
def reload_scripts(): reload_scripts = load_scripts # compatibility alias
global scripts_txt2img, scripts_img2img, scripts_postproc
load_scripts()
scripts_txt2img = ScriptRunner()
scripts_img2img = ScriptRunner()
scripts_postproc = scripts_postprocessing.ScriptPostprocessingRunner()
def add_classes_to_gradio_component(comp): def add_classes_to_gradio_component(comp):
......
...@@ -98,7 +98,6 @@ def setup_model(): ...@@ -98,7 +98,6 @@ def setup_model():
if not os.path.exists(model_path): if not os.path.exists(model_path):
os.makedirs(model_path) os.makedirs(model_path)
list_models()
enable_midas_autodownload() enable_midas_autodownload()
......
...@@ -232,10 +232,19 @@ class ExtraNetworksPage: ...@@ -232,10 +232,19 @@ class ExtraNetworksPage:
return None return None
def intialize(): def initialize():
extra_pages.clear() 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: class ExtraNetworksUi:
def __init__(self): def __init__(self):
self.pages = None self.pages = None
......
This diff is collapsed.
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment