Prevent upscaling when None is selected for SD upscale

2022-10-02 16:10:41 +02:00 · 2022-10-02 16:10:41 +02:00 · 2ef69df9a7
parent c0389eb307
commit 2ef69df9a7
1 changed files with 5 additions and 1 deletions
--- 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()