浏览代码

only pull gguf model if already exists (#1817)

Bruce MacDonald 1 年之前
父节点
当前提交
3a9f447141
共有 1 个文件被更改,包括 13 次插入12 次删除
  1. 13 12
      cmd/cmd.go

+ 13 - 12
cmd/cmd.go

@@ -158,18 +158,19 @@ func RunHandler(cmd *cobra.Command, args []string) error {
 		}
 	case err != nil:
 		return err
-	}
-
-	if 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 {
-			// this is a library model, log some info
-			fmt.Fprintln(os.Stderr, "This model is no longer compatible with Ollama. Pulling a new version...")
-		}
-		if err := PullHandler(cmd, []string{name}); err != nil {
-			fmt.Printf("Error: %s\n", err)
-			return fmt.Errorf("unsupported model, please update this model to gguf format") // relay the original error
+	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 {
+				// this is a library model, log some info
+				fmt.Fprintln(os.Stderr, "This model is no longer compatible with Ollama. Pulling a new version...")
+			}
+			if err := PullHandler(cmd, []string{name}); err != nil {
+				fmt.Printf("Error: %s\n", err)
+				return fmt.Errorf("unsupported model, please update this model to gguf format") // relay the original error
+			}
 		}
 	}