fix(models_tokens): add llama2 and llama3 sizes explicitly

2026-07-01 21:00:48 +08:00 · 2024-08-21 09:32:59 +02:00 · 2024-08-21 09:32:59 +02:00 · b05ec16b25
commit b05ec16b25
parent dd406dff0c
1 changed files with 4 additions and 0 deletions
--- a/scrapegraphai/helpers/models_tokens.py
+++ b/scrapegraphai/helpers/models_tokens.py
@ -58,7 +58,11 @@ models_tokens = {
            "deepseek-coder:33b": 16000, 
            "falcon": 2048,
            "llama2": 4096,
+            "llama2:7b": 4096,
+            "llama2:13b": 4096,
+            "llama2:70b": 4096,
            "llama3": 8192,
+            "llama3:8b": 8192, 
            "llama3:70b": 8192,
            "llama3.1":128000,
            "llama3.1:8b": 128000,