create the blobs directory correctly (ollama#508)

bitsnaps · Sep 11, 2023 · 45ac07c · 45ac07c
1 parent 7d749cc
commit 45ac07c
Showing 1 changed file with 6 additions and 1 deletion.
diff --git 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
 	}