deepgeek/CosyVoice
2
0
Fork 0
mirror of https://github.com/FunAudioLLM/CosyVoice.git synced 2026-02-05 18:09:24 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix(bug).when generating text that contains only punctuation marks or whitespace characters, the CPU usage reaches 100%, and the process crashes.

Browse Source
This commit is contained in:
0xCAFEBABE0
2024-12-12 16:55:43 +08:00
parent f56c2583e8
commit 014fed4405
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
cosyvoice/cli/model.py
View File

@@ -146,8 +146,10 @@ class CosyVoiceModel:
llm_prompt_speech_token=torch.zeros(1, 0, dtype=torch.int32), llm_prompt_speech_token=torch.zeros(1, 0, dtype=torch.int32),
flow_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): 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): 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 is used to track variables related to this inference thread
this_uuid = str(uuid.uuid1()) this_uuid = str(uuid.uuid1())
with self.lock: with self.lock: