Text Generation
Transformers
Safetensors
7 languages
stablelm
causal-lm
Inference Endpoints
12 papers
jon-tow commited on
Commit
1909ae1
1 Parent(s): 21ee10d

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"