fix(modeling): use correct `base_model_prefix` name
Browse files
modeling_stablelm_epoch.py
CHANGED
|
@@ -542,7 +542,7 @@ class StableLMEpochPreTrainedModel(PreTrainedModel):
|
|
| 542 |
"""
|
| 543 |
|
| 544 |
config_class = StableLMEpochConfig
|
| 545 |
-
base_model_prefix = "
|
| 546 |
supports_gradient_checkpointing = True
|
| 547 |
_no_split_modules = ["DecoderLayer"]
|
| 548 |
_skip_keys_device_placement = "past_key_values"
|
|
|
|
| 542 |
"""
|
| 543 |
|
| 544 |
config_class = StableLMEpochConfig
|
| 545 |
+
base_model_prefix = "model"
|
| 546 |
supports_gradient_checkpointing = True
|
| 547 |
_no_split_modules = ["DecoderLayer"]
|
| 548 |
_skip_keys_device_placement = "past_key_values"
|