jon-tow commited on
Commit
fa88b77
1 Parent(s): 9e73e07

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