fix ruined SD upscale

AUTOMATIC1111 · Dec 25, 2022 · c6f347b · c6f347b
1 parent 7b7f7e9
commit c6f347b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/scripts/sd_upscale.py b/scripts/sd_upscale.py
@@ -35,8 +35,9 @@ def run(self, p, _, overlap, upscaler_index, scale_factor):
         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