accurate loading bar for resume download

This commit is contained in:
Bruce MacDonald 2023-06-29 15:13:11 -04:00
parent 61e39bf5d9
commit 28b47e6068

View file

@ -90,7 +90,7 @@ def download_file(download_url, models_home, location, file_size):
response = requests.get(download_url, headers=header, stream=True)
response.raise_for_status()
total_size = int(response.headers.get('content-length', 0))
total_size = int(response.headers.get('content-length', 0)) + first_byte
with open(local_filename, 'ab' if first_byte else 'wb') as file, tqdm(
total=total_size,