Mirrors/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-11-25 17:29:22 +01:00
Code Issues Packages Projects Releases Wiki Activity

Fixed the ZeroDivisionError when downloading a model (#2797)

Browse Source
This commit is contained in:
Gaurav Bhagchandani 2023-06-21 11:31:50 -04:00 committed by GitHub
parent 90be1d9fe1
commit 89fb6f9236
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
download-model.py
View File

@ -149,7 +149,7 @@ class ModelDownloader:
for data in r.iter_content(block_size): for data in r.iter_content(block_size):
t.update(len(data)) t.update(len(data))
f.write(data) f.write(data)
if self.progress_bar is not None: if total_size != 0 and self.progress_bar is not None:
count += len(data) count += len(data)
self.progress_bar(float(count) / float(total_size), f"Downloading {filename}") self.progress_bar(float(count) / float(total_size), f"Downloading {filename}")