mirror of
https://github.com/ollama/ollama.git
synced 2025-05-11 18:36:41 +02:00
explicitly decode maxarraysize 1024
This commit is contained in:
parent
ced7d0e53d
commit
340448d2d1
4 changed files with 6 additions and 6 deletions
|
@ -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)
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue