Skip to content

Commit dcd216e

Browse files
fix: removed duplicate from ollama dictionary
Co-Authored-By: Matteo Vedovati <[email protected]>
1 parent c44b701 commit dcd216e

File tree

1 file changed

+27
-44
lines changed

1 file changed

+27
-44
lines changed

scrapegraphai/helpers/models_tokens.py

Lines changed: 27 additions & 44 deletions
Original file line numberDiff line numberDiff line change
@@ -35,50 +35,33 @@
3535
"gemini-1.5-pro-latest":128000,
3636
"models/embedding-001": 2048
3737
},
38-
39-
"ollama": {
40-
"command-r": 12800,
41-
"command-r-plus": 12800,
42-
"codellama": 16000,
43-
"dbrx": 32768,
44-
"dbrx:instruct": 32768,
45-
"deepseek-coder:33b": 16000,
46-
"dolphin-mixtral": 32000,
47-
"llama2": 4096,
48-
"llama3": 8192,
49-
"llama3:70b-instruct": 8192,
50-
"llava": 4096,
51-
"llava:34b": 4096,
52-
"llava_next": 4096,
53-
"mistral": 8192,
54-
"falcon": 2048,
55-
"codellama": 16000,
56-
"dolphin-mixtral": 32000,
57-
"mistral-openorca": 32000,
58-
"stablelm-zephyr": 8192,
59-
"command-r-plus": 12800,
60-
"command-r": 12800,
61-
"mistral:7b-instruct": 32768,
62-
"mistral-openorca": 32000,
63-
"mixtral:8x22b-instruct": 65536,
64-
"nous-hermes2:34b": 4096,
65-
"orca-mini": 2048,
66-
"phi3:3.8b": 12800,
67-
"phi3:14b": 12800,
68-
"qwen:0.5b": 32000,
69-
"qwen:1.8b": 32000,
70-
"qwen:4b": 32000,
71-
"qwen:14b": 32000,
72-
"qwen:32b": 32000,
73-
"qwen:72b": 32000,
74-
"qwen:110b": 32000,
75-
"stablelm-zephyr": 8192,
76-
"wizardlm2:8x22b": 65536,
77-
# embedding models
78-
"nomic-embed-text": 8192,
79-
"snowflake-arctic-embed:335m": 8192,
80-
"snowflake-arctic-embed:l": 8192,
81-
"mxbai-embed-large": 512,
38+
"ollama": { "command-r": 12800,
39+
"codellama": 16000,
40+
"dbrx": 32768,
41+
"deepseek-coder:33b": 16000,
42+
"falcon": 2048,
43+
"llama2": 4096,
44+
"llama3": 8192,
45+
"scrapegraph": 8192,
46+
"llava": 4096,
47+
"mixtral:8x22b-instruct": 65536,
48+
"mistral-openorca": 32000,
49+
"nomic-embed-text": 8192,
50+
"nous-hermes2:34b": 4096,
51+
"orca-mini": 2048,
52+
"phi3:3.8b": 12800,
53+
"qwen:0.5b": 32000,
54+
"qwen:1.8b": 32000,
55+
"qwen:4b": 32000,
56+
"qwen:14b": 32000,
57+
"qwen:32b": 32000,
58+
"qwen:72b": 32000,
59+
"qwen:110b": 32000,
60+
"stablelm-zephyr": 8192,
61+
"wizardlm2:8x22b": 65536,
62+
# embedding models
63+
"snowflake-arctic-embed": 8192,
64+
"mxbai-embed-large": 512
8265
},
8366
"oneapi": {
8467
"qwen-turbo": 16380

0 commit comments

Comments
 (0)