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

only pull gguf model if already exists (#1817)

Browse source
This commit is contained in:
Bruce MacDonald 2024-01-05 18:50:00 -05:00 committed by GitHub
parent 9c2941e61b
commit 3a9f447141
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 13 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

25
cmd/cmd.go
View file

@ -158,18 +158,19 @@ func RunHandler(cmd *cobra.Command, args []string) error {
} }
case err != nil: case err != nil:
return err return err
} default:
// the model was found, check if it is in the correct format
if model.Details.Format != "gguf" { if model.Details.Format != "" && model.Details.Format != "gguf" {
// pull and retry to see if the model has been updated // pull and retry to see if the model has been updated
parts := strings.Split(name, string(os.PathSeparator)) parts := strings.Split(name, string(os.PathSeparator))
if len(parts) == 1 { if len(parts) == 1 {
// this is a library model, log some info // this is a library model, log some info
fmt.Fprintln(os.Stderr, "This model is no longer compatible with Ollama. Pulling a new version...") fmt.Fprintln(os.Stderr, "This model is no longer compatible with Ollama. Pulling a new version...")
} }
if err := PullHandler(cmd, []string{name}); err != nil { if err := PullHandler(cmd, []string{name}); err != nil {
fmt.Printf("Error: %s\n", err) fmt.Printf("Error: %s\n", err)
return fmt.Errorf("unsupported model, please update this model to gguf format") // relay the original error return fmt.Errorf("unsupported model, please update this model to gguf format") // relay the original error
}
} }
} }