Browse Source

allow sdupscale to accept upscaler name

noodleanon 2 years ago
parent
commit
eadd1bf06a
1 changed files with 2 additions and 0 deletions
  1. 2 0
      scripts/sd_upscale.py

+ 2 - 0
scripts/sd_upscale.py

@@ -25,6 +25,8 @@ class Script(scripts.Script):
         return [info, overlap, upscaler_index, scale_factor]
         return [info, overlap, upscaler_index, scale_factor]
 
 
     def run(self, p, _, overlap, upscaler_index, scale_factor):
     def run(self, p, _, overlap, upscaler_index, scale_factor):
+        if isinstance(upscaler_index, str):
+            upscaler_index = [x.name.lower() for x in shared.sd_upscalers].index(upscaler_index.lower())
         processing.fix_seed(p)
         processing.fix_seed(p)
         upscaler = shared.sd_upscalers[upscaler_index]
         upscaler = shared.sd_upscalers[upscaler_index]