NEMOtools / app /test.py
AndrewKof's picture
πŸš€ Update: new UI design, classification plot, and attention improvements
c34dda4
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])