GeoGraph / README.md
jeanq1's picture
Update README.md
55219b8 verified
metadata
license: mit

GeoGraph

Pytorch checkpoint for the GeoGraph model (raw PyTorch/Lightning-style checkpoint).

Usage

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)