fix model
Browse files- modeling_ernie4_5.py +1 -1
modeling_ernie4_5.py
CHANGED
|
@@ -680,7 +680,7 @@ class Ernie4_5_PretrainedModel(PreTrainedModel):
|
|
| 680 |
"""Base class for ERNIE pretrained models."""
|
| 681 |
|
| 682 |
config_class = Ernie4_5_Config
|
| 683 |
-
base_model_prefix = "
|
| 684 |
|
| 685 |
|
| 686 |
class Ernie4_5_Model(Ernie4_5_PretrainedModel):
|
|
|
|
| 680 |
"""Base class for ERNIE pretrained models."""
|
| 681 |
|
| 682 |
config_class = Ernie4_5_Config
|
| 683 |
+
base_model_prefix = "model"
|
| 684 |
|
| 685 |
|
| 686 |
class Ernie4_5_Model(Ernie4_5_PretrainedModel):
|