aboutsummaryrefslogtreecommitdiff
path: root/modules/shared.py
diff options
context:
space:
mode:
Diffstat (limited to 'modules/shared.py')
-rw-r--r--modules/shared.py7
1 files changed, 7 insertions, 0 deletions
diff --git a/modules/shared.py b/modules/shared.py
index 667c3441..7bad7a6d 100644
--- a/modules/shared.py
+++ b/modules/shared.py
@@ -307,6 +307,13 @@ class TotalTQDM:
self.reset()
self._tqdm.update()
+ def updateTotal(self, new_total):
+ if not opts.multiple_tqdm:
+ return
+ if self._tqdm is None:
+ self.reset()
+ self._tqdm.total=new_total
+
def clear(self):
if self._tqdm is not None:
self._tqdm.close()