diff options
| author | noodleanon <122053346+noodleanon@users.noreply.github.com> | 2023-01-05 21:22:04 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-01-05 21:22:04 +0000 |
| commit | eadd1bf06adbd7263875640a6446d3b0184d1561 (patch) | |
| tree | 52baac0997757593308ee99693c75dd0d065e606 /scripts/sd_upscale.py | |
| parent | b5253f0dab529707f1fe2e11211a10ce2f264617 (diff) | |
allow sdupscale to accept upscaler name
Diffstat (limited to 'scripts/sd_upscale.py')
| -rw-r--r-- | scripts/sd_upscale.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/scripts/sd_upscale.py b/scripts/sd_upscale.py index 9b8ffd85..332d76d9 100644 --- a/scripts/sd_upscale.py +++ b/scripts/sd_upscale.py @@ -25,6 +25,8 @@ class Script(scripts.Script): return [info, 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)
upscaler = shared.sd_upscalers[upscaler_index]
|