Update modeling_mplug_owl2_huggingface.py

modeling_mplug_owl2_huggingface.py

@@ -31,6 +31,7 @@ from transformers.modeling_outputs import CausalLMOutputWithPast
 
 from .configuration_mplug_owl2 import MPLUGOwl2Config, MplugOwlVisionConfig, MplugOwlVisualAbstractorConfig
 from .visual_encoder import MplugOwlVisionModel, MplugOwlVisualAbstractorModel
+from .modeling_attn_mask_utils import *
 from .modeling_llama2 import replace_llama_modality_adaptive
 IGNORE_INDEX = -100
 IMAGE_TOKEN_INDEX = -200