Fixed the ZeroDivisionError when downloading a model (#2797)

2023-06-21 11:31:50 -04:00 · 2023-06-21 11:31:50 -04:00 · 89fb6f9236
commit 89fb6f9236
parent 90be1d9fe1
1 changed files with 1 additions and 1 deletions
--- a/download-model.py
+++ b/download-model.py
@ -149,7 +149,7 @@ class ModelDownloader:
                    for data in r.iter_content(block_size):
                        t.update(len(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)
                            self.progress_bar(float(count) / float(total_size), f"Downloading {filename}")