jeans/ollama
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

short circuit

Browse Source
This commit is contained in:
Josh Yan 2024-07-23 17:14:34 -07:00
parent b8c3d54f7a
commit e39be4f63a
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
server/model.go
View File

@ -245,7 +245,12 @@ func parseFromFile(ctx context.Context, file *os.File, digest string, fn func(ap
mediatype = "application/vnd.ollama.image.projector"
}
layer, err := NewLayer(io.NewSectionReader(file, offset, n), mediatype)
var layer *Layer
if digest != "" && n == stat.Size() {
layer, err = NewLayerFromLayer(digest, mediatype, file.Name())
} else {
layer, err = NewLayer(io.NewSectionReader(file, offset, n), mediatype)
}
if err != nil {
return nil, err
}