Skip to content

Commit d9752b1

Browse files
committed
chore: update models_tokens.py with new model configurations
1 parent fc56d6b commit d9752b1

File tree

1 file changed

+15
-2
lines changed

1 file changed

+15
-2
lines changed

scrapegraphai/helpers/models_tokens.py

Lines changed: 15 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
"""
2-
Models token
2+
Models token
33
"""
44

55
models_tokens = {
@@ -39,7 +39,20 @@
3939
"dolphin-mixtral": 32000,
4040
"mistral-openorca": 32000,
4141
"stablelm-zephyr": 8192,
42-
"nomic-embed-text": 8192
42+
"command-r-plus": 12800,
43+
"command-r": 12800,
44+
"mistral:7b-instruct": 32768,
45+
"llama3:70b-instruct": 8192,
46+
"mixtral:8x22b-instruct": 65536,
47+
"wizardlm2:8x22b": 65536,
48+
"dbrx": 32768,
49+
"dbrx:instruct": 32768,
50+
"nous-hermes2:34b": 4096,
51+
# embedding models
52+
"nomic-embed-text": 8192,
53+
"snowflake-arctic-embed:335m": 8192,
54+
"snowflake-arctic-embed:l": 8192,
55+
"mxbai-embed-large": 512,
4356
},
4457
"groq": {
4558
"llama3-8b-8192": 8192,

0 commit comments

Comments
 (0)