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"
|