create the blobs directory correctly (#508)

45ac07cd · Patrick Devine · GitHub · 7d749cc7 · 45ac07cd
Unverified Commit 45ac07cd authored Sep 11, 2023 by Patrick Devine Committed by GitHub Sep 11, 2023
Show whitespace changes
Inline Side-by-side

Showing with 6 additions and 1 deletion

server/modelpath.go server/modelpath.go +6 -1

No files found.
--- a/server/modelpath.go
+++ b/server/modelpath.go
@@ -133,7 +133,12 @@ func GetBlobsPath(digest string) (string, error) {
 	}

 	path := filepath.Join(home, ".ollama", "models", "blobs", digest)
-	if err := os.MkdirAll(filepath.Dir(path), 0o755); err != nil {
+	dirPath := filepath.Dir(path)
+	if digest == "" {
+		dirPath = path
+	}
+
+	if err := os.MkdirAll(dirPath, 0o755); err != nil {
 		return "", err
 	}