Update modeling_llama2.py
Browse files- 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,
|
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 *
|