CorrDiff model for Continental US

[gowtham-talluru]

Hi Team

Is trained CorrDiff model available for continental US?

Thanks
Gowtham

[NickGeneva]

Hi @gowtham-talluru

Not that is a part of Earth2Studio right as this moment. There is a checkpoint that us provided as part the the NIM developer program / Nvidia AIE but its not integrated with the package yet.

The corrdiff code base has had a lot of improvements, so we are waiting for that to be fully complete before any assessments of adding a support for it in E2S with the improved inference code. PhysicsNeMo has the recipes for training. Thanks!