diff --git a/scrapegraphai/helpers/models_tokens.py b/scrapegraphai/helpers/models_tokens.py index 43598785..510777fe 100644 --- a/scrapegraphai/helpers/models_tokens.py +++ b/scrapegraphai/helpers/models_tokens.py @@ -154,5 +154,15 @@ "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, } }