浏览代码

Prevent upscaling when None is selected for SD upscale

Jocke 2 年之前
父节点
当前提交
2ef69df9a7
共有 1 个文件被更改,包括 5 次插入1 次删除
  1. 5 1
      scripts/sd_upscale.py

+ 5 - 1
scripts/sd_upscale.py

@@ -34,7 +34,11 @@ class Script(scripts.Script):
         seed = p.seed
         seed = p.seed
 
 
         init_img = p.init_images[0]
         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()
         devices.torch_gc()