From 0a0f6b7f6b18605be32449277fe823e0f8dc960f Mon Sep 17 00:00:00 2001
From: David Errington <davecerr@users.noreply.github.com>
Date: Tue, 28 Jul 2026 07:26:29 -0600
Subject: [PATCH] Fetch hub/config.json so HuggingFace download stats are
 counted

---
 hub/config.json              | 11 +++++++++++
 nesso/main.py                | 20 ++++++++++++++++++++
 nesso/model/models/nesso1.py | 13 +++++++++++++
 3 files changed, 44 insertions(+)
 create mode 100644 hub/config.json

diff --git a/hub/config.json b/hub/config.json
new file mode 100644
index 0000000..c7cd4ed
--- /dev/null
+++ b/hub/config.json
@@ -0,0 +1,11 @@
+{
+  "model_type": "nesso1",
+  "architectures": ["Nesso1"],
+  "library_name": "nesso",
+  "default_revision": "v1.0.0",
+  "files": {
+    "hparams": "hparams.json",
+    "weights": "model.safetensors",
+    "ccd": "ccd.pkl"
+  }
+}
diff --git a/nesso/main.py b/nesso/main.py
index 1238424..c29a46e 100644
--- a/nesso/main.py
+++ b/nesso/main.py
@@ -39,6 +39,9 @@
 FALLBACK_REVISION = "main"
 MODEL_WEIGHTS_NAME = "model.safetensors"
 MODEL_HPARAMS_NAME = "hparams.json"
+# Root query file used by the Hub for download counting (GET/HEAD on this path).
+# See https://huggingface.co/docs/hub/models-download-stats
+MODEL_CONFIG_NAME = "config.json"
 
 
 def get_default_model_revision() -> str:
@@ -109,6 +112,23 @@ def ensure_cache(
     if checkpoint is not None:
         model_dir = checkpoint
     else:
+        # Fetch the Hub query file so this load is counted in download stats.
+        # Always from ``main`` so version tags need not re-copy this file.
+        try:
+            from huggingface_hub.errors import EntryNotFoundError
+
+            hf_hub_download(
+                repo_id=HF_REPO_ID,
+                filename=MODEL_CONFIG_NAME,
+                revision="main",
+                cache_dir=hf_cache_dir,
+            )
+        except EntryNotFoundError as exc:
+            warnings.warn(
+                f"Could not download Hub query file {MODEL_CONFIG_NAME!r} "
+                f"(download stats will not increment): {exc}",
+                stacklevel=2,
+            )
         weights = Path(
             hf_hub_download(
                 repo_id=HF_REPO_ID,
diff --git a/nesso/model/models/nesso1.py b/nesso/model/models/nesso1.py
index 1c1f841..c52a441 100644
--- a/nesso/model/models/nesso1.py
+++ b/nesso/model/models/nesso1.py
@@ -483,6 +483,19 @@ def from_pretrained(
                 except PackageNotFoundError:
                     hub_revision = "main"
 
+            # Root ``config.json`` is the Hub download-stats query file; fetch it
+            # from ``main`` so Hub loads are counted (hparams/weights alone are not).
+            try:
+                from huggingface_hub.errors import EntryNotFoundError
+
+                hf_hub_download(
+                    repo_id=str(pretrained_model_name_or_path),
+                    filename="config.json",
+                    revision="main",
+                    cache_dir=cache_dir,
+                )
+            except EntryNotFoundError:
+                pass
             hparams_path = Path(
                 hf_hub_download(
                     repo_id=str(pretrained_model_name_or_path),