diff --git a/scripts/sd_upscale.py b/scripts/sd_upscale.py index e614c23b..64e34cd9 100644 --- a/scripts/sd_upscale.py +++ b/scripts/sd_upscale.py @@ -98,4 +98,6 @@ class Script(scripts.Script): processed = Processed(p, result_images, seed, initial_info) + p.n_iter = upscale_count + return processed