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