congcuong-cse commited on
Commit
7a6836d
·
1 Parent(s): 74328a8

add tqdm for infer_batch_process

Browse files
Files changed (1) hide show
  1. src/f5_tts/infer/utils_infer.py +2 -2
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