only pull gguf model if already exists (#1817)

This commit is contained in:
Bruce MacDonald
2024-01-05 18:50:00 -05:00
committed by GitHub
parent 9c2941e61b
commit 3a9f447141

View File

@@ -158,9 +158,9 @@ func RunHandler(cmd *cobra.Command, args []string) error {
}
case err != nil:
return err
}
if model.Details.Format != "gguf" {
default:
// the model was found, check if it is in the correct format
if model.Details.Format != "" && model.Details.Format != "gguf" {
// pull and retry to see if the model has been updated
parts := strings.Split(name, string(os.PathSeparator))
if len(parts) == 1 {
@@ -172,6 +172,7 @@ func RunHandler(cmd *cobra.Command, args []string) error {
return fmt.Errorf("unsupported model, please update this model to gguf format") // relay the original error
}
}
}
return RunGenerate(cmd, args)
}