Restore last generation params

This commit is contained in:
Trung Ngo
2022-10-08 22:57:19 -05:00
committed by AUTOMATIC1111
parent 04c0e643f2
commit e72adc999b
4 changed files with 14 additions and 1 deletions

View File

@@ -324,6 +324,10 @@ def process_images(p: StableDiffusionProcessing) -> Processed:
else:
assert p.prompt is not None
with open(os.path.join(shared.script_path, "params.txt"), "w", encoding="utf8") as file:
processed = Processed(p, [], p.seed, "")
file.write(processed.infotext(p, 0))
devices.torch_gc()
seed = get_fixed_seed(p.seed)