Andrei Panferov commited on
Commit
c0d7cc2
·
1 Parent(s): c43662f

correct import

Browse files
Files changed (1) hide show
  1. 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():