neucodec 0.0.3

A package for NeuCodec, based on xcodec2.

HuggingFace: Model, Distilled Model

Model Details

NeuCodec is a Finite Scalar Quantisation (FSQ) based 0.8kbps audio codec for speech tokenization. It takes advantage of the following features:

• It uses both audio (BigCodec) and semantic (Wav2Vec2-BERT) encoders.
• We make use of Finite Scalar Quantisation (FSQ) resulting in a single vector for the quantised output, which makes it ideal for downstream modeling with Speech Language Models.
• At 50 tokens/sec and 16 bits per token, the overall bit-rate is 0.8kbps.
• The codec takes in 16kHz input and outputs 24kHz using an upsampling decoder.

Our work largely based on extending the work of X-Codec2.0.

• Developed by: Neuphonic
• Model type: Neural Audio Codec
• License: apache-2.0
• Repository: https://github.com/neuphonic/neucodec
• Paper: Coming soon!

Get Started

Use the code below to get started with the model.

To install from pypi in a dedicated environment:

conda create -n neucodec python>3.9
conda activate neucodec
pip install neucodec

Then, to use in python:

import librosa
import torch
import torchaudio
from torchaudio import transforms as T
from neucodec import NeuCodec
 
model = NeuCodec.from_pretrained("neuphonic/neucodec")
model.eval().cuda()   
 
y, sr = torchaudio.load(librosa.ex("libri1"))
if sr != 16_000:
    y = T.Resample(sr, 16_000)(y)[None, ...] # (B, 1, T_16)

with torch.no_grad():
    fsq_codes = model.encode_code(y)
    # fsq_codes = model.encode_code(librosa.ex("libri1")) # or directly pass your filepath!
    print(f"Codes shape: {fsq_codes.shape}")  
    recon = model.decode_code(fsq_codes).cpu() # (B, 1, T_24)

torchaudio.save("reconstructed.wav", recon[0, :, :], 24_000)

Training Details

The model was trained using the following data:

• Emilia-YODAS
• MLS
• LibriTTS
• Fleurs
• CommonVoice
• HUI
• Additional proprietary set

All publically available data was covered by either the CC-BY-4.0 or CC0 license.