mirror of
https://github.com/kohya-ss/sd-scripts.git
synced 2026-04-06 13:47:06 +00:00
19 lines
468 B
Python
19 lines
468 B
Python
import json
|
|
import argparse
|
|
from safetensors import safe_open
|
|
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument("--model", type=str, required=True)
|
|
args = parser.parse_args()
|
|
|
|
with safe_open(args.model, framework="pt") as f:
|
|
metadata = f.metadata()
|
|
|
|
if metadata is None:
|
|
print("No metadata found")
|
|
else:
|
|
# metadata is json dict, but not pretty printed
|
|
# sort by key and pretty print
|
|
print(json.dumps(metadata, indent=4, sort_keys=True))
|
|
|
|
|