aboutsummaryrefslogtreecommitdiff
path: root/scripts
diff options
context:
space:
mode:
Diffstat (limited to 'scripts')
-rw-r--r--scripts/sd_upscale.py3
1 files changed, 2 insertions, 1 deletions
diff --git a/scripts/sd_upscale.py b/scripts/sd_upscale.py
index 28bd96b3..e8c80a6c 100644
--- a/scripts/sd_upscale.py
+++ b/scripts/sd_upscale.py
@@ -35,8 +35,9 @@ class Script(scripts.Script):
seed = p.seed
init_img = p.init_images[0]
+ init_img = images.flatten(init_img, opts.img2img_background_color)
- if (upscaler.name != "None"):
+ if upscaler.name != "None":
img = upscaler.scaler.upscale(init_img, scale_factor, upscaler.data_path)
else:
img = init_img