Commit 5e7fd690 authored by Bruce MacDonald's avatar Bruce MacDonald Committed by Daniel Hiltgen
Browse files

Update images.go

parent 811b1f03
...@@ -421,7 +421,11 @@ func CreateModel(ctx context.Context, name, modelFileDir string, commands []pars ...@@ -421,7 +421,11 @@ func CreateModel(ctx context.Context, name, modelFileDir string, commands []pars
// if the model is not in gguf format, pull the base model to try and get it in gguf format // if the model is not in gguf format, pull the base model to try and get it in gguf format
if fromConfig.ModelFormat != "gguf" { if fromConfig.ModelFormat != "gguf" {
fn(api.ProgressResponse{Status: "updating base model"}) fn(api.ProgressResponse{Status: "updating base model"})
if err := PullModel(ctx, c.Args, &RegistryOptions{}, fn); err != nil { parent, err := GetModel(c.Args)
if err != nil {
return err
}
if err := PullModel(ctx, parent.OriginalModel, &RegistryOptions{}, fn); err != nil {
log.Printf("error pulling model: %v", err) log.Printf("error pulling model: %v", err)
} }
// Reset the file pointer to the beginning of the file // Reset the file pointer to the beginning of the file
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment