Spaces:
Runtime error
Runtime error
Commit
·
7a6836d
1
Parent(s):
74328a8
add tqdm for infer_batch_process
Browse files
src/f5_tts/infer/utils_infer.py
CHANGED
|
@@ -473,7 +473,7 @@ def infer_batch_process(
|
|
| 473 |
|
| 474 |
if len(ref_text[-1].encode("utf-8")) == 1:
|
| 475 |
ref_text = ref_text + " "
|
| 476 |
-
for i, gen_text in enumerate(progress.tqdm(gen_text_batches)):
|
| 477 |
# Prepare the text
|
| 478 |
text_list = [ref_text + gen_text]
|
| 479 |
final_text_list = convert_char_to_pinyin(text_list)
|
|
@@ -520,7 +520,7 @@ def infer_batch_process(
|
|
| 520 |
final_wave = np.concatenate(generated_waves)
|
| 521 |
else:
|
| 522 |
final_wave = generated_waves[0]
|
| 523 |
-
for i in range(1, len(generated_waves)):
|
| 524 |
prev_wave = final_wave
|
| 525 |
next_wave = generated_waves[i]
|
| 526 |
|
|
|
|
| 473 |
|
| 474 |
if len(ref_text[-1].encode("utf-8")) == 1:
|
| 475 |
ref_text = ref_text + " "
|
| 476 |
+
for i, gen_text in enumerate(progress.tqdm(gen_text_batches, 'generate_wave')):
|
| 477 |
# Prepare the text
|
| 478 |
text_list = [ref_text + gen_text]
|
| 479 |
final_text_list = convert_char_to_pinyin(text_list)
|
|
|
|
| 520 |
final_wave = np.concatenate(generated_waves)
|
| 521 |
else:
|
| 522 |
final_wave = generated_waves[0]
|
| 523 |
+
for i in progress.tqdm(range(1, len(generated_waves)), 'combined_spectrogram'):
|
| 524 |
prev_wave = final_wave
|
| 525 |
next_wave = generated_waves[i]
|
| 526 |
|