Update modeling_mpt.py

modeling_mpt.py

@@ -27,9 +27,9 @@ 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 .mpt.configuration import ExtendedMPTConfig
-from .mpt.attention import attn_bias_shape, build_attn_bias
-from .mpt.blocks import MPTBlock
+from .configuration import ExtendedMPTConfig
+from .attention import attn_bias_shape, build_attn_bias
+from .blocks import MPTBlock
 from .utils import instantiate_from_config
 
 Tokenizer = Union[PreTrainedTokenizer, PreTrainedTokenizerFast]