Files
ChatTTS/tools/audio/np.py
T
github-actions[bot] d089bd2492 chore(format): run black on dev (#560)
Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
2024-07-10 16:53:29 +09:00

12 lines
254 B
Python

import math
import numpy as np
from numba import jit
@jit
def float_to_int16(audio: np.ndarray) -> np.ndarray:
am = int(math.ceil(float(np.abs(audio).max())) * 32768)
am = 32767 * 32768 // am
return np.multiply(audio, am).astype(np.int16)