CodeCow/Scrapegraph-ai
1
0
Fork 0
mirror of https://github.com/VinciGit00/Scrapegraph-ai.git synced 2026-07-06 21:11:37 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

chore(models_tokens): add mistral models

Browse Source
This commit is contained in:
Federico Aguzzi 2024-08-06 17:22:39 +02:00
parent 8eb66f6e22
commit 5e824327c3
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
scrapegraphai/helpers/models_tokens.py
View File

@ -146,7 +146,14 @@ models_tokens = {
"cohere.embed-multilingual-v3": 512 "cohere.embed-multilingual-v3": 512
}, },
"mistral": { "mistral": {
"mistralai/Mistral-7B-Instruct-v0.2": 32000 "mistral-large-latest": 128000,
"open-mistral-nemo": 128000,
"codestral-latest": 32000,
"mistral-embed": 8000,
"open-mistral-7b": 32000,
"open-mixtral-8x7b": 32000,
"open-mixtral-8x22b": 64000,
"open-codestral-mamba": 256000
}, },
"hugging_face": { "hugging_face": {
"xai-org/grok-1": 8192, "xai-org/grok-1": 8192,