show download progress in bytes (#18)

pull/21/head
Alethea Power 2023-02-25 07:31:14 -08:00 zatwierdzone przez GitHub
rodzic dfb5df895a
commit 3b7f4d180b
Nie znaleziono w bazie danych klucza dla tego podpisu
ID klucza GPG: 4AEE18F83AFDEB23
1 zmienionych plików z 1 dodań i 0 usunięć

Wyświetl plik

@ -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):