This commit is contained in:
Younes Beriane 2026-10-12 16:56:15 +05:30 • committed by GitHub
commit f138046504
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 12 additions and 2 deletions

View file

@ -80,6 +80,15 @@ func TestResolveGemma4Renderer(t *testing.T) {
},
want: gemma4RendererLarge,
},
{
name: "legacy 11.9B model type with size-free name resolves large",
model: &Model{
Name: "my-gemma",
ShortName: "my-gemma",
Config: testConfigWithRendererAndType(gemma4RendererLegacy, "11.9B"),
},
want: gemma4RendererLarge,
},
{
name: "legacy unknown defaults small",
model: &Model{

View file

@ -14,8 +14,9 @@ const (
// Gemma 4 small templates cover the e2b/e4b family, while 12b/26b/31b use
// the large template. Default to the small prompt unless the model is
// clearly in the large range.
gemma4LargeMinParameterCount = 12_000_000_000
// clearly in the large range. The 12b model reports 11.9B parameters, so
// the threshold sits just below that and above e4b.
gemma4LargeMinParameterCount = 11_500_000_000
)
func resolveRendererName(m *Model) string {