try compile again

allenai · Dec 3, 2024 · 8ba7f58 · 8ba7f58
1 parent d552a3a
commit 8ba7f58
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/scripts/train/OLMo2-26B.py b/src/scripts/train/OLMo2-26B.py
@@ -22,7 +22,7 @@
 
 
 def build_model_config(common: CommonComponents) -> TransformerConfig:
-    compile = False
+    compile = True
     return TransformerConfig.olmo2_26B(
         vocab_size=common.tokenizer.padded_vocab_size(),
         compile=compile,