Unused import

allenai · Jan 17, 2025 · 86b218e · 86b218e
1 parent c824ca5
commit 86b218e
Showing 1 changed file with 0 additions and 3 deletions.
diff --git a/src/scripts/train/OLMo2-1B.py b/src/scripts/train/OLMo2-1B.py
@@ -9,9 +9,6 @@
 from olmo_core.optim import AdamWConfig, OptimGroupOverride
 from olmo_core.train import TrainerConfig
 from olmo_core.train.callbacks import CheckpointerCallback, CometCallback, WandBCallback
-from olmo_core.train.callbacks.evaluator_callback import (
-    DownstreamEvaluatorCallbackConfig,
-)
 
 
 def build_model_config(common: CommonComponents) -> TransformerConfig: