from safetensors import safe_open | |
resolved_archive_file = "model.safetensors" | |
try: | |
with safe_open(resolved_archive_file, framework="pt") as f: | |
metadata = f.metadata() | |
print(metadata) | |
except Exception as e: | |
print(f"Failed to open safetensor file: {e}") |