highperfocused/ollama
1
0
Fork 0
mirror of https://github.com/ollama/ollama.git synced 2025-07-28 16:44:39 +02:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #5051 from ollama/mxyng/capabilities

Browse Source 
add model capabilities
This commit is contained in:
Michael Yang
2024-07-02 14:26:07 -07:00
committed by GitHub
parent 400056e154 da8e2a0447
commit dddb58a38b
31 changed files with 354 additions and 192 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
server/model.go
View File

@@ -15,7 +15,7 @@ import (
"github.com/ollama/ollama/api"
"github.com/ollama/ollama/convert"
"github.com/ollama/ollama/llm"
"github.com/ollama/ollama/templates"
"github.com/ollama/ollama/template"
"github.com/ollama/ollama/types/model"
)
@@ -256,7 +256,7 @@ func parseFromFile(ctx context.Context, file *os.File, digest string, fn func(ap
func detectChatTemplate(layers []*layerGGML) ([]*layerGGML, error) {
for _, layer := range layers {
if s := layer.GGML.KV().ChatTemplate(); s != "" {
if t, err := templates.NamedTemplate(s); err != nil {
if t, err := template.Named(s); err != nil {
slog.Debug("template detection", "error", err)
} else {
tmpl, err := NewLayer(t.Reader(), "application/vnd.ollama.image.template")