Update modeling_llama2.py
Browse files- modeling_llama2.py +2 -0
modeling_llama2.py
CHANGED
|
@@ -16,6 +16,8 @@ import sys
|
|
| 16 |
dir_path = os.path.dirname(os.path.realpath(__file__))
|
| 17 |
sys.path.insert(0, dir_path)
|
| 18 |
|
|
|
|
|
|
|
| 19 |
import transformers
|
| 20 |
from transformers.models.llama.modeling_llama import *
|
| 21 |
|
|
|
|
| 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 *
|
| 23 |
|