Method encode

moss_tts_delay/llama_cpp/pipeline.py:104–118

(self, waveform, n_quantizers=N_QUANTIZERS)

    class _Enc:
        def encode(self, waveform, n_quantizers=N_QUANTIZERS):
            if waveform.ndim == 1:
                waveform = waveform[np.newaxis, np.newaxis, :]
            elif waveform.ndim == 2:
                waveform = waveform[np.newaxis, :]
            T = waveform.shape[-1]
            padded = ((T + DOWNSAMPLE_RATE - 1) // DOWNSAMPLE_RATE) * DOWNSAMPLE_RATE
            if padded != T:
                waveform = np.concatenate(
                    [waveform, np.zeros((1, 1, padded - T), dtype=np.float32)], axis=-1,
                )
            waveform = waveform.astype(np.float32)
            nq = np.array(n_quantizers, dtype=np.int64)
            r = engine.run(outs, {ins[0]: waveform, ins[1]: nq})
            return r[0][:, 0, :int(r[1][0])].T.astype(np.int64)

        def close(self):
            engine.close()