InvokeAI/scripts/probe-model.py

#!/bin/env python

import argparse
import sys
from pathlib import Path
from invokeai.backend.model_management.model_probe import ModelProbe

parser = argparse.ArgumentParser(description="Probe model type")
parser.add_argument(
    "model_path",
    type=Path,
)
args = parser.parse_args()

info = ModelProbe().probe(args.model_path)
print(info)
fix "unknown model type" error when rebasing a model with API - Add command-line model probing script for dev use - Minor documentation tweak 2023-07-25 12:36:57 +00:00			`#!/bin/env python`

			`import argparse`
			`import sys`
			`from pathlib import Path`
			`from invokeai.backend.model_management.model_probe import ModelProbe`

			`parser = argparse.ArgumentParser(description="Probe model type")`
			`parser.add_argument(`
reformat with black 2023-07-27 19:01:00 +00:00			`"model_path",`
fix "unknown model type" error when rebasing a model with API - Add command-line model probing script for dev use - Minor documentation tweak 2023-07-25 12:36:57 +00:00			`type=Path,`
			`)`
reformat with black 2023-07-27 19:01:00 +00:00			`args = parser.parse_args()`
fix "unknown model type" error when rebasing a model with API - Add command-line model probing script for dev use - Minor documentation tweak 2023-07-25 12:36:57 +00:00
			`info = ModelProbe().probe(args.model_path)`
			`print(info)`