Merge pull request #347 from duke147/ernie

add the Ernie context window
2026-06-23 21:00:30 +08:00 · 2024-06-05 13:22:26 +02:00 · 2024-06-05 13:22:26 +02:00 · 9ef73d7363
commit 9ef73d7363
parent 95725789ff 1a404e358d
1 changed files with 10 additions and 0 deletions
--- a/scrapegraphai/helpers/models_tokens.py
+++ b/scrapegraphai/helpers/models_tokens.py
@ -154,5 +154,15 @@ models_tokens = {
    "deepseek": {
        "deepseek-chat": 32768,
        "deepseek-coder": 16384
+    },
+    "ernie": {
+        "ernie-bot-turbo": 4096,
+        "ernie-bot": 4096,
+        "ernie-bot-2": 4096,
+        "ernie-bot-2-base": 4096,
+        "ernie-bot-2-base-zh": 4096,
+        "ernie-bot-2-base-en": 4096,
+        "ernie-bot-2-base-en-zh": 4096,
+        "ernie-bot-2-base-zh-en": 4096,
    }
 }