Update precious3_gpt_multi_modal.py

precious3_gpt_multi_modal.py

@@ -13,14 +13,12 @@ from transformers import PreTrainedTokenizerFast
 import os
 import torch.nn.functional as F
 
-import sys
-sys.path.append("./mpt_7b")
-from mpt_7b.modeling_mpt import MPTModel, MPTForCausalLM, gen_attention_mask_in_length
-from mpt_7b.configuration_mpt import MPTConfig
-from mpt_7b.blocks import MPTBlock
-from mpt_7b.norm import NORM_CLASS_REGISTRY
-from mpt_7b.custom_embedding import SharedEmbedding
-from mpt_7b.attention import ATTN_CLASS_REGISTRY, attn_bias_shape, build_attn_bias, gen_slopes
+from modeling_mpt import MPTModel, MPTForCausalLM, gen_attention_mask_in_length
+from configuration_mpt import MPTConfig
+from blocks import MPTBlock
+from norm import NORM_CLASS_REGISTRY
+from custom_embedding import SharedEmbedding
+from attention import ATTN_CLASS_REGISTRY, attn_bias_shape, build_attn_bias, gen_slopes
 
 import logging
 log = logging.getLogger(__name__)