Upload README.md with huggingface_hub

README.md

@@ -20,21 +20,25 @@ Ivan Lee, Cheng Yang, Taylor Berg-Kirkpatrick
 
 ## Available Checkpoints
 
-| Checkpoint | Objective | Hybrid Tokens | Training | PPL |
-|------------|-----------|---------------|----------|-----|
-| `vision_base_hybrid0_reconstruction` | Reconstruction | 0 | Direct | 1.03 |
-| `vision_base_hybrid0_lm_direct` | Language Modeling | 0 | Direct (no recon init) | 5.08 |
-| `vision_base_hybrid0_lm_recon_init` | Language Modeling | 0 | Initialized from reconstruction | 5.06 |
+| Checkpoint | Objective | Hybrid | Training | PPL |
+|------------|-----------|--------|----------|-----|
+| `vision_base_h0_recon` | Reconstruction | 0 | - | 1.03 |
+| `vision_base_h0_lm` | LM | 0 | Direct | 5.08 |
+| `vision_base_h0_lm_recon-init` | LM | 0 | From reconstruction | 5.06 |
 
-### Naming Convention
+## Naming Convention
 
-`{regime}_{size}_{hybrid}_[reconstruction|lm]_[direct|recon_init]`
+```
+{regime}_{config}_h{N}_{objective}[_recon-init]
+```
 
-- **regime**: vision, conv1d_residual, meanpool, text
-- **size**: tiny, small, base, large (for vision); compression target (for others)
-- **hybrid**: hybrid0 (pure vision/compression) or hybrid100 (100 text tokens + vision)
-- **objective**: reconstruction or lm (language modeling)
-- **training**: direct (trained from scratch) or recon_init (initialized from reconstruction checkpoint)
+| Field | Values | Description |
+|-------|--------|-------------|
+| regime | vision, conv1d, meanpool, text | Compression architecture |
+| config | base/small/tiny/large, t500/t250, w10s10, ctx525 | Regime-specific config |
+| h{N} | h0, h100 | Hybrid text tokens (0 = pure vision) |
+| objective | recon, lm | Training objective |
+| recon-init | (optional) | LM initialized from reconstruction checkpoint |
 
 ## Model Details
 
@@ -51,7 +55,7 @@ from huggingface_hub import hf_hub_download
 # Download a specific checkpoint
 checkpoint_path = hf_hub_download(
     repo_id="ivnle/bad-autoencoding",
-    filename="vision_base_hybrid0_lm_direct/model.pt",
+    filename="vision_base_h0_lm/model.pt",
     repo_type="model"
 )
 ```