Commit 8f1b3153 authored by AUTOMATIC's avatar AUTOMATIC

fix bugs in the PR

parent 25414bcd
......@@ -96,7 +96,7 @@ def list_models():
print(f"Checkpoint in --ckpt argument not found (Possible it was moved to {model_path}: {cmd_ckpt}", file=sys.stderr)
for filename in model_list:
h = model_hash(filename)
title = modeltitle(filename, h)
title, model_name = modeltitle(filename, h)
checkpoints_list[title] = CheckpointInfo(filename, title, h, model_name)
def get_closet_checkpoint_match(searchString):
......
......@@ -60,7 +60,7 @@ class Upscaler:
break
img = self.do_upscale(img, selected_model)
if img.width != dest_w or img.height != dest_h:
img = img.resize(dest_w, dest_h, resample=LANCZOS)
img = img.resize((dest_w, dest_h), resample=LANCZOS)
return img
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment