| license: mit | |
| # GeoGraph | |
| Pytorch checkpoint for the [GeoGraph](https://github.com/instadeepai/GeoGraph) model (raw PyTorch/Lightning-style checkpoint). | |
| ## Usage | |
| ```python | |
| import torch | |
| from huggingface_hub import hf_hub_download | |
| # Download the checkpoint into a local folder (or use the default cache) | |
| p = hf_hub_download( | |
| repo_id="InstaDeepAI/GeoGraph", | |
| filename="model.ckpt", | |
| local_dir="./weights" | |
| ) | |
| # Load like any local PyTorch file | |
| ckpt = torch.load(p, map_location="cpu", weights_only=False) | |