Update modeling_mpt.py

modeling_mpt.py

@@ -27,10 +27,10 @@ from llmfoundry.models.layers.custom_embedding import SharedEmbedding
 from llmfoundry.models.layers.norm import NORM_CLASS_REGISTRY
 from llmfoundry.models.utils.param_init_fns import MODEL_INIT_REGISTRY
 
-from extended_mind_transformers.mpt.configuration import ExtendedMPTConfig
-from extended_mind_transformers.mpt.attention import attn_bias_shape, build_attn_bias
-from extended_mind_transformers.mpt.blocks import MPTBlock
-from extended_mind_transformers.utils import instantiate_from_config
+from .mpt.configuration import ExtendedMPTConfig
+from .mpt.attention import attn_bias_shape, build_attn_bias
+from .mpt.blocks import MPTBlock
+from .utils import instantiate_from_config
 
 Tokenizer = Union[PreTrainedTokenizer, PreTrainedTokenizerFast]