Update scripts/verify_checkpoint_template.py

Co-authored-by: Eugene Brodsky <ebr@users.noreply.github.com>
This commit is contained in:
Kent Keirsey 2023-08-10 16:00:33 -04:00 committed by GitHub
parent 2f68a1a76c
commit 6c8e898f09
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -12,7 +12,7 @@ from pathlib import Path
from invokeai.backend.model_management.models.base import read_checkpoint_meta from invokeai.backend.model_management.models.base import read_checkpoint_meta
parser = argparse.ArgumentParser(description="Create a .json template from checkpoint/safetensors model") parser = argparse.ArgumentParser(description="Compare a checkpoint/safetensors file to a JSON metadata template.")
parser.add_argument("--checkpoint", "--in", type=Path, help="Path to the input checkpoint/safetensors file") parser.add_argument("--checkpoint", "--in", type=Path, help="Path to the input checkpoint/safetensors file")
parser.add_argument("--template", "--out", type=Path, help="Path to the template .json file to match against") parser.add_argument("--template", "--out", type=Path, help="Path to the template .json file to match against")