Graph Machine Learning
AnemoI
English

Update checkpoint to latest anemoi-* versions

#11
by rodrigoalmeida1994 - opened

Hello!

I have been implementing running AIFS-ENS (and AIFS-Single) in earth2studio https://github.com/NVIDIA/earth2studio/pull/483 - and have found the versions required to run the models with this checkpoint to be quite restrictive (and also apparently non-compatible with AIFS-Single). Is there any chance you could release a new checkpoint that makes use of the latest anemoi package versions? Or a compatible package version to the AIFS-Single checkpoint?

Thanks

Hello Rodrigo! good to hear from you! The Single and ENS cycle implementation are so far not synced so it's likely these two models use slightly different versions. So far we don't have plants to update those checkpoints with the latest main. Could you extend on why would you need the two models to use the same version of anemoi models ?

Hi Ana, nice to hear from you too! I see - it would make the dependency management easier, and enable the same environment to run both AIFS-Single and AIFS-ENS. In the earth2studio package (which integrates a long list of weather models) there are a lot of dependencies already - I think I can work around this in any case.

Sign up or log in to comment