Skip to content

Commit

Permalink
default to highres
Browse files Browse the repository at this point in the history
  • Loading branch information
b8raoult committed Aug 30, 2024
1 parent 3cae85c commit ef96388
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions src/ai_models_aurora/model.py
Original file line number Diff line number Diff line change
Expand Up @@ -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

0 comments on commit ef96388

Please sign in to comment.