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,