diff --git a/rslearn/models/olmoearth_pretrain/model.py b/rslearn/models/olmoearth_pretrain/model.py index 002e0f03..b191db51 100644 --- a/rslearn/models/olmoearth_pretrain/model.py +++ b/rslearn/models/olmoearth_pretrain/model.py @@ -40,6 +40,7 @@ ModelID.OLMOEARTH_V1_NANO: 128, ModelID.OLMOEARTH_V1_TINY: 192, ModelID.OLMOEARTH_V1_BASE: 768, + ModelID.OLMOEARTH_V1_LARGE: 1024, }