From 086aad5b49e7ef39d043c6b0e12ac33c20773ab8 Mon Sep 17 00:00:00 2001
From: mrq <mrq@ecker.tech>
Date: Thu, 4 May 2023 22:59:57 +0000
Subject: [PATCH] quick hotfix to remove offending codesmell (will actually
clean it when I finish eating)
---
tortoise/api.py | 2 +-
tortoise/utils/diffusion.py | 2 +-
2 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/tortoise/api.py b/tortoise/api.py
index 9b091ca..7df0e7e 100755
--- a/tortoise/api.py
+++ b/tortoise/api.py
@@ -91,7 +91,7 @@ def tqdm_override(arr, verbose=False, progress=None, desc=None):
if progress is None:
return tqdm(arr, disable=not verbose)
- return progress.tqdm(arr, desc=f'{progress.msg_prefix} {desc}' if hasattr(progress, 'msg_prefix') else desc, track_tqdm=True)
+ return progress.tqdm(arr, desc=f'{progress.msg_prefix} {desc}' if hasattr(progress, 'msg_prefix') else desc)
def download_models(specific_models=None):
"""
diff --git a/tortoise/utils/diffusion.py b/tortoise/utils/diffusion.py
index c706416..b344899 100755
--- a/tortoise/utils/diffusion.py
+++ b/tortoise/utils/diffusion.py
@@ -21,7 +21,7 @@ def tqdm_override(arr, verbose=False, progress=None, desc=None):
if progress is None:
return tqdm(arr, disable=not verbose)
- return progress.tqdm(arr, desc=f'{progress.msg_prefix} {desc}' if hasattr(progress, 'msg_prefix') else desc, track_tqdm=True)
+ return progress.tqdm(arr, desc=f'{progress.msg_prefix} {desc}' if hasattr(progress, 'msg_prefix') else desc)
def normal_kl(mean1, logvar1, mean2, logvar2):
"""