[BUG] Fix progress bar unit. (PaddlePaddle#3177)

luotao1 · Jun 11, 2024 · 6e63998 · 6e63998
1 parent cf8727b
commit 6e63998
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/paddlespeech/cli/download.py b/paddlespeech/cli/download.py
@@ -133,10 +133,10 @@ def _get_download(url, fullname):
     total_size = req.headers.get('content-length')
     with open(tmp_fullname, 'wb') as f:
         if total_size:
-            with tqdm(total=(int(total_size) + 1023) // 1024) as pbar:
+            with tqdm(total=(int(total_size)), unit='B', unit_scale=True) as pbar:
                 for chunk in req.iter_content(chunk_size=1024):
                     f.write(chunk)
-                    pbar.update(1)
+                    pbar.update(len(chunk))
         else:
             for chunk in req.iter_content(chunk_size=1024):
                 if chunk: