Update README.md

README.md

@@ -82,8 +82,8 @@ Hugging Face: 07/27/2026 via [https://huggingface.co/nvidia/corrdiff-cosmo-era5]
 **Network Architecture:** Diffusion Transformer (DiT) with axial 2D rotary
 position embeddings (RoPE), localized neighborhood attention (NATTEN, kernel size
 23), and a patch-size-2 tokenizer. The deterministic (mean) model is a regression
-DiT (~98M parameters) with a convolutional detokenizer head; the generative
-(ensemble) model is an EDM-preconditioned diffusion DiT (~174M parameters) with
+DiT (98M parameters) with a convolutional detokenizer head; the generative
+(ensemble) model is an EDM-preconditioned diffusion DiT (174M parameters) with
 the standard patch detokenizer, sampled with an 18-step deterministic (Heun)
 sampler. Both resolutions (REA6, REA2) share this architecture. Note: although
 part of the CorrDiff downscaling family, this model uses a Diffusion Transformer