From 3b7f4d180bb125a76b2f4f7b3a74268e5ec5f131 Mon Sep 17 00:00:00 2001
From: Alethea Power <10975066+aletheap@users.noreply.github.com>
Date: Sat, 25 Feb 2023 07:31:14 -0800
Subject: [PATCH] show download progress in bytes (#18)

---
 utils.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/utils.py b/utils.py
index 6d4c9af..e4d0aea 100644
--- a/utils.py
+++ b/utils.py
@@ -33,6 +33,7 @@ def download_gpt2_files(model_size, model_dir):
                 desc="Fetching " + filename,
                 total=file_size,
                 unit_scale=True,
+                unit="b",
             ) as pbar:
                 # 1k for chunk_size, since Ethernet packet size is around 1500 bytes
                 for chunk in r.iter_content(chunk_size=chunk_size):