example.py

#!/usr/bin/env python3
import torch
# weights_only=False required for full pickle support
ckpt = torch.load("pytorch_model.bin", weights_only=False, map_location="cpu")
print(f"Model loaded: {len(ckpt['model_state_dict'])} layers")