From ad8dd21e1d47df4ca235c7b555a27d65feacc75d Mon Sep 17 00:00:00 2001
From: Phil Culliton <philculliton@google.com>
Date: Fri, 14 Feb 2025 08:59:08 -0800
Subject: [PATCH] Internal change.

PiperOrigin-RevId: 726947349
---
 gemma/configs.h | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/gemma/configs.h b/gemma/configs.h
index 42693e6..25735d2 100644
--- a/gemma/configs.h
+++ b/gemma/configs.h
@@ -236,6 +236,7 @@ struct VitConfig : public IFields {
     visitor(patch_width);
     visitor(image_size);
     visitor(layer_configs);
+    visitor(pool_dim);
   }
 
   uint32_t model_dim = 0;
@@ -243,6 +244,8 @@ struct VitConfig : public IFields {
   uint32_t num_scales = 0;
   uint32_t patch_width = 14;
   uint32_t image_size = 224;
+  // Average pooling; 1 for no pooling.
+  uint32_t pool_dim = 1;
   std::vector<LayerConfig> layer_configs;
 };