fyfyfyfy28 commited on
Commit
3da990e
1 Parent(s): c0e74da

Update modeling_mpt.py

Browse files

remove one line of import to evade a bug of bloom model

Files changed (1) hide show
  1. modeling_mpt.py +1 -1
modeling_mpt.py CHANGED
@@ -20,7 +20,7 @@ from .ffn import build_ffn as build_ffn
20
  from .norm import NORM_CLASS_REGISTRY
21
  from .configuration_mpt import MPTConfig
22
  from .adapt_tokenizer import AutoTokenizerForMOD, adapt_tokenizer_for_denoising
23
- from .hf_prefixlm_converter import add_bidirectional_mask_if_missing, convert_hf_causal_lm_to_prefix_lm
24
  from .meta_init_context import init_empty_weights
25
  from .param_init_fns import generic_param_init_fn_, MODEL_INIT_REGISTRY
26
  try:
 
20
  from .norm import NORM_CLASS_REGISTRY
21
  from .configuration_mpt import MPTConfig
22
  from .adapt_tokenizer import AutoTokenizerForMOD, adapt_tokenizer_for_denoising
23
+ # from .hf_prefixlm_converter import add_bidirectional_mask_if_missing, convert_hf_causal_lm_to_prefix_lm
24
  from .meta_init_context import init_empty_weights
25
  from .param_init_fns import generic_param_init_fn_, MODEL_INIT_REGISTRY
26
  try: