fix: removed duplicate from ollama dictionary

Co-Authored-By: Matteo Vedovati <[email protected]>
ScrapeGraphAI · Jun 15, 2024 · dcd216e · dcd216e
1 parent c44b701
commit dcd216e
Showing 1 changed file with 27 additions and 44 deletions.
diff --git a/scrapegraphai/helpers/models_tokens.py b/scrapegraphai/helpers/models_tokens.py
@@ -35,50 +35,33 @@
         "gemini-1.5-pro-latest":128000,
         "models/embedding-001": 2048
     },
-
-    "ollama": {
-        "command-r": 12800,
-        "command-r-plus": 12800,
-        "codellama": 16000,
-        "dbrx": 32768,
-        "dbrx:instruct": 32768,
-        "deepseek-coder:33b": 16000,
-        "dolphin-mixtral": 32000,
-        "llama2": 4096,
-        "llama3": 8192,
-        "llama3:70b-instruct": 8192,
-        "llava": 4096,
-        "llava:34b": 4096,
-        "llava_next": 4096,
-        "mistral": 8192,
-        "falcon": 2048,
-        "codellama": 16000,
-        "dolphin-mixtral": 32000,
-        "mistral-openorca": 32000,
-        "stablelm-zephyr": 8192,
-        "command-r-plus": 12800,
-        "command-r": 12800,
-        "mistral:7b-instruct": 32768,
-        "mistral-openorca": 32000,
-        "mixtral:8x22b-instruct": 65536,
-        "nous-hermes2:34b": 4096,
-        "orca-mini": 2048,
-        "phi3:3.8b": 12800,
-        "phi3:14b": 12800,
-        "qwen:0.5b": 32000,
-        "qwen:1.8b": 32000,
-        "qwen:4b": 32000,
-        "qwen:14b": 32000,
-        "qwen:32b": 32000,
-        "qwen:72b": 32000,
-        "qwen:110b": 32000,
-        "stablelm-zephyr": 8192,
-        "wizardlm2:8x22b": 65536,
-        # embedding models
-        "nomic-embed-text": 8192,
-        "snowflake-arctic-embed:335m": 8192,
-        "snowflake-arctic-embed:l": 8192,
-        "mxbai-embed-large": 512,
+    "ollama": { "command-r": 12800, 
+               "codellama": 16000, 
+               "dbrx": 32768, 
+               "deepseek-coder:33b": 16000, 
+               "falcon": 2048, 
+               "llama2": 4096, 
+               "llama3": 8192, 
+               "scrapegraph": 8192, 
+               "llava": 4096, 
+               "mixtral:8x22b-instruct": 65536, 
+               "mistral-openorca": 32000, 
+               "nomic-embed-text": 8192, 
+               "nous-hermes2:34b": 4096, 
+               "orca-mini": 2048, 
+               "phi3:3.8b": 12800, 
+               "qwen:0.5b": 32000, 
+               "qwen:1.8b": 32000, 
+               "qwen:4b": 32000, 
+               "qwen:14b": 32000, 
+               "qwen:32b": 32000, 
+               "qwen:72b": 32000, 
+               "qwen:110b": 32000, 
+               "stablelm-zephyr": 8192, 
+               "wizardlm2:8x22b": 65536, 
+               # embedding models 
+               "snowflake-arctic-embed": 8192, 
+               "mxbai-embed-large": 512 
     },
     "oneapi": {
         "qwen-turbo": 16380