| | --- |
| | 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") |
| | ``` |
