fix bugs in the PR
This commit is contained in:
parent
25414bcd05
commit
8f1b315318
|
@ -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):
|
||||||
|
|
|
@ -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
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue
Block a user