--- license: mit tags: - model_hub_mixin - pytorch_model_hub_mixin base_model: - Alethia/BigCodec pipeline_tag: feature-extraction --- # BigCodec Python library for bigcodec. This is something that the author of bigcodec didn't do, so I'm doing it for him. ## Setup ``` pip install bigcodec ``` ## Usage ``` import torch.accelerator from bigcodec import BigCodec import torchaudio codec = BigCodec.from_pretrained("intexcp/bigcodec") wav = torchaudio.load("enc.wav")[0] wav = wav.unsqueeze(0) device = torch.accelerator.current_accelerator().type if torch.accelerator.is_available() else "cpu" codec = codec.to(device) wav = wav.to(device) with torch.no_grad(): enc = codec.encode(wav) dec = codec.decode(enc) print(enc) torchaudio.save("dec.wav", dec.squeeze(0).cpu(), 16000, encoding="PCM_F") ```