mirror of
https://github.com/kohya-ss/sd-scripts.git
synced 2026-04-06 05:44:56 +00:00
5f6bf29e52
* Add get_my_logger() * Use logger instead of print * Fix log level * Removed line-breaks for readability * Use setup_logging() * Add rich to requirements.txt * Make simple * Use logger instead of print --------- Co-authored-by: Kohya S <52813779+kohya-ss@users.noreply.github.com>
24 lines
584 B
Python
24 lines
584 B
Python
import json
|
|
import argparse
|
|
from safetensors import safe_open
|
|
from library.utils import setup_logging
|
|
setup_logging()
|
|
import logging
|
|
logger = logging.getLogger(__name__)
|
|
|
|
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:
|
|
logger.error("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))
|
|
|
|
|