Internal change. Add deduction for 270M.

PiperOrigin-RevId: 795041810
This commit is contained in:
Phil Culliton 2025-08-14 08:04:10 -07:00 committed by Copybara-Service
parent d044801c1d
commit 78573b6718
1 changed files with 3 additions and 0 deletions

View File

@ -750,6 +750,9 @@ Model DeduceModel(const Path& blob_path, size_t layers, int layer_types) {
switch (layers) { switch (layers) {
case 2: case 2:
return Model::GEMMA_TINY; return Model::GEMMA_TINY;
case 18:
return Model::GEMMA3_270M;
case 26: case 26:
if (layer_types & kDeducedGriffin) return Model::GRIFFIN_2B; if (layer_types & kDeducedGriffin) return Model::GRIFFIN_2B;
if (layer_types & kDeducedViT) return Model::GEMMA3_1B; if (layer_types & kDeducedViT) return Model::GEMMA3_1B;