happyz
/
gemma.cpp
mirror of https://github.com/google/gemma.cpp.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Internal change. 

PiperOrigin-RevId: 726947349
This commit is contained in:
Phil Culliton 2025-02-14 08:59:08 -08:00 committed by Copybara-Service
parent d854471ae2
commit ad8dd21e1d
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
gemma/configs.h
View File

@ -236,6 +236,7 @@ struct VitConfig : public IFields {
visitor(patch_width); visitor(patch_width);
visitor(image_size); visitor(image_size);
visitor(layer_configs); visitor(layer_configs);
visitor(pool_dim);
} }
uint32_t model_dim = 0; uint32_t model_dim = 0;
@ -243,6 +244,8 @@ struct VitConfig : public IFields {
uint32_t num_scales = 0; uint32_t num_scales = 0;
uint32_t patch_width = 14; uint32_t patch_width = 14;
uint32_t image_size = 224; uint32_t image_size = 224;
// Average pooling; 1 for no pooling.
uint32_t pool_dim = 1;
std::vector<LayerConfig> layer_configs; std::vector<LayerConfig> layer_configs;
}; };