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

only add a layer if there is actual data (#535)

Browse source
This commit is contained in:
Patrick Devine 2023-09-18 13:47:45 -07:00 committed by GitHub
parent 80dd44e80a
commit 24580df958
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 9 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
server/images.go
View file

@ -391,7 +391,7 @@ func CreateModel(ctx context.Context, name string, path string, fn func(resp api
return err return err
} }
// copie the model metadata // copy the model metadata
config.ModelFamily = source.ModelFamily config.ModelFamily = source.ModelFamily
config.ModelType = source.ModelType config.ModelType = source.ModelType
config.ModelFormat = source.ModelFormat config.ModelFormat = source.ModelFormat
@ -461,8 +461,10 @@ func CreateModel(ctx context.Context, name string, path string, fn func(resp api
return err return err
} }
layer.MediaType = mediaType if layer.Size > 0 {
layers = append(layers, layer) layer.MediaType = mediaType
layers = append(layers, layer)
}
case "template", "system", "prompt": case "template", "system", "prompt":
fn(api.ProgressResponse{Status: fmt.Sprintf("creating model %s layer", c.Name)}) fn(api.ProgressResponse{Status: fmt.Sprintf("creating model %s layer", c.Name)})
// remove the layer if one exists // remove the layer if one exists
@ -474,8 +476,10 @@ func CreateModel(ctx context.Context, name string, path string, fn func(resp api
return err return err
} }
layer.MediaType = mediaType if layer.Size > 0 {
layers = append(layers, layer) layer.MediaType = mediaType
layers = append(layers, layer)
}
default: default:
// runtime parameters, build a list of args for each parameter to allow multiple values to be specified (ex: multiple stop sequences) // runtime parameters, build a list of args for each parameter to allow multiple values to be specified (ex: multiple stop sequences)
params[c.Name] = append(params[c.Name], c.Args) params[c.Name] = append(params[c.Name], c.Args)