Andrei Panferov
commited on
Commit
·
c0d7cc2
1
Parent(s):
c43662f
correct import
Browse files- modeling_llama.py +1 -1
modeling_llama.py
CHANGED
@@ -42,7 +42,7 @@ from transformers.utils import (
|
|
42 |
replace_return_docstrings,
|
43 |
)
|
44 |
|
45 |
-
from configuration_llama import LlamaConfig
|
46 |
from src.inference import FinalizedQuantizedLinear
|
47 |
|
48 |
if is_flash_attn_available():
|
|
|
42 |
replace_return_docstrings,
|
43 |
)
|
44 |
|
45 |
+
from .configuration_llama import LlamaConfig
|
46 |
from src.inference import FinalizedQuantizedLinear
|
47 |
|
48 |
if is_flash_attn_available():
|