Hugging Face's logo

mudhil
/
diffusion-downscaling-checkpoints

PyTorch
climate
downscaling
diffusion
Model card Files
xet
 Community
diffusion-downscaling-checkpoints / README.md
mudhil's picture
mudhil
Update README.md
c19d4bd verified
preview code
|
Raw History Blame Contribute Delete
820 Bytes
metadata 
license: mit
tags:
  - climate
  - downscaling
  - diffusion
  - pytorch

ERA5→CONUS404 Diffusion Downscaling — Checkpoint Archive

Source code: https://github.com/milhud/diffusion_downscaling_model

This repo holds a flat archive of checkpoints for the three-stage "Latent CorrDiff" downscaling pipeline (DRN → VAE → latent diffusion; see the repo README/docs/ARCHITECTURE.md for the full model description).

For the current, most up-to-date diffusion checkpoint, use mudhil/era5-conus404-diffusion-downscaling instead — this repo's diffusion_best.pt/diffusion_latest.pt are an older diffusion checkpoint (confirmed by sha256, different from the current local weights). The DRN and VAE files in this repo are current.