mirror of
https://github.com/remsky/Kokoro-FastAPI.git
synced 2025-04-13 09:39:17 +00:00
Fix audio chunk concatenation and dtype conversion confliction
- Modify audio chunk concatenation to handle float32 audio data - Add explicit conversion from float32 to int16 using amplitude scaling - Remove unnecessary dtype specification in np.concatenate
This commit is contained in:
parent
5c8f941f06
commit
2dc9b81ad5
1 changed files with 6 additions and 1 deletions
|
@ -23,10 +23,15 @@ class AudioChunk:
|
|||
output=AudioChunk(audio_chunk_list[0].audio,audio_chunk_list[0].word_timestamps)
|
||||
|
||||
for audio_chunk in audio_chunk_list[1:]:
|
||||
output.audio=np.concatenate((output.audio,audio_chunk.audio),dtype=np.int16)
|
||||
output.audio=np.concatenate((output.audio, audio_chunk.audio))
|
||||
if output.word_timestamps is not None:
|
||||
output.word_timestamps+=audio_chunk.word_timestamps
|
||||
|
||||
# 从float32转换为int16,使用音频的方式
|
||||
if output.audio.dtype == np.float32:
|
||||
output.audio = (output.audio * 32767).astype(np.int16)
|
||||
|
||||
|
||||
return output
|
||||
|
||||
class ModelBackend(ABC):
|
||||
|
|
Loading…
Add table
Reference in a new issue