aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorAUTOMATIC <16777216c@gmail.com>2022-12-25 09:47:24 +0300
committerAUTOMATIC <16777216c@gmail.com>2022-12-25 09:47:34 +0300
commitc6f347b81f584b6c0d44af7a209983284dbb52d2 (patch)
treeccb8b89d64480a4bd224b311702ffeb13b8fe754
parent7b7f7e9361bda6e184fdee52c2fd277048a46f49 (diff)
fix ruined SD upscale
-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