Prevent upscaling when None is selected for SD upscale

2ef69df9 · Jocke · AUTOMATIC1111 · c0389eb3 · 2ef69df9
Commit 2ef69df9 authored Oct 02, 2022 by Jocke Committed by AUTOMATIC1111 Oct 02, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 5 additions and 1 deletion

scripts/sd_upscale.py scripts/sd_upscale.py +5 -1

No files found.
--- a/scripts/sd_upscale.py
+++ b/scripts/sd_upscale.py
@@ -34,7 +34,11 @@ class Script(scripts.Script):
        seed = p.seed

        init_img = p.init_images[0]
-        img = upscaler.scaler.upscale(init_img, 2, upscaler.data_path)
+        
+        if(upscaler.name != "None"): 
+            img = upscaler.scaler.upscale(init_img, 2, upscaler.data_path)
+        else:
+            img = init_img

        devices.torch_gc()