damerajee commited on
Commit
a064c2e
1 Parent(s): 3aa3114

Update modeling_Llamoe.py

Browse files
Files changed (1) hide show
  1. modeling_Llamoe.py +2 -2
modeling_Llamoe.py CHANGED
@@ -33,10 +33,10 @@ from transformers.utils import (
33
  replace_return_docstrings,
34
  )
35
  from transformers.utils.import_utils import is_torch_fx_available
36
- from .configuration_Llamoe import LlmoeConfig
37
 
38
  from math import sqrt as math_sqrt
39
- _CONFIG_FOR_DOC = "LlmoeConfig"
40
 
41
 
42
  if is_flash_attn_2_available():
 
33
  replace_return_docstrings,
34
  )
35
  from transformers.utils.import_utils import is_torch_fx_available
36
+ from .configuration_Llamoe import LlamoeConfig
37
 
38
  from math import sqrt as math_sqrt
39
+ _CONFIG_FOR_DOC = "LlamoeConfig"
40
 
41
 
42
  if is_flash_attn_2_available():