Commit fb97acef authored by Cody Brownstein's avatar Cody Brownstein
Browse files

Update error message WRT missing checkpoint file

The Safetensors format is also supported.
parent 226d840e
Loading
Loading
Loading
Loading
+1 −1
Original line number Original line Diff line number Diff line
@@ -158,7 +158,7 @@ def select_checkpoint():
        print(f" - directory {model_path}", file=sys.stderr)
        print(f" - directory {model_path}", file=sys.stderr)
        if shared.cmd_opts.ckpt_dir is not None:
        if shared.cmd_opts.ckpt_dir is not None:
            print(f" - directory {os.path.abspath(shared.cmd_opts.ckpt_dir)}", file=sys.stderr)
            print(f" - directory {os.path.abspath(shared.cmd_opts.ckpt_dir)}", file=sys.stderr)
        print("Can't run without a checkpoint. Find and place a .ckpt file into any of those locations. The program will exit.", file=sys.stderr)
        print("Can't run without a checkpoint. Find and place a .ckpt or .safetensors file into any of those locations. The program will exit.", file=sys.stderr)
        exit(1)
        exit(1)


    checkpoint_info = next(iter(checkpoints_list.values()))
    checkpoint_info = next(iter(checkpoints_list.values()))