From 039fe3c6d91978f70baedfef407bda912a285aed Mon Sep 17 00:00:00 2001 From: Marco Vinciguerra Date: Wed, 11 Sep 2024 11:45:44 +0200 Subject: [PATCH] fix: models tokens --- scrapegraphai/helpers/models_tokens.py | 1 + 1 file changed, 1 insertion(+) diff --git a/scrapegraphai/helpers/models_tokens.py b/scrapegraphai/helpers/models_tokens.py index 514b3357..ea35ed71 100644 --- a/scrapegraphai/helpers/models_tokens.py +++ b/scrapegraphai/helpers/models_tokens.py @@ -154,6 +154,7 @@ models_tokens = { "mistral.mistral-7b-instruct-v0:2": 32768, "mistral.mixtral-8x7b-instruct-v0:1": 32768, "mistral.mistral-large-2402-v1:0": 32768, + "mistral.mistral-small-2402-v1:0": 32768, "amazon.titan-embed-text-v1": 8000, "amazon.titan-embed-text-v2:0": 8000, "cohere.embed-english-v3": 512,