not-lain commited on
Commit
1288f9a
1 Parent(s): 71e2f68

fix config name

Browse files
Files changed (1) hide show
  1. modeling_cerule_gemma.py +1 -1
modeling_cerule_gemma.py CHANGED
@@ -853,7 +853,7 @@ from transformers.utils import (
853
  replace_return_docstrings,
854
  )
855
  from transformers.utils.import_utils import is_torch_fx_available
856
- from .configuration_gemma import GemmaConfig
857
 
858
 
859
  if is_flash_attn_2_available():
 
853
  replace_return_docstrings,
854
  )
855
  from transformers.utils.import_utils import is_torch_fx_available
856
+ from .configuration_gemma import CeruleGemmaConfig
857
 
858
 
859
  if is_flash_attn_2_available():