mrq/tortoise-tts
4
6
Fork 15
Code Issues 9 Pull Requests 1 Packages Projects Releases Wiki Activity

quick hotfix to remove offending codesmell (will actually clean it when I finish eating)

Browse Source
This commit is contained in:
mrq 2023-05-04 22:59:57 +00:00
parent 04b7049811
commit 086aad5b49
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tortoise/api.py
View File

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

2
tortoise/utils/diffusion.py
View File

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