diff options
| author | Kohaku-Blueleaf <59680068+KohakuBlueleaf@users.noreply.github.com> | 2023-08-14 13:34:51 +0800 |
|---|---|---|
| committer | Kohaku-Blueleaf <59680068+KohakuBlueleaf@users.noreply.github.com> | 2023-08-14 13:34:51 +0800 |
| commit | e7c03ccdcefb2a80129703931ef1f8455708945b (patch) | |
| tree | 6769bdb3a32bdf99cf25422206ff03457bdfa86c /modules/scripts.py | |
| parent | d9cc27cb29926c9cc5dce331da8fbaf996cf4973 (diff) | |
| parent | 007ecfbb29771aa7cdcf0263ab1811bc75fa5446 (diff) | |
Merge branch 'dev' into extra-norm-module
Diffstat (limited to 'modules/scripts.py')
| -rw-r--r-- | modules/scripts.py | 17 |
1 files changed, 16 insertions, 1 deletions
diff --git a/modules/scripts.py b/modules/scripts.py index c6459b45..d4a9da94 100644 --- a/modules/scripts.py +++ b/modules/scripts.py @@ -106,9 +106,16 @@ class Script: pass
+ def setup(self, p, *args):
+ """For AlwaysVisible scripts, this function is called when the processing object is set up, before any processing starts.
+ args contains all values returned by components from ui().
+ """
+ pass
+
+
def before_process(self, p, *args):
"""
- This function is called very early before processing begins for AlwaysVisible scripts.
+ This function is called very early during processing begins for AlwaysVisible scripts.
You can modify the processing object (p) here, inject hooks, etc.
args contains all values returned by components from ui()
"""
@@ -706,6 +713,14 @@ class ScriptRunner: except Exception:
errors.report(f"Error running before_hr: {script.filename}", exc_info=True)
+ def setup_scrips(self, p):
+ for script in self.alwayson_scripts:
+ try:
+ script_args = p.script_args[script.args_from:script.args_to]
+ script.setup(p, *script_args)
+ except Exception:
+ errors.report(f"Error running setup: {script.filename}", exc_info=True)
+
scripts_txt2img: ScriptRunner = None
scripts_img2img: ScriptRunner = None
|