change process_one virtual function for script to process_batch, add extra args and docs

This commit is contained in:
AUTOMATIC
2022-11-04 11:21:40 +03:00
parent 99043f3360
commit eeb0733013
2 changed files with 12 additions and 6 deletions

View File

@@ -73,9 +73,15 @@ class Script:
pass
def process_one(self, p, n, *args):
def process_batch(self, p, *args, **kwargs):
"""
Same as process(), but called for every iteration
Same as process(), but called for every batch.
**kwargs will have those items:
- batch_number - index of current batch, from 0 to number of batches-1
- prompts - list of prompts for current batch; you can change contents of this list but changing the number of entries will likely break things
- seeds - list of seeds for current batch
- subseeds - list of subseeds for current batch
"""
pass
@@ -303,13 +309,13 @@ class ScriptRunner:
print(f"Error running process: {script.filename}", file=sys.stderr)
print(traceback.format_exc(), file=sys.stderr)
def process_one(self, p, n):
def process_batch(self, p, **kwargs):
for script in self.alwayson_scripts:
try:
script_args = p.script_args[script.args_from:script.args_to]
script.process_one(p, n, *script_args)
script.process_batch(p, *script_args, **kwargs)
except Exception:
print(f"Error running process_one: {script.filename}", file=sys.stderr)
print(f"Error running process_batch: {script.filename}", file=sys.stderr)
print(traceback.format_exc(), file=sys.stderr)
def postprocess(self, p, processed):