add number of itter

2 files changed, 4 insertions, 4 deletions

diff --git a/modules/processing.py b/modules/processing.py
index 72a2ee4e..17f4a5ec 100644
--- a/modules/processing.py
+++ b/modules/processing.py
@@ -510,7 +510,7 @@ def process_images_inner(p: StableDiffusionProcessing) -> Processed:
                 break

 

             if p.scripts is not None:

-                p.scripts.process_one(p)

+                p.scripts.process_one(p, n)

 

             with devices.autocast():

                 uc = prompt_parser.get_learned_conditioning(shared.sd_model, len(prompts) * [p.negative_prompt], p.steps)

diff --git a/modules/scripts.py b/modules/scripts.py
index 9f82efea..7aa0d56a 100644
--- a/modules/scripts.py
+++ b/modules/scripts.py
@@ -70,7 +70,7 @@ class Script:
 

         pass

 

-    def process_one(self, p, *args):

+    def process_one(self, p, n, *args):

         """

         Same as process(), but called for every iteration

         """

@@ -301,11 +301,11 @@ class ScriptRunner:
                 print(f"Error running process: {script.filename}", file=sys.stderr)

                 print(traceback.format_exc(), file=sys.stderr)

 

-    def process_one(self, p):

+    def process_one(self, p, n):

         for script in self.alwayson_scripts:

             try:

                 script_args = p.script_args[script.args_from:script.args_to]

-                script.process_one(p, *script_args)

+                script.process_one(p, n, *script_args)

             except Exception:

                 print(f"Error running process_one: {script.filename}", file=sys.stderr)

                 print(traceback.format_exc(), file=sys.stderr)