for img2img, use None as upscaler instead of erroring out if the desired upscaler is not found
This commit is contained in:
parent
532ac22b38
commit
4c1ad743e3
|
@ -261,9 +261,12 @@ def resize_image(resize_mode, im, width, height, upscaler_name=None):
|
||||||
|
|
||||||
if scale > 1.0:
|
if scale > 1.0:
|
||||||
upscalers = [x for x in shared.sd_upscalers if x.name == upscaler_name]
|
upscalers = [x for x in shared.sd_upscalers if x.name == upscaler_name]
|
||||||
assert len(upscalers) > 0, f"could not find upscaler named {upscaler_name}"
|
if len(upscalers) == 0:
|
||||||
|
upscaler = shared.sd_upscalers[0]
|
||||||
|
print(f"could not find upscaler named {upscaler_name or '<empty string>'}, using {upscaler.name} as a fallback")
|
||||||
|
else:
|
||||||
|
upscaler = upscalers[0]
|
||||||
|
|
||||||
upscaler = upscalers[0]
|
|
||||||
im = upscaler.scaler.upscale(im, scale, upscaler.data_path)
|
im = upscaler.scaler.upscale(im, scale, upscaler.data_path)
|
||||||
|
|
||||||
if im.width != w or im.height != h:
|
if im.width != w or im.height != h:
|
||||||
|
|
Loading…
Reference in New Issue