jon-tow commited on
Commit
e20b0d6
1 Parent(s): 756dc76

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
@@ -544,7 +544,7 @@ class StableLMEpochPreTrainedModel(PreTrainedModel):
544
  """
545
 
546
  config_class = StableLMEpochConfig
547
- base_model_prefix = "transformer"
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"