aboutsummaryrefslogtreecommitdiffstats
path: root/modules
diff options
context:
space:
mode:
authorLeon <yongliang.zhao@happyelements.com>2023-09-14 10:42:56 +0000
committerLeon <yongliang.zhao@happyelements.com>2023-09-14 10:42:56 +0000
commitab3d3528a18ea1a81f1af22ea71bfc0d8c710dde (patch)
tree732d82c0fd8635dce8b8b004b282a7cb5164abc0 /modules
parent102b6617dacffdcc89c56badcaae6c5e83c3ff21 (diff)
downloadstable-diffusion-webui-gfx803-ab3d3528a18ea1a81f1af22ea71bfc0d8c710dde.tar.gz
stable-diffusion-webui-gfx803-ab3d3528a18ea1a81f1af22ea71bfc0d8c710dde.tar.bz2
stable-diffusion-webui-gfx803-ab3d3528a18ea1a81f1af22ea71bfc0d8c710dde.zip
add --skip-load-model-at-start
Diffstat (limited to 'modules')
-rw-r--r--modules/cmd_args.py1
-rw-r--r--modules/initialize.py4
2 files changed, 3 insertions, 2 deletions
diff --git a/modules/cmd_args.py b/modules/cmd_args.py
index 5be879dd..4e602a84 100644
--- a/modules/cmd_args.py
+++ b/modules/cmd_args.py
@@ -117,3 +117,4 @@ parser.add_argument('--api-server-stop', action='store_true', help='enable serve
parser.add_argument('--timeout-keep-alive', type=int, default=30, help='set timeout_keep_alive for uvicorn')
parser.add_argument("--disable-all-extensions", action='store_true', help="prevent all extensions from running regardless of any other settings", default=False)
parser.add_argument("--disable-extra-extensions", action='store_true', help="prevent all extensions except built-in from running regardless of any other settings", default=False)
+parser.add_argument("--skip-load-model-at-start", action='store_true', help="if load a model at web start, only take effect when --nowebui", )
diff --git a/modules/initialize.py b/modules/initialize.py
index f24f7637..ac95fc6f 100644
--- a/modules/initialize.py
+++ b/modules/initialize.py
@@ -151,8 +151,8 @@ def initialize_rest(*, reload_script_modules=False):
from modules import devices
devices.first_time_calculation()
-
- Thread(target=load_model).start()
+ if not shared.cmd_opts.skip_load_model_at_start:
+ Thread(target=load_model).start()
from modules import shared_items
shared_items.reload_hypernetworks()