From 28b47e6068cec49d14ff459fa12d83f718882459 Mon Sep 17 00:00:00 2001 From: Bruce MacDonald Date: Thu, 29 Jun 2023 15:13:11 -0400 Subject: [PATCH] accurate loading bar for resume download --- ollama/model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ollama/model.py b/ollama/model.py index 41256c04..5f99a13a 100644 --- a/ollama/model.py +++ b/ollama/model.py @@ -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,