q-future
/

co-instruct

Image-Text-to-Text

feature-extraction

Model card Files Files and versions

teowu commited on Jan 15, 2024

Commit

edd457f

·

verified ·

1 Parent(s): 1c087ef

Update modeling_llama2.py

Files changed (1) hide show

modeling_llama2.py +1 -1

modeling_llama2.py CHANGED Viewed

@@ -16,7 +16,7 @@ import sys
 dir_path = os.path.dirname(os.path.realpath(__file__))
 sys.path.insert(0, dir_path)
-from .modeling_attn_mask_utils import _prepare_4d_causal_attention_mask, _prepare_4d_causal_attention_mask_for_sdpq
 import transformers
 from transformers.models.llama.modeling_llama import *

 dir_path = os.path.dirname(os.path.realpath(__file__))
 sys.path.insert(0, dir_path)
+from .modeling_attn_mask_utils import _prepare_4d_causal_attention_mask, _prepare_4d_causal_attention_mask_for_sdpa
 import transformers
 from transformers.models.llama.modeling_llama import *