Upload folder using huggingface_hub

README.md

@@ -6,12 +6,13 @@ tags:
   - era5
   - pytorch
   - sonny
+  - stepnets
 pipeline_tag: other
 ---
 
-# Sonny — 69-channel ERA5-like backbone (120×240, 6 h)
+# Sonny
 
-**Sonny** is a ViT-style one-step weather backbone trained at 1.5°-style resolution (lat×lon = 120×240), 69 ERA5-like variables, patch size 2, EMA weights at epoch 50.
+**Sonny** is a **StepNets**-style one-step weather backbone: variable-aware embedding, then a two-stage trunk (narrow **Step 1** “slow path” over dynamics-heavy channels, full-width **Step 2** “fast path” after fusion), timestep conditioning, and patch-wise prediction. Trained at 1.5°-style resolution (**120×240**), **69** ERA5-like variables, patch size **2**, **EMA** weights at epoch **50**.
 
 ## Loading (with the `stormer` codebase)
 
@@ -28,7 +29,7 @@ Or download files manually and build :class:`~stormer.models.hub.sonny.Sonny`(**
 
 ## Files
 
-- `config.json` — hyperparameters / variable list for :class:`Sonny`
+- `config.json` — hyperparameters / variable list for **Sonny**
 - `model.safetensors` — EMA ``ema_model_state_dict``
 - `normalization_note.txt` — reminder to use your local ``normalization_constants`` from the training repo for real inference
 

config.json

@@ -1,6 +1,8 @@
 {
+  "model_name": "Sonny",
   "sonny_version": "1.0",
-  "architecture_name": "sonny",
+  "architecture_name": "Sonny",
+  "architecture_family": "StepNets",
   "class_qualname": "stormer.models.hub.sonny.Sonny",
   "source_checkpoint": "stormbreaker5_checkpoint_ema_epoch_050.pth",
   "source_epoch": 50,