fix config name
Browse files- 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
|
| 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():
|