From 014fed4405592b113cbe60104ad5543d6f821cf5 Mon Sep 17 00:00:00 2001 From: 0xCAFEBABE0 Date: Thu, 12 Dec 2024 16:55:43 +0800 Subject: [PATCH] fix(bug).when generating text that contains only punctuation marks or whitespace characters, the CPU usage reaches 100%, and the process crashes. --- cosyvoice/cli/model.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/cosyvoice/cli/model.py b/cosyvoice/cli/model.py index d878fc1..850e5db 100644 --- a/cosyvoice/cli/model.py +++ b/cosyvoice/cli/model.py @@ -146,8 +146,10 @@ class CosyVoiceModel: llm_prompt_speech_token=torch.zeros(1, 0, dtype=torch.int32), flow_prompt_speech_token=torch.zeros(1, 0, dtype=torch.int32), prompt_speech_feat=torch.zeros(1, 0, 80), stream=False, speed=1.0, **kwargs): + # When generating text that contains only punctuation marks or whitespace characters + # - Returning 10ms of silence ensures consistent processing logic. if is_only_punctuation(text): - return {'tts_speech': torch.zeros(1, int(0.01 * 22050))} #返回10ms空白音频，保证了一致的上下游处理逻辑 + return {'tts_speech': torch.zeros(1, int(0.01 * 22050))} # this_uuid is used to track variables related to this inference thread this_uuid = str(uuid.uuid1()) with self.lock: