From a72f2f8efb72b90a2827d2f02cae91b070790dd2 Mon Sep 17 00:00:00 2001 From: SWivid Date: Sat, 22 Feb 2025 08:33:10 +0800 Subject: [PATCH] 0.6.1 fix tqdm func check with difference call behavior from gr.Progress() --- pyproject.toml | 2 +- src/f5_tts/infer/utils_infer.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/pyproject.toml b/pyproject.toml index 9f2da4f..b5d3f19 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta" [project] name = "f5-tts" -version = "0.6.0" +version = "0.6.1" description = "F5-TTS: A Fairytaler that Fakes Fluent and Faithful Speech with Flow Matching" readme = "README.md" license = {text = "MIT License"} diff --git a/src/f5_tts/infer/utils_infer.py b/src/f5_tts/infer/utils_infer.py index 0c5ca66..6a65654 100644 --- a/src/f5_tts/infer/utils_infer.py +++ b/src/f5_tts/infer/utils_infer.py @@ -500,13 +500,13 @@ def infer_batch_process( yield generated_wave, generated_mel_spec[0].cpu().numpy() if streaming: - for gen_text in progress.tqdm(gen_text_batches) if progress else gen_text_batches: + for gen_text in progress.tqdm(gen_text_batches) if progress is not None else gen_text_batches: for chunk in process_batch(gen_text): yield chunk else: with ThreadPoolExecutor() as executor: futures = [executor.submit(process_batch, gen_text) for gen_text in gen_text_batches] - for future in progress.tqdm(futures) if progress else futures: + for future in progress.tqdm(futures) if progress is not None else futures: result = future.result() if result: generated_wave, generated_mel_spec = next(result)