From ef9638811b2d90a602643002ad8a2caceac4403f Mon Sep 17 00:00:00 2001 From: Baudouin Raoult Date: Fri, 30 Aug 2024 09:05:10 +0000 Subject: [PATCH] default to highres --- src/ai_models_aurora/model.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/ai_models_aurora/model.py b/src/ai_models_aurora/model.py index b37841a..612b19f 100644 --- a/src/ai_models_aurora/model.py +++ b/src/ai_models_aurora/model.py @@ -202,10 +202,10 @@ class Aurora0p1(AuroraModel): # https://microsoft.github.io/aurora/models.html#aurora-0-1-fine-tuned -class Aurora0p1FineTuned(UseIFSMixin, Aurora0p1): +class Aurora0p1FineTuned(Aurora0p1): klass = AuroraHighRes checkpoint = "aurora-0.1-finetuned.ckpt" -model = Aurora2p5FineTuned +model = Aurora0p1FineTuned