diff options
| author | Jairo Correa <jn.j41r0@gmail.com> | 2022-10-02 21:31:19 +0000 |
|---|---|---|
| committer | Jairo Correa <jn.j41r0@gmail.com> | 2022-10-02 21:31:19 +0000 |
| commit | ad0cc85d1f0bd52877963f296eb1257a0c2b012b (patch) | |
| tree | c7703e74e1964800bc1dbf11654c055f0dc21f8a /modules/scripts.py | |
| parent | ad1fbbae93fa17f797a76bc59220d074990b85b4 (diff) | |
| parent | 4c2eccf8e96825333ed400f8a8a2be78141ed8ec (diff) | |
| download | stable-diffusion-webui-gfx803-ad0cc85d1f0bd52877963f296eb1257a0c2b012b.tar.gz stable-diffusion-webui-gfx803-ad0cc85d1f0bd52877963f296eb1257a0c2b012b.tar.bz2 stable-diffusion-webui-gfx803-ad0cc85d1f0bd52877963f296eb1257a0c2b012b.zip | |
Merge branch 'master' into stable
Diffstat (limited to 'modules/scripts.py')
| -rw-r--r-- | modules/scripts.py | 34 |
1 files changed, 34 insertions, 0 deletions
diff --git a/modules/scripts.py b/modules/scripts.py index 7c3bd5e7..45230f9a 100644 --- a/modules/scripts.py +++ b/modules/scripts.py @@ -162,6 +162,40 @@ class ScriptRunner: return processed
+ def reload_sources(self):
+ for si, script in list(enumerate(self.scripts)):
+ with open(script.filename, "r", encoding="utf8") as file:
+ args_from = script.args_from
+ args_to = script.args_to
+ filename = script.filename
+ text = file.read()
+
+ from types import ModuleType
+
+ compiled = compile(text, filename, 'exec')
+ module = ModuleType(script.filename)
+ exec(compiled, module.__dict__)
+
+ for key, script_class in module.__dict__.items():
+ if type(script_class) == type and issubclass(script_class, Script):
+ self.scripts[si] = script_class()
+ self.scripts[si].filename = filename
+ self.scripts[si].args_from = args_from
+ self.scripts[si].args_to = args_to
scripts_txt2img = ScriptRunner()
scripts_img2img = ScriptRunner()
+
+def reload_script_body_only():
+ scripts_txt2img.reload_sources()
+ scripts_img2img.reload_sources()
+
+
+def reload_scripts(basedir):
+ global scripts_txt2img, scripts_img2img
+
+ scripts_data.clear()
+ load_scripts(basedir)
+
+ scripts_txt2img = ScriptRunner()
+ scripts_img2img = ScriptRunner()
|