Files
ChatTTS/tools/audio/np.py
T
2024-10-10 21:47:12 +09:00

12 lines
269 B
Python

import math
import numpy as np
from numba import jit
@jit(nopython=True)
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)