1import torch
2from huggingface_hub import hf_hub_download
3
4# Download the checkpoint into a local folder (or use the default cache)
5p = hf_hub_download(
6 repo_id="InstaDeepAI/GeoGraph",
7 filename="model.ckpt",
8 local_dir="./weights"
9)
10
11# Load like any local PyTorch file
12ckpt = torch.load(p, map_location="cpu", weights_only=False)