baalajimaestro/ollama
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

fix: allow importing a model from name reference (#3005)

Browse source
This commit is contained in:
Bruce MacDonald 2024-03-08 12:27:47 -05:00 committed by GitHub
parent 0e4669b04f
commit 0cebc79cba
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 4 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
server/images.go
View file

@ -10,6 +10,7 @@ import (
"errors" "errors"
"fmt" "fmt"
"io" "io"
"io/fs"
"log" "log"
"log/slog" "log/slog"
"net/http" "net/http"
@ -322,9 +323,12 @@ func CreateModel(ctx context.Context, name, modelFileDir string, commands []pars
ggufName, err := convertSafetensors(name, pathName) ggufName, err := convertSafetensors(name, pathName)
if err != nil { if err != nil {
var pathErr *fs.PathError
switch { switch {
case errors.Is(err, zip.ErrFormat): case errors.Is(err, zip.ErrFormat):
// it's not a safetensor archive // it's not a safetensor archive
case errors.As(err, &pathErr):
// it's not a file on disk, could be a model reference
default: default:
return err return err
} }