precommit

mosaicml · Oct 30, 2023 · fe05f6e · fe05f6e
1 parent a5fe55b
commit fe05f6e
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/tests/test_model.py b/tests/test_model.py
@@ -25,8 +25,7 @@
 from transformers.modeling_outputs import CausalLMOutputWithPast
 from transformers.models.bloom.modeling_bloom import build_alibi_tensor
 
-from llmfoundry import (COMPOSER_MODEL_REGISTRY, ComposerHFCausalLM,
-                        ComposerHFPrefixLM)
+from llmfoundry import COMPOSER_MODEL_REGISTRY
 from llmfoundry.models.hf.model_wrapper import HuggingFaceModelWithZLoss
 from llmfoundry.models.layers import NORM_CLASS_REGISTRY, build_alibi_bias
 from llmfoundry.models.layers.blocks import MPTBlock