q-future
/

one-align

Zero-Shot Image Classification

feature-extraction

Model card Files Files and versions Community

teowu commited on Jan 15

Commit

15f7bb4

•

1 Parent(s): 7817ae3

Update modeling_llama2.py

Files changed (1) hide show

modeling_llama2.py +1 -1

modeling_llama2.py CHANGED Viewed

@@ -22,7 +22,7 @@ from transformers.models.llama.modeling_llama import *
 from transformers.configuration_utils import PretrainedConfig
 from transformers.utils import logging
-from .modeling_attn_mask_utils import _prepare_4d_causal_attention_mask
 from .configuration_mplug_owl2 import LlamaConfig
 class MultiwayNetwork(nn.Module):

 from transformers.configuration_utils import PretrainedConfig
 from transformers.utils import logging
+from .modeling_attn_mask_utils import _prepare_4d_causal_attention_mask, _prepare_4d_causal_attention_mask_for_sdpa
 from .configuration_mplug_owl2 import LlamaConfig
 class MultiwayNetwork(nn.Module):