example defaults to JiT-H/16

example_sr_op.py

@@ -2,7 +2,7 @@
 """
 Genesis quick start — super-resolution + outpainting on a live Esri tile.
 
-Downloads the main JiT-B/16 checkpoints from https://huggingface.co/MVRL/genesis,
+Downloads the main JiT-H/16 checkpoints from https://huggingface.co/MVRL/genesis,
 fetches one real Esri World Imagery tile (Sydney Opera House / Circular Quay,
 zoom 16), then:
 
@@ -53,7 +53,7 @@ from utils.genesis_common import (  # demos/utils/genesis_common.py
 )
 
 HF_REPO = "MVRL/genesis"
-ARCH = "JiT-B/16"  # main B/16 checkpoints (H/16 SR alone is 23 GB)
+ARCH = "JiT-H/16"  # main H/16 checkpoints (~49 GB download; set "JiT-B/16" for the 7 GB variant)
 
 # Fixed scenic spot: Sydney Opera House / Circular Quay waterfront.
 LAT, LON, ZOOM = -33.8568, 151.2153, 16  # training zoom levels are 10..19
@@ -65,8 +65,8 @@ def main() -> None:
 
     # 1. Checkpoints from the HF Hub (cached under ~/.cache/huggingface).
     #    The main SR model is DINOv3-conditioned; the main OP model is no-DINO.
-    sr_ckpt = hf_hub_download(HF_REPO, "main/superresolution/main_B16/sr-full-tile-stage3-step0800000.ckpt")
-    op_ckpt = hf_hub_download(HF_REPO, "main/outpainting/main_B16/op-new-stage3-step0800000.ckpt")
+    sr_ckpt = hf_hub_download(HF_REPO, "main/superresolution/main_H16/sr-full-tile-stage3-step0800000.ckpt")
+    op_ckpt = hf_hub_download(HF_REPO, "main/outpainting/main_H16/op-new-stage3-step0800000.ckpt")
     dino = hf_hub_download(HF_REPO, "dinov3_vitl16_pretrain_sat493m-eadcf0ff.pth")
 
     # 2. One real 256x256 Esri World Imagery tile as the input.