同步千帆的模型定义

silvbb · Jul 3, 2024 · 455f68c · 455f68c
1 parent afc752d
commit 455f68c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/utils/gpt_model/my_wenxinworkshop.py b/utils/gpt_model/my_wenxinworkshop.py
@@ -28,7 +28,7 @@ def __init__(self, data):
                     "ERNIEBot_4_0": LLMAPI.ERNIEBot_4_0,
                     "ERNIE_SPEED_128K": LLMAPI.ERNIEBot_4_0,
                     "ERNIE_SPEED_8K": LLMAPI.ERNIE_SPEED_8K,
-                    "ERNIE_LITE_8K_0308": LLMAPI.ERNIE_LITE_8K_0308,
+                    "ERNIE_LITE_8K": LLMAPI.ERNIE_LITE_8K,
                     "ERNIE_LITE_8K_0922": LLMAPI.ERNIE_LITE_8K_0922,
                     "ERNIE_TINY_8K": LLMAPI.ERNIE_TINY_8K,
                     "BLOOMZ_7B": LLMAPI.BLOOMZ_7B,

diff --git a/webui.py b/webui.py
@@ -4170,7 +4170,7 @@ def save_config():
                             "ERNIEBot_4_0",
                             "ERNIE_SPEED_128K",
                             "ERNIE_SPEED_8K",
-                            "ERNIE_LITE_8K_0308",
+                            "ERNIE_LITE_8K",
                             "ERNIE_LITE_8K_0922",
                             "ERNIE_TINY_8K",
                             "BLOOMZ_7B",