From f020e1d5194a753c7e17f7997cc9f9b4cb85e4c8 Mon Sep 17 00:00:00 2001
From: Bruce MacDonald <brucewmacdonald@gmail.com>
Date: Mon, 14 Aug 2023 13:09:20 -0300
Subject: [PATCH] always remove from in progress map on download

---
 server/download.go | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/server/download.go b/server/download.go
index 932e29a0..72aa0d77 100644
--- a/server/download.go
+++ b/server/download.go
@@ -113,6 +113,7 @@ var chunkSize = 1024 * 1024 // 1 MiB in bytes
 
 // doDownload downloads a blob from the registry and stores it in the blobs directory
 func doDownload(ctx context.Context, mp ModelPath, regOpts *RegistryOptions, f *FileDownload, fn func(api.ProgressResponse)) error {
+	defer inProgress.Delete(f.Digest)
 	var size int64
 
 	fi, err := os.Stat(f.FilePath + "-partial")
@@ -208,8 +209,6 @@ outerLoop:
 		inProgress.Store(f.Digest, f)
 	}
 
-	inProgress.Delete(f.Digest)
-
 	log.Printf("success getting %s\n", f.Digest)
 	return nil
 }