aboutsummaryrefslogtreecommitdiffstats
path: root/modules/processing.py
diff options
context:
space:
mode:
authorAUTOMATIC1111 <16777216c@gmail.com>2023-03-11 11:33:38 +0000
committerGitHub <noreply@github.com>2023-03-11 11:33:38 +0000
commitda3f942ab2171e11adf47cd21182db644b9c400a (patch)
tree195ee0c27ce6fdb54146b91eb070f26294443cd7 /modules/processing.py
parentaaa367e35ce4e823219c2954ca141ca1ed14800e (diff)
parenta2d635ad135241a0a40f67f7e1638c9c8a4ded04 (diff)
downloadstable-diffusion-webui-gfx803-da3f942ab2171e11adf47cd21182db644b9c400a.tar.gz
stable-diffusion-webui-gfx803-da3f942ab2171e11adf47cd21182db644b9c400a.tar.bz2
stable-diffusion-webui-gfx803-da3f942ab2171e11adf47cd21182db644b9c400a.zip
Merge pull request #8017 from space-nuko/before-process-batch
Add `before_process_batch` script callback
Diffstat (limited to 'modules/processing.py')
-rw-r--r--modules/processing.py3
1 files changed, 3 insertions, 0 deletions
diff --git a/modules/processing.py b/modules/processing.py
index 0b2f7e60..06e7a440 100644
--- a/modules/processing.py
+++ b/modules/processing.py
@@ -597,6 +597,9 @@ def process_images_inner(p: StableDiffusionProcessing) -> Processed:
seeds = p.all_seeds[n * p.batch_size:(n + 1) * p.batch_size]
subseeds = p.all_subseeds[n * p.batch_size:(n + 1) * p.batch_size]
+ if p.scripts is not None:
+ p.scripts.before_process_batch(p, batch_number=n, prompts=prompts, seeds=seeds, subseeds=subseeds)
+
if len(prompts) == 0:
break