From b1032ebf5f951e13651cac57a83636319ce21ff0 Mon Sep 17 00:00:00 2001
From: Jan Wassenberg <janwas@google.com>
Date: Thu, 20 Mar 2025 05:06:45 -0700
Subject: [PATCH] Fix PromptWrapping for gemma3 1B, thanks @ufownl

PiperOrigin-RevId: 738759611
---
 gemma/common.cc | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/gemma/common.cc b/gemma/common.cc
index 0d8977b..17483d9 100644
--- a/gemma/common.cc
+++ b/gemma/common.cc
@@ -26,7 +26,6 @@
 
 #include "compression/shared.h"
 #include "hwy/base.h"
-#include "hwy/contrib/thread_pool/thread_pool.h"
 
 namespace gcpp {
 
@@ -80,7 +79,7 @@ constexpr PromptWrapping kPromptWrapping[] = {
     PromptWrapping::PALIGEMMA, PromptWrapping::PALIGEMMA,  // PG2 3B 224/448
     PromptWrapping::PALIGEMMA, PromptWrapping::PALIGEMMA,  // PG2 10B 224/448
     PromptWrapping::GEMMA_VLM,                             // Gemma3 4B
-    PromptWrapping::GEMMA_PT,                              // Gemma3 1B
+    PromptWrapping::GEMMA_IT,                              // Gemma3 1B
     PromptWrapping::GEMMA_VLM,                             // Gemma3 12B
     PromptWrapping::GEMMA_VLM,                             // Gemma3 27B
 };