fix

allenai · Nov 27, 2024 · 2d5cd3e · 2d5cd3e
1 parent 3ce2f25
commit 2d5cd3e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/scripts/train/OLMo2-ladder.py b/src/scripts/train/OLMo2-ladder.py
@@ -36,7 +36,7 @@ class BaselineModelLadder(ModelLadder):
     }
 
     def get_model_config(self, *, size: ModelSize) -> TransformerConfig:
-        return getattr(TransformerConfig, f"olmo_{size}")(
+        return getattr(TransformerConfig, f"olmo2_{size}")(
             vocab_size=self.tokenizer.padded_vocab_size(),
             init_seed=self.init_seed,
             compile=True,