Add release configuration and document download statistics

README.md

@@ -10,6 +10,7 @@ The released inference bundle is available in [weights/](https://huggingface.co/
 
 | File | Purpose |
 |---|---|
+| `config.json` | Release configuration and model-bundle file map |
 | `weights/la_ribo_base.pt` | Base flow model, exported EMA weights |
 | `weights/la_ribo_tri.pt` | Tri flow model, exported EMA weights |
 | `weights/la_ribo_vae.pt` | Shared autoencoder weights |
@@ -24,6 +25,7 @@ Run this from the local [La-Ribo inference repository](https://github.com/GENTEL
 
 ```bash
 uvx --from huggingface_hub hf download GENTEL-Lab/La-Ribo \
+  config.json \
   weights/la_ribo_base.pt \
   weights/la_ribo_tri.pt \
   weights/la_ribo_vae.pt \
@@ -34,6 +36,12 @@ uvx --from huggingface_hub hf download GENTEL-Lab/La-Ribo \
 
 Both variants use the shared autoencoder and both statistics files. Choose `config/la-ribo-base.yaml` with `la_ribo_base.pt`, or `config/la-ribo-tri.yaml` with `la_ribo_tri.pt`. Installation and CPU/CUDA inference commands are in the [GitHub README](https://github.com/GENTEL-lab/La-Ribo#readme). The default sampling protocol uses 200 steps and scaffold/latent powers 1.25/2.0.
 
+`config.json` describes the released variants, weight files, architecture dimensions, and default sampling settings. It contains public inference configuration only.
+
+### Download statistics
+
+Hugging Face uses access to the root `config.json` as the default download-counting signal for this model repository. The command above includes that file. Direct downloads of only `.pt` files or files under `data/` do not trigger this default model counter. HF counts matching GET/HEAD requests server-side; the displayed value is not a count of unique users or guaranteed completed weight transfers. See the [official counting rules](https://huggingface.co/docs/hub/models-download-stats).
+
 The data below are optional for inference.
 
 ## Data

config.json

@@ -0,0 +1,135 @@
+{
+  "format": "la-ribo-inference-bundle",
+  "format_version": 1,
+  "code_repository": "https://github.com/GENTEL-lab/La-Ribo",
+  "paper": "https://arxiv.org/abs/2610.12236",
+  "checkpoint_format": "pytorch_state_dict",
+  "variants": {
+    "base": {
+      "checkpoint": "weights/la_ribo_base.pt",
+      "weights": "ema",
+      "inference_config": "config/la-ribo-base.yaml",
+      "flow_model": {
+        "latent_dim": 16,
+        "explicit_backbone_mode": "p_c4prime_n",
+        "coarse_dim": 9,
+        "relpos_dim": 64,
+        "dist_num_bins": 16,
+        "dist_min": 0.0,
+        "dist_max": 32.0,
+        "time_emb_dim": 64,
+        "token_dim": 512,
+        "pair_dim": 256,
+        "cond_dim": 128,
+        "num_layers": 12,
+        "num_heads": 8,
+        "use_qkln": true,
+        "transition_expansion": 4,
+        "pair_update": {
+          "enabled": false
+        }
+      },
+      "autoencoder": {
+        "num_nt_types": 4,
+        "num_torsions": 10,
+        "latent_dim": 16,
+        "explicit_backbone": {
+          "mode": "p_c4prime_n",
+          "zero_center": true
+        },
+        "decoder_conf": {
+          "relpos_dim": 64,
+          "dist_num_bins": 16,
+          "dist_min": 0.0,
+          "dist_max": 32.0,
+          "token_dim": 256,
+          "pair_dim": 128,
+          "cond_dim": 64,
+          "num_layers": 6,
+          "num_heads": 8,
+          "use_qkln": true
+        }
+      }
+    },
+    "tri": {
+      "checkpoint": "weights/la_ribo_tri.pt",
+      "weights": "ema",
+      "inference_config": "config/la-ribo-tri.yaml",
+      "flow_model": {
+        "latent_dim": 16,
+        "explicit_backbone_mode": "p_c4prime_n",
+        "coarse_dim": 9,
+        "relpos_dim": 64,
+        "dist_num_bins": 16,
+        "dist_min": 0.0,
+        "dist_max": 32.0,
+        "time_emb_dim": 64,
+        "token_dim": 512,
+        "pair_dim": 256,
+        "cond_dim": 128,
+        "num_layers": 12,
+        "num_heads": 8,
+        "use_qkln": true,
+        "transition_expansion": 4,
+        "pair_update": {
+          "enabled": true,
+          "every": 2,
+          "use_cuequivariance": true,
+          "expansion": 2,
+          "token_to_pair": "add"
+        }
+      },
+      "autoencoder": {
+        "num_nt_types": 4,
+        "num_torsions": 10,
+        "latent_dim": 16,
+        "explicit_backbone": {
+          "mode": "p_c4prime_n",
+          "zero_center": true
+        },
+        "decoder_conf": {
+          "relpos_dim": 64,
+          "dist_num_bins": 16,
+          "dist_min": 0.0,
+          "dist_max": 32.0,
+          "token_dim": 256,
+          "pair_dim": 128,
+          "cond_dim": 64,
+          "num_layers": 6,
+          "num_heads": 8,
+          "use_qkln": true
+        }
+      }
+    }
+  },
+  "shared_files": {
+    "autoencoder": "weights/la_ribo_vae.pt",
+    "latent_stats": "weights/latent_stats.pt",
+    "coarse_stats": "weights/coarse_stats.pt"
+  },
+  "sampling": {
+    "solver": "forward_euler",
+    "num_timesteps": 200,
+    "scaffold_power": 1.25,
+    "latent_power": 2.0,
+    "self_condition": true
+  },
+  "supported_devices": [
+    "cpu",
+    "cuda"
+  ],
+  "benchmark_lengths": [
+    40,
+    50,
+    60,
+    70,
+    80,
+    90,
+    100,
+    110,
+    120,
+    130,
+    140,
+    150
+  ]
+}