explicitly decode maxarraysize 1024

This commit is contained in:
Michael Yang 2025-04-25 16:08:25 -07:00 committed by Michael Yang
parent ced7d0e53d
commit 340448d2d1
4 changed files with 6 additions and 6 deletions

View file

@ -75,7 +75,7 @@ func (m *Model) Capabilities() []model.Capability {
if err == nil {
defer r.Close()
f, _, err := ggml.Decode(r, 0)
f, _, err := ggml.Decode(r, 1024)
if err == nil {
if _, ok := f.KV()[fmt.Sprintf("%s.pooling_type", f.KV().Architecture())]; ok {
capabilities = append(capabilities, model.CapabilityEmbedding)