diff --git a/scrapegraphai/helpers/models_tokens.py b/scrapegraphai/helpers/models_tokens.py
index 0d18991c..852a9ce2 100644
--- 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,