AI/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-10-01 01:26:03 -04:00
Code Issues Packages Projects Releases Wiki Activity

Downloader: handle one more retry case after 5770e06c48

Browse Source
This commit is contained in:
oobabooga 2024-05-04 19:25:04 -07:00
parent cb31998605
commit 0b193b8553
1 changed files with 10 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
download-model.py
View File

@ -190,17 +190,17 @@ class ModelDownloader:
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
headers = {'Range': f'bytes={output_path.stat().st_size}-'}
mode = 'ab'
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'
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))