From 86b218e65ef18313ebb4070544c9e899f545988a Mon Sep 17 00:00:00 2001 From: Dirk Groeneveld Date: Fri, 17 Jan 2025 13:10:26 -0800 Subject: [PATCH] Unused import --- src/scripts/train/OLMo2-1B.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/src/scripts/train/OLMo2-1B.py b/src/scripts/train/OLMo2-1B.py index 52fc6a0c..1285df7c 100644 --- 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: