| 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)) | |