Commit 8f1b3153 authored by AUTOMATIC's avatar AUTOMATIC
Browse files

fix bugs in the PR

parent 25414bcd
Loading
Loading
Loading
Loading
+1 −1
Original line number Diff line number Diff line
@@ -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):
+1 −1
Original line number Diff line number Diff line
@@ -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