stefan-insilico commited on
Commit
8009c4a
·
verified ·
1 Parent(s): 3158d44

Update precious3_gpt_multi_modal.py

Browse files
Files changed (1) hide show
  1. precious3_gpt_multi_modal.py +7 -7
precious3_gpt_multi_modal.py CHANGED
@@ -14,13 +14,13 @@ import os
14
  import torch.nn.functional as F
15
 
16
  import sys
17
- sys.path.append("./mpt_7b/")
18
- from modeling_mpt import MPTModel, MPTForCausalLM, gen_attention_mask_in_length
19
- from configuration_mpt import MPTConfig
20
- from blocks import MPTBlock
21
- from norm import NORM_CLASS_REGISTRY
22
- from custom_embedding import SharedEmbedding
23
- from attention import ATTN_CLASS_REGISTRY, attn_bias_shape, build_attn_bias, gen_slopes
24
 
25
  import logging
26
  log = logging.getLogger(__name__)
 
14
  import torch.nn.functional as F
15
 
16
  import sys
17
+ sys.path.append("./mpt_7b")
18
+ from mpt_7b.modeling_mpt import MPTModel, MPTForCausalLM, gen_attention_mask_in_length
19
+ from mpt_7b.configuration_mpt import MPTConfig
20
+ from mpt_7b.blocks import MPTBlock
21
+ from mpt_7b.norm import NORM_CLASS_REGISTRY
22
+ from mpt_7b.custom_embedding import SharedEmbedding
23
+ from mpt_7b.attention import ATTN_CLASS_REGISTRY, attn_bias_shape, build_attn_bias, gen_slopes
24
 
25
  import logging
26
  log = logging.getLogger(__name__)