Update README.md
Browse files
README.md
CHANGED
|
@@ -19,38 +19,4 @@ README/`docs/ARCHITECTURE.md` for the full model description).
|
|
| 19 |
[mudhil/era5-conus404-diffusion-downscaling](https://huggingface.co/mudhil/era5-conus404-diffusion-downscaling)
|
| 20 |
instead** — this repo's `diffusion_best.pt`/`diffusion_latest.pt` are an
|
| 21 |
older diffusion checkpoint (confirmed by sha256, different from the current
|
| 22 |
-
local weights). The DRN and VAE files in this repo *are* current.
|
| 23 |
-
|
| 24 |
-
## Files (verified by sha256 against local `checkpoints/`, 2026-09-27)
|
| 25 |
-
|
| 26 |
-
| File | Size | Status |
|
| 27 |
-
|---|---|---|
|
| 28 |
-
| `drn_best.pt` | 596 MB | ✅ current |
|
| 29 |
-
| `drn_latest.pt` | 596 MB | ✅ current |
|
| 30 |
-
| `vae_best.pt` | 476 MB | ✅ current |
|
| 31 |
-
| `vae_latest.pt` | 476 MB | ✅ current |
|
| 32 |
-
| `vae_best_prespectral.pt` | 476 MB | archival — VAE before the spectral reconstruction loss term was added |
|
| 33 |
-
| `vae_latest_prespectral.pt` | 476 MB | archival — same as above |
|
| 34 |
-
| `diffusion_best_oldvae.pt` | 2.3 GB | archival — diffusion model paired with a since-replaced VAE version |
|
| 35 |
-
| `diffusion_latest_oldvae.pt` | 2.3 GB | archival — same as above |
|
| 36 |
-
| `diffusion_best.pt` | 2.3 GB | ⚠️ older diffusion checkpoint — see note above |
|
| 37 |
-
| `diffusion_latest.pt` | 2.3 GB | ⚠️ older diffusion checkpoint — see note above |
|
| 38 |
-
|
| 39 |
-
## What this model does
|
| 40 |
-
|
| 41 |
-
ERA5 (0.25°, ~27km) → CONUS404 (4km) downscaling over CONUS. Six variables:
|
| 42 |
-
t2m/d2m/u10/v10/sp/tp → T2/TD2/U10/V10/PSFC/PREC_ACC_NC. DRN (49.6M params)
|
| 43 |
-
predicts the conditional mean; a VAE (39.7M params) compresses the residual
|
| 44 |
-
to a 64×64×8 latent; an EDM diffusion UNet (142.4M params) generates the
|
| 45 |
-
latent residual.
|
| 46 |
-
|
| 47 |
-
**Known limitation:** precipitation generation is a confirmed failure mode
|
| 48 |
-
(unphysical ensemble spikes). See the project's `event_benchmark_output/README.md`
|
| 49 |
-
for full completed results, including a comparison against self-trained
|
| 50 |
-
CorrDiff-style and R2-D2-style baselines.
|
| 51 |
-
|
| 52 |
-
## Usage
|
| 53 |
-
|
| 54 |
-
```bash
|
| 55 |
-
hf download mudhil/diffusion-downscaling-checkpoints --local-dir checkpoints/
|
| 56 |
-
```
|
|
|
|
| 19 |
[mudhil/era5-conus404-diffusion-downscaling](https://huggingface.co/mudhil/era5-conus404-diffusion-downscaling)
|
| 20 |
instead** — this repo's `diffusion_best.pt`/`diffusion_latest.pt` are an
|
| 21 |
older diffusion checkpoint (confirmed by sha256, different from the current
|
| 22 |
+
local weights). The DRN and VAE files in this repo *are* current.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|