Downloader: handle one more retry case after 5770e06

oobabooga · May 5, 2024 · 0b193b8 · 0b193b8
1 parent cb31998
commit 0b193b8
Showing 1 changed file with 9 additions and 9 deletions.
diff --git a/download-model.py b/download-model.py
@@ -190,17 +190,17 @@ def get_single_file(self, url, output_folder, start_from_scratch=False):
             headers = {}
             mode = 'wb'
 
-            if output_path.exists() and not start_from_scratch:
-                # Resume download
-                r = session.get(url, stream=True, timeout=20)
-                total_size = int(r.headers.get('content-length', 0))
-                if output_path.stat().st_size >= total_size:
-                    return
+            try:
+                if output_path.exists() and not start_from_scratch:
+                    # Resume download
+                    r = session.get(url, stream=True, timeout=20)
+                    total_size = int(r.headers.get('content-length', 0))
+                    if output_path.stat().st_size >= total_size:
+                        return
 
-                headers = {'Range': f'bytes={output_path.stat().st_size}-'}
-                mode = 'ab'
+                    headers = {'Range': f'bytes={output_path.stat().st_size}-'}
+                    mode = 'ab'
 
-            try:
                 with session.get(url, stream=True, headers=headers, timeout=30) as r:
                     r.raise_for_status()  # If status is not 2xx, raise an error
                     total_size = int(r.headers.get('content-length', 0))