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

model probe detects sdxl lora models

Browse Source
This commit is contained in:
Lincoln Stein
2023-08-03 10:26:52 -04:00
committed by Kent Keirsey
parent 1ac14a1e43
commit 1d5d187ba1
5 changed files with 39 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
scripts/probe-model.py
View File

@ -9,8 +9,10 @@ parser = argparse.ArgumentParser(description="Probe model type")
parser.add_argument(
"model_path",
type=Path,
nargs="+",
)
args = parser.parse_args()
info = ModelProbe().probe(args.model_path)
print(info)
for path in args.model_path:
info = ModelProbe().probe(path)
print(f"{path}: {info}")