From f1c6da98c12e6584943704764df7a39fd6c62fe6 Mon Sep 17 00:00:00 2001 From: "github-actions[bot]" <41898282+github-actions[bot]@users.noreply.github.com> Date: Sat, 6 Jul 2024 00:42:17 +0900 Subject: [PATCH] chore(format): run black on dev (#536) Co-authored-by: github-actions[bot] --- examples/api/main.py | 4 +++- examples/cmd/stream.py | 3 ++- tools/audio/np.py | 1 + 3 files changed, 6 insertions(+), 2 deletions(-) diff --git a/examples/api/main.py b/examples/api/main.py index 1395f2f..d407a5d 100644 --- a/examples/api/main.py +++ b/examples/api/main.py @@ -97,7 +97,9 @@ async def generate_voice(params: ChatTTSParams): # zip all of the audio files together buf = io.BytesIO() - with zipfile.ZipFile(buf, "a", compression=zipfile.ZIP_DEFLATED, allowZip64=False) as f: + with zipfile.ZipFile( + buf, "a", compression=zipfile.ZIP_DEFLATED, allowZip64=False + ) as f: for idx, wav in enumerate(wavs): f.writestr(wav_arr_to_mp3_view(wav), f"{idx}.mp3") logger.info("Audio generation successful.") diff --git a/examples/cmd/stream.py b/examples/cmd/stream.py index 75296a1..dd4b9b3 100644 --- a/examples/cmd/stream.py +++ b/examples/cmd/stream.py @@ -3,12 +3,13 @@ import threading import time import random -import pyaudio # please install it manually +import pyaudio # please install it manually import numpy as np import ChatTTS from tools.audio import batch_unsafe_float_to_int16 + # 流式声音处理器 class AudioStreamer: def __init__(self): diff --git a/tools/audio/np.py b/tools/audio/np.py index 32ba820..527a140 100644 --- a/tools/audio/np.py +++ b/tools/audio/np.py @@ -13,6 +13,7 @@ def unsafe_float_to_int16(audio: np.ndarray) -> np.ndarray: audio16 = audio.astype(np.int16) return audio16 + @jit def batch_unsafe_float_to_int16(audios: list[np.ndarray]) -> list[np.ndarray]: """