File size: 378 Bytes
c34dda4 |
1 2 3 4 5 6 7 8 9 10 11 |
from safetensors.torch import load_file
from pprint import pprint
import os
path = os.path.expanduser("~/.cache/huggingface/hub/models--Arew99--dinov2-costum/snapshots/055a10af249d426a5b9a6ac07550f011e5739bbf/model.safetensors")
print(f"Loading checkpoint: {path}")
sd = load_file(path)
print(f"✅ Loaded {len(sd)} tensors")
print("Sample keys:")
pprint(list(sd.keys())[:30])
|