Merge pull request #769 from ScrapeGraphAI/add-mistral-models

feat: add new mistral models
ScrapeGraphAI · Oct 27, 2024 · 58bb0dc · 58bb0dc
2 parents eee131e + 6914170
commit 58bb0dc
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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,