From 972ed965ad44ee583021fddc4e018ad2b0feb6f8 Mon Sep 17 00:00:00 2001
From: Anna Pfohl <anna.pfohl@databricks.com>
Date: Mon, 27 Nov 2023 16:36:44 -0800
Subject: [PATCH] remove circular dependency

---
 llmfoundry/utils/builders.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/llmfoundry/utils/builders.py b/llmfoundry/utils/builders.py
index 793ae8c99f..21913a6cdd 100644
--- a/llmfoundry/utils/builders.py
+++ b/llmfoundry/utils/builders.py
@@ -27,7 +27,6 @@
 from torch.optim.optimizer import Optimizer
 from transformers import AutoTokenizer, PreTrainedTokenizerBase
 
-from llmfoundry import ComposerHFCausalLM
 from llmfoundry.callbacks import (EvalGauntlet, FDiffMetrics, GlobalLRScaling,
                                   HuggingFaceCheckpointer, LayerFreezing,
                                   MonolithicCheckpointSaver,
@@ -43,7 +42,7 @@
 
 def build_eval_loader(
     eval_loader_config: Union[DictConfig, ListConfig],
-    model: Union[Any, ComposerHFCausalLM],
+    model: Any,
     tokenizer: PreTrainedTokenizerBase,
     device_eval_batch_size: int,
 ) -> List[Evaluator]: