aboutsummaryrefslogtreecommitdiffstats
path: root/modules/scripts.py
diff options
context:
space:
mode:
authorArtem Zagidulin <p0chta@yandex.ru>2022-11-02 16:05:01 +0000
committerArtem Zagidulin <p0chta@yandex.ru>2022-11-02 16:05:01 +0000
commita9e979977a8e3999b01b6a086bb1332ab7ab308b (patch)
treeb3c8721d1f6e2f0d029503a9bf2c0450a3aba917 /modules/scripts.py
parent172c4bc09f0866e7dd114068ebe0f9abfe79ef33 (diff)
downloadstable-diffusion-webui-gfx803-a9e979977a8e3999b01b6a086bb1332ab7ab308b.tar.gz
stable-diffusion-webui-gfx803-a9e979977a8e3999b01b6a086bb1332ab7ab308b.tar.bz2
stable-diffusion-webui-gfx803-a9e979977a8e3999b01b6a086bb1332ab7ab308b.zip
process_one
Diffstat (limited to 'modules/scripts.py')
-rw-r--r--modules/scripts.py16
1 files changed, 16 insertions, 0 deletions
diff --git a/modules/scripts.py b/modules/scripts.py
index 533db45c..9f82efea 100644
--- a/modules/scripts.py
+++ b/modules/scripts.py
@@ -70,6 +70,13 @@ class Script:
pass
+ def process_one(self, p, *args):
+ """
+ Same as process(), but called for every iteration
+ """
+
+ pass
+
def postprocess(self, p, processed, *args):
"""
This function is called after processing ends for AlwaysVisible scripts.
@@ -294,6 +301,15 @@ class ScriptRunner:
print(f"Error running process: {script.filename}", file=sys.stderr)
print(traceback.format_exc(), file=sys.stderr)
+ def process_one(self, p):
+ for script in self.alwayson_scripts:
+ try:
+ script_args = p.script_args[script.args_from:script.args_to]
+ script.process_one(p, *script_args)
+ except Exception:
+ print(f"Error running process_one: {script.filename}", file=sys.stderr)
+ print(traceback.format_exc(), file=sys.stderr)
+
def postprocess(self, p, processed):
for script in self.alwayson_scripts:
try: