common : clean up map after successful download

Signed-off-by: Adrien Gallouët <angt@huggingface.co>

Author: angt

common/download.cpp

@@ -490,15 +490,16 @@ static void print_progress(size_t current, size_t total) {
 
     static std::mutex mutex;
     static std::map<std::thread::id, int> lines;
+    static int max_line = 0;
 
     std::lock_guard<std::mutex> lock(mutex);
     std::thread::id id = std::this_thread::get_id();
 
     if (lines.find(id) == lines.end()) {
-        lines[id] = lines.size();
+        lines[id] = max_line++;
         std::cout << "\n";
     }
-    int lines_up = lines.size() - lines[id];
+    int lines_up = max_line - lines[id];
 
     size_t width = 50;
     size_t pct = (100 * current) / total;
@@ -519,6 +520,13 @@ static void print_progress(size_t current, size_t total) {
               << "\033[u";
 
     std::cout.flush();
+
+    if (current == total) {
+        lines.erase(id);
+        if (lines.empty()) {
+            max_line = 0;
+        }
+    }
 }
 
 static bool common_pull_file(httplib::Client & cli,