Skip to content

Commit 69850fa

Browse files
committed
fix. address #179
1 parent aaf1fa7 commit 69850fa

File tree

3 files changed

+6
-9
lines changed

3 files changed

+6
-9
lines changed

gradio_app.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -158,9 +158,8 @@ def infer_batch(ref_audio, ref_text, gen_text_batches, exp_name, remove_silence,
158158

159159
# Calculate duration
160160
ref_audio_len = audio.shape[-1] // hop_length
161-
zh_pause_punc = r"。,、;:?!"
162-
ref_text_len = len(ref_text.encode('utf-8')) + 3 * len(re.findall(zh_pause_punc, ref_text))
163-
gen_text_len = len(gen_text.encode('utf-8')) + 3 * len(re.findall(zh_pause_punc, gen_text))
161+
ref_text_len = len(ref_text.encode('utf-8'))
162+
gen_text_len = len(gen_text.encode('utf-8'))
164163
duration = ref_audio_len + int(ref_audio_len / ref_text_len * gen_text_len / speed)
165164

166165
# inference

inference-cli.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -250,9 +250,8 @@ def infer_batch(ref_audio, ref_text, gen_text_batches, model,ckpt_file,file_voca
250250

251251
# Calculate duration
252252
ref_audio_len = audio.shape[-1] // hop_length
253-
zh_pause_punc = r"。,、;:?!"
254-
ref_text_len = len(ref_text.encode('utf-8')) + 3 * len(re.findall(zh_pause_punc, ref_text))
255-
gen_text_len = len(gen_text.encode('utf-8')) + 3 * len(re.findall(zh_pause_punc, gen_text))
253+
ref_text_len = len(ref_text.encode('utf-8'))
254+
gen_text_len = len(gen_text.encode('utf-8'))
256255
duration = ref_audio_len + int(ref_audio_len / ref_text_len * gen_text_len / speed)
257256

258257
# inference

model/utils.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -296,9 +296,8 @@ def get_inference_prompt(
296296
# # test vocoder resynthesis
297297
# ref_audio = gt_audio
298298
else:
299-
zh_pause_punc = r"。,、;:?!"
300-
ref_text_len = len(prompt_text.encode('utf-8')) + 3 * len(re.findall(zh_pause_punc, prompt_text))
301-
gen_text_len = len(gt_text.encode('utf-8')) + 3 * len(re.findall(zh_pause_punc, gt_text))
299+
ref_text_len = len(prompt_text.encode('utf-8'))
300+
gen_text_len = len(gt_text.encode('utf-8'))
302301
total_mel_len = ref_mel_len + int(ref_mel_len / ref_text_len * gen_text_len / speed)
303302

304303
# to mel spectrogram

0 commit comments

Comments
 (0)