teowu commited on
Commit
edd457f
1 Parent(s): 1c087ef

Update modeling_llama2.py

Browse files
Files changed (1) hide show
  1. modeling_llama2.py +1 -1
modeling_llama2.py CHANGED
@@ -16,7 +16,7 @@ import sys
16
  dir_path = os.path.dirname(os.path.realpath(__file__))
17
  sys.path.insert(0, dir_path)
18
 
19
- from .modeling_attn_mask_utils import _prepare_4d_causal_attention_mask, _prepare_4d_causal_attention_mask_for_sdpq
20
 
21
  import transformers
22
  from transformers.models.llama.modeling_llama import *
 
16
  dir_path = os.path.dirname(os.path.realpath(__file__))
17
  sys.path.insert(0, dir_path)
18
 
19
+ from .modeling_attn_mask_utils import _prepare_4d_causal_attention_mask, _prepare_4d_causal_attention_mask_for_sdpa
20
 
21
  import transformers
22
  from transformers.models.llama.modeling_llama import *