gitea-mirror/F5-TTS
1
0
Fork 0
mirror of https://github.com/SWivid/F5-TTS.git synced 2026-03-12 21:02:50 -07:00
Code Issues Packages Projects Releases Wiki Activity

add tqdm in convert text to pinyin

Browse Source
This commit is contained in:
ZhikangNiu
2026-01-22 16:34:39 +08:00
parent 2aefa7c5f7
commit 5d473e980c
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/f5_tts/train/datasets/prepare_csv_wavs.py
View File

@@ -92,7 +92,11 @@ def process_audio_file(audio_path, text, polyphone):
def batch_convert_texts(texts, polyphone, batch_size=BATCH_SIZE): def batch_convert_texts(texts, polyphone, batch_size=BATCH_SIZE):
"""Convert a list of texts to pinyin in batches.""" """Convert a list of texts to pinyin in batches."""
converted_texts = [] converted_texts = []
for i in range(0, len(texts), batch_size): for i in tqdm(
range(0, len(texts), batch_size),
total=(len(texts) + batch_size - 1) // batch_size,
desc="Converting texts to pinyin",
):
batch = texts[i : i + batch_size] batch = texts[i : i + batch_size]
converted_batch = convert_char_to_pinyin(batch, polyphone=polyphone) converted_batch = convert_char_to_pinyin(batch, polyphone=polyphone)
converted_texts.extend(converted_batch) converted_texts.extend(converted_batch)