circular import

mosaicml · Oct 25, 2024 · 8ea0d99 · 8ea0d99
1 parent e1f61c5
commit 8ea0d99
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/llmfoundry/models/llm_embed/modeling_llm_embed.py b/llmfoundry/models/llm_embed/modeling_llm_embed.py
@@ -28,7 +28,7 @@
 from transformers import PreTrainedTokenizerBase
 from transformers.modeling_outputs import CausalLMOutputWithPast
 
-from llmfoundry import MPTConfig
+from llmfoundry.models.mpt.configuration_mpt import MPTConfig
 from llmfoundry.models.hf.hf_causal_lm import ComposerHFCausalLM
 from llmfoundry.models.mpt.modeling_mpt import MPTForCausalLM
 from llmfoundry.models.utils.config_moe_args import create_set_process_group