diff options
author | Leon <yongliang.zhao@happyelements.com> | 2023-09-14 10:42:56 +0000 |
---|---|---|
committer | Leon <yongliang.zhao@happyelements.com> | 2023-09-14 10:42:56 +0000 |
commit | ab3d3528a18ea1a81f1af22ea71bfc0d8c710dde (patch) | |
tree | 732d82c0fd8635dce8b8b004b282a7cb5164abc0 | |
parent | 102b6617dacffdcc89c56badcaae6c5e83c3ff21 (diff) | |
download | stable-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
-rw-r--r-- | modules/cmd_args.py | 1 | ||||
-rw-r--r-- | modules/initialize.py | 4 |
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()
|