remove circular dependency

mosaicml · Nov 28, 2023 · 972ed96 · 972ed96
1 parent 1a0a3da
commit 972ed96
Showing 1 changed file with 1 addition and 2 deletions.
diff --git 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]: