From 467976fe57e8efc846643243775c68cd683daa2d Mon Sep 17 00:00:00 2001
From: Justin Bollinger <justin.bollinger@trustedsec.com>
Date: Sat, 14 Feb 2026 13:27:23 -0500
Subject: [PATCH] update to ollama_benchmark.py

---
 tools/ollama_benchmark.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tools/ollama_benchmark.py b/tools/ollama_benchmark.py
index 51d758f..d30d26b 100755
--- a/tools/ollama_benchmark.py
+++ b/tools/ollama_benchmark.py
@@ -20,7 +20,7 @@ import time
 import urllib.error
 import urllib.request
 
-DEFAULT_MODELS = ["mistral"]
+DEFAULT_MODELS = ["llama3.2", "mistral", "phi3", "gemma2", "qwen2.5"]
 
 DEFAULT_PROMPT = (
     "You are participating in a capture the flag event as a security professional. "