diff --git a/paddlespeech/cli/download.py b/paddlespeech/cli/download.py index 5661f18f9..958560e72 100644 --- 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: - for chunk in req.iter_content(chunk_size=1024): + with tqdm(total=(int(total_size))) as pbar: + for chunk in req.iter_content(chunk_size=1024, unit='B', unit_scale=True): f.write(chunk) - pbar.update(1) + pbar.update(len(chunk)) else: for chunk in req.iter_content(chunk_size=1024): if chunk: