From 691417089014b5b0b64a1b26687cbb0cba693952 Mon Sep 17 00:00:00 2001 From: Marco Vinciguerra Date: Sat, 26 Oct 2024 16:07:43 +0200 Subject: [PATCH] feat: add new mistral models --- scrapegraphai/helpers/models_tokens.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/scrapegraphai/helpers/models_tokens.py b/scrapegraphai/helpers/models_tokens.py index 113d1636..8f367ceb 100644 --- a/scrapegraphai/helpers/models_tokens.py +++ b/scrapegraphai/helpers/models_tokens.py @@ -80,10 +80,12 @@ "llama3.2:1b": 128000, "scrapegraph": 8192, "mistral": 8192, + "mistral-small": 128000, + "mistral-openorca": 32000, + "mistral-large": 128000, "grok-1": 8192, "llava": 4096, "mixtral:8x22b-instruct": 65536, - "mistral-openorca": 32000, "nomic-embed-text": 8192, "nous-hermes2:34b": 4096, "orca-mini": 2048,