Commit de64146a authored by Artem Zagidulin's avatar Artem Zagidulin

add number of itter

parent a9e97997
...@@ -510,7 +510,7 @@ def process_images_inner(p: StableDiffusionProcessing) -> Processed: ...@@ -510,7 +510,7 @@ def process_images_inner(p: StableDiffusionProcessing) -> Processed:
break break
if p.scripts is not None: if p.scripts is not None:
p.scripts.process_one(p) p.scripts.process_one(p, n)
with devices.autocast(): with devices.autocast():
uc = prompt_parser.get_learned_conditioning(shared.sd_model, len(prompts) * [p.negative_prompt], p.steps) uc = prompt_parser.get_learned_conditioning(shared.sd_model, len(prompts) * [p.negative_prompt], p.steps)
......
...@@ -70,7 +70,7 @@ class Script: ...@@ -70,7 +70,7 @@ class Script:
pass pass
def process_one(self, p, *args): def process_one(self, p, n, *args):
""" """
Same as process(), but called for every iteration Same as process(), but called for every iteration
""" """
...@@ -301,11 +301,11 @@ class ScriptRunner: ...@@ -301,11 +301,11 @@ class ScriptRunner:
print(f"Error running process: {script.filename}", file=sys.stderr) print(f"Error running process: {script.filename}", file=sys.stderr)
print(traceback.format_exc(), 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: for script in self.alwayson_scripts:
try: try:
script_args = p.script_args[script.args_from:script.args_to] 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: except Exception:
print(f"Error running process_one: {script.filename}", file=sys.stderr) print(f"Error running process_one: {script.filename}", file=sys.stderr)
print(traceback.format_exc(), file=sys.stderr) print(traceback.format_exc(), file=sys.stderr)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment