import torch
ckpt = torch.load('storage/proto_lm/checkpoint.pt', map_location='cpu')
print(f"Step: {ckpt['step']}")
print(f"Vocab size: {len(ckpt['vocab']['tokens'])}")
print(f"Keys: {list(ckpt.keys())}")

# Check if it's the fresh one we created
if ckpt['step'] == 5:
    print("✓ Fresh checkpoint loaded")
else:
    print(f"✗ OLD checkpoint loaded (step {ckpt['step']})")