diff --git a/paddlespeech/cli/download.py b/paddlespeech/cli/download.py index 5661f18f..e77a05d2 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: + 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: