aboutsummaryrefslogtreecommitdiffstats
path: root/modules/api/api.py
diff options
context:
space:
mode:
authorlanyeeee <1210347077@qq.com>2023-12-30 17:09:13 +0000
committerlanyeeee <1210347077@qq.com>2023-12-30 17:09:13 +0000
commitf651405427dfc6d4ef96ecba7f9c2ceb580263fd (patch)
tree4df9e78acdcd37d8ea730e67b7f573d782d4bf5d /modules/api/api.py
parent91560e98c47f8271d444556ef4ae6505dece9aba (diff)
downloadstable-diffusion-webui-gfx803-f651405427dfc6d4ef96ecba7f9c2ceb580263fd.tar.gz
stable-diffusion-webui-gfx803-f651405427dfc6d4ef96ecba7f9c2ceb580263fd.tar.bz2
stable-diffusion-webui-gfx803-f651405427dfc6d4ef96ecba7f9c2ceb580263fd.zip
remove locks, move init code to __init__
Diffstat (limited to 'modules/api/api.py')
-rw-r--r--modules/api/api.py37
1 files changed, 19 insertions, 18 deletions
diff --git a/modules/api/api.py b/modules/api/api.py
index d202cb8d..fc3921c2 100644
--- a/modules/api/api.py
+++ b/modules/api/api.py
@@ -251,8 +251,21 @@ class Api:
self.default_script_arg_txt2img = []
self.default_script_arg_img2img = []
- self.txt2img_script_arg_init_lock = Lock()
- self.img2img_script_arg_init_lock = Lock()
+ txt2img_script_runner = scripts.scripts_txt2img
+ img2img_script_runner = scripts.scripts_img2img
+
+ if not txt2img_script_runner.scripts or not img2img_script_runner.scripts:
+ ui.create_ui()
+
+ if not txt2img_script_runner.scripts:
+ txt2img_script_runner.initialize_scripts(False)
+ if not self.default_script_arg_txt2img:
+ self.default_script_arg_txt2img = self.init_default_script_args(txt2img_script_runner)
+
+ if not img2img_script_runner.scripts:
+ img2img_script_runner.initialize_scripts(True)
+ if not self.default_script_arg_img2img:
+ self.default_script_arg_img2img = self.init_default_script_args(img2img_script_runner)
@@ -418,16 +431,10 @@ class Api:
task_id = txt2imgreq.force_task_id or create_task_id("txt2img")
script_runner = scripts.scripts_txt2img
- with self.txt2img_script_arg_init_lock:
- if not script_runner.scripts:
- script_runner.initialize_scripts(False)
- ui.create_ui()
- infotext_script_args = {}
- self.apply_infotext(txt2imgreq, "txt2img", script_runner=script_runner, mentioned_script_args=infotext_script_args)
+ infotext_script_args = {}
+ self.apply_infotext(txt2imgreq, "txt2img", script_runner=script_runner, mentioned_script_args=infotext_script_args)
- if not self.default_script_arg_txt2img:
- self.default_script_arg_txt2img = self.init_default_script_args(script_runner)
selectable_scripts, selectable_script_idx = self.get_selectable_script(txt2imgreq.script_name, script_runner)
populate = txt2imgreq.copy(update={ # Override __init__ params
@@ -488,16 +495,10 @@ class Api:
mask = decode_base64_to_image(mask)
script_runner = scripts.scripts_img2img
- with self.img2img_script_arg_init_lock:
- if not script_runner.scripts:
- script_runner.initialize_scripts(True)
- ui.create_ui()
- infotext_script_args = {}
- self.apply_infotext(img2imgreq, "img2img", script_runner=script_runner, mentioned_script_args=infotext_script_args)
+ infotext_script_args = {}
+ self.apply_infotext(img2imgreq, "img2img", script_runner=script_runner, mentioned_script_args=infotext_script_args)
- if not self.default_script_arg_img2img:
- self.default_script_arg_img2img = self.init_default_script_args(script_runner)
selectable_scripts, selectable_script_idx = self.get_selectable_script(img2imgreq.script_name, script_runner)
populate = img2imgreq.copy(update={ # Override __init__ params