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):
     """