From 63b824376c49013880ff44c260ea426e2899511e Mon Sep 17 00:00:00 2001
From: AUTOMATIC <16777216c@gmail.com>
Date: Sat, 21 Jan 2023 18:47:54 +0300
Subject: [PATCH] add --gradio-queue option to enable gradio queue
---
modules/shared.py | 2 ++
webui.py | 3 +++
2 files changed, 5 insertions(+)
diff --git a/modules/shared.py b/modules/shared.py
index 72fb1934..52bbb807 100644
--- a/modules/shared.py
+++ b/modules/shared.py
@@ -100,6 +100,8 @@ parser.add_argument("--cors-allow-origins-regex", type=str, help="Allowed CORS o
parser.add_argument("--tls-keyfile", type=str, help="Partially enables TLS, requires --tls-certfile to fully function", default=None)
parser.add_argument("--tls-certfile", type=str, help="Partially enables TLS, requires --tls-keyfile to fully function", default=None)
parser.add_argument("--server-name", type=str, help="Sets hostname of server", default=None)
+parser.add_argument("--gradio-queue", action='store_true', help="Uses gradio queue; experimental option; breaks restart UI button")
+
script_loading.preload_extensions(extensions.extensions_dir, parser)
script_loading.preload_extensions(extensions.extensions_builtin_dir, parser)
diff --git a/webui.py b/webui.py
index 88d04840..d235da74 100644
--- a/webui.py
+++ b/webui.py
@@ -169,6 +169,9 @@ def webui():
shared.demo = modules.ui.create_ui()
+ if cmd_opts.gradio_queue:
+ shared.demo.queue(64)
+
app, local_url, share_url = shared.demo.launch(
share=cmd_opts.share,
server_name=server_name,
--
GitLab