fix bugs in the PR

This commit is contained in:
AUTOMATIC 2022-09-30 09:46:52 +03:00
parent 25414bcd05
commit 8f1b315318
2 changed files with 2 additions and 2 deletions

View File

@ -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) 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: for filename in model_list:
h = model_hash(filename) h = model_hash(filename)
title = modeltitle(filename, h) title, model_name = modeltitle(filename, h)
checkpoints_list[title] = CheckpointInfo(filename, title, h, model_name) checkpoints_list[title] = CheckpointInfo(filename, title, h, model_name)
def get_closet_checkpoint_match(searchString): def get_closet_checkpoint_match(searchString):

View File

@ -60,7 +60,7 @@ class Upscaler:
break break
img = self.do_upscale(img, selected_model) img = self.do_upscale(img, selected_model)
if img.width != dest_w or img.height != dest_h: 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 return img