jon-tow commited on
Commit
7c4583d
1 Parent(s): c77592f

fix(modeling): use correct `base_model_prefix` name

Browse files
Files changed (1) hide show
  1. modeling_stablelm_epoch.py +1 -1
modeling_stablelm_epoch.py CHANGED
@@ -542,7 +542,7 @@ class StableLMEpochPreTrainedModel(PreTrainedModel):
542
  """
543
 
544
  config_class = StableLMEpochConfig
545
- base_model_prefix = "transformer"
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"