Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix model probing for controlnet checkpoint legacy config files

Browse Source
This commit is contained in:
Lincoln Stein
2023-11-25 15:53:22 -05:00
parent 19baea1883
commit ec510d34b5
5 changed files with 269 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
scripts/probe-model.py
View File

@ -1,9 +1,13 @@
#!/bin/env python
"""Little command-line utility for probing a model on disk."""
import argparse
from pathlib import Path
from invokeai.backend.model_manager.probe import ModelProbe
from invokeai.backend.model_manager import ModelProbe, InvalidModelConfigException
parser = argparse.ArgumentParser(description="Probe model type")
parser.add_argument(
@ -14,5 +18,8 @@ parser.add_argument(
args = parser.parse_args()
for path in args.model_path:
info = ModelProbe().probe(path)
print(f"{path}: {info}")
try:
info = ModelProbe.probe(path)
print(f"{path}:{info.model_dump_json(indent=4)}")
except InvalidModelConfigException as exc:
print(exc)