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