From 1909ae19b3adb0a79d1ed313154cb67afe35d587 Mon Sep 17 00:00:00 2001
From: Jonathan Tow <jon-tow@users.noreply.huggingface.co>
Date: Mon, 29 Jan 2024 19:49:03 +0000
Subject: [PATCH] fix(modeling): use correct `base_model_prefix` name

---
 modeling_stablelm_epoch.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/modeling_stablelm_epoch.py b/modeling_stablelm_epoch.py
index e7fde9f..b179cfb 100755
--- a/modeling_stablelm_epoch.py
+++ b/modeling_stablelm_epoch.py
@@ -545,7 +545,7 @@ class StableLMEpochPreTrainedModel(PreTrainedModel):
     """
 
     config_class = StableLMEpochConfig
-    base_model_prefix = "transformer"
+    base_model_prefix = "model"
     supports_gradient_checkpointing = True
     _no_split_modules = ["DecoderLayer"]
     _skip_keys_device_placement = "past_key_values"