初始化项目,由ModelHub XC社区提供模型
Model: RL-gang/vocence-orpheus-amir Source: Original Platform
This commit is contained in:
35
snac/.gitattributes
vendored
Normal file
35
snac/.gitattributes
vendored
Normal file
@@ -0,0 +1,35 @@
|
||||
*.7z filter=lfs diff=lfs merge=lfs -text
|
||||
*.arrow filter=lfs diff=lfs merge=lfs -text
|
||||
*.bin filter=lfs diff=lfs merge=lfs -text
|
||||
*.bz2 filter=lfs diff=lfs merge=lfs -text
|
||||
*.ckpt filter=lfs diff=lfs merge=lfs -text
|
||||
*.ftz filter=lfs diff=lfs merge=lfs -text
|
||||
*.gz filter=lfs diff=lfs merge=lfs -text
|
||||
*.h5 filter=lfs diff=lfs merge=lfs -text
|
||||
*.joblib filter=lfs diff=lfs merge=lfs -text
|
||||
*.lfs.* filter=lfs diff=lfs merge=lfs -text
|
||||
*.mlmodel filter=lfs diff=lfs merge=lfs -text
|
||||
*.model filter=lfs diff=lfs merge=lfs -text
|
||||
*.msgpack filter=lfs diff=lfs merge=lfs -text
|
||||
*.npy filter=lfs diff=lfs merge=lfs -text
|
||||
*.npz filter=lfs diff=lfs merge=lfs -text
|
||||
*.onnx filter=lfs diff=lfs merge=lfs -text
|
||||
*.ot filter=lfs diff=lfs merge=lfs -text
|
||||
*.parquet filter=lfs diff=lfs merge=lfs -text
|
||||
*.pb filter=lfs diff=lfs merge=lfs -text
|
||||
*.pickle filter=lfs diff=lfs merge=lfs -text
|
||||
*.pkl filter=lfs diff=lfs merge=lfs -text
|
||||
*.pt filter=lfs diff=lfs merge=lfs -text
|
||||
*.pth filter=lfs diff=lfs merge=lfs -text
|
||||
*.rar filter=lfs diff=lfs merge=lfs -text
|
||||
*.safetensors filter=lfs diff=lfs merge=lfs -text
|
||||
saved_model/**/* filter=lfs diff=lfs merge=lfs -text
|
||||
*.tar.* filter=lfs diff=lfs merge=lfs -text
|
||||
*.tar filter=lfs diff=lfs merge=lfs -text
|
||||
*.tflite filter=lfs diff=lfs merge=lfs -text
|
||||
*.tgz filter=lfs diff=lfs merge=lfs -text
|
||||
*.wasm filter=lfs diff=lfs merge=lfs -text
|
||||
*.xz filter=lfs diff=lfs merge=lfs -text
|
||||
*.zip filter=lfs diff=lfs merge=lfs -text
|
||||
*.zst filter=lfs diff=lfs merge=lfs -text
|
||||
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
||||
71
snac/README.md
Normal file
71
snac/README.md
Normal file
@@ -0,0 +1,71 @@
|
||||
---
|
||||
license: mit
|
||||
tags:
|
||||
- audio
|
||||
---
|
||||
|
||||
# SNAC 🍿
|
||||
|
||||
Multi-**S**cale **N**eural **A**udio **C**odec (SNAC) compressess audio into discrete codes at a low bitrate.
|
||||
|
||||
👉 This model was primarily trained on speech data, and its recommended use case is speech synthesis. See below for other pretrained models.
|
||||
|
||||
🔗 GitHub repository: https://github.com/hubertsiuzdak/snac/
|
||||
|
||||
## Overview
|
||||
|
||||
SNAC encodes audio into hierarchical tokens similarly to SoundStream, EnCodec, and DAC. However, SNAC introduces a simple change where coarse tokens are sampled less frequently,
|
||||
covering a broader time span.
|
||||
|
||||
This model compresses 24 kHz audio into discrete codes at a 0.98 kbps bitrate. It uses 3 RVQ levels with token rates of 12, 23, and
|
||||
47 Hz.
|
||||
|
||||
## Pretrained models
|
||||
|
||||
Currently, all models support only single audio channel (mono).
|
||||
|
||||
| Model | Bitrate | Sample Rate | Params | Recommended use case |
|
||||
|-----------------------------------------------------------------------------|-----------|-------------|--------|--------------------------|
|
||||
| hubertsiuzdak/snac_24khz (this model) | 0.98 kbps | 24 kHz | 19.8 M | 🗣️ Speech |
|
||||
| [hubertsiuzdak/snac_32khz](https://huggingface.co/hubertsiuzdak/snac_32khz) | 1.9 kbps | 32 kHz | 54.5 M | 🎸 Music / Sound Effects |
|
||||
| [hubertsiuzdak/snac_44khz](https://huggingface.co/hubertsiuzdak/snac_44khz) | 2.6 kbps | 44 kHz | 54.5 M | 🎸 Music / Sound Effects |
|
||||
|
||||
## Usage
|
||||
|
||||
Install it using:
|
||||
|
||||
```bash
|
||||
pip install snac
|
||||
```
|
||||
To encode (and decode) audio with SNAC in Python, use the following code:
|
||||
|
||||
```python
|
||||
import torch
|
||||
from snac import SNAC
|
||||
|
||||
model = SNAC.from_pretrained("hubertsiuzdak/snac_24khz").eval().cuda()
|
||||
audio = torch.randn(1, 1, 24000).cuda() # B, 1, T
|
||||
|
||||
with torch.inference_mode():
|
||||
codes = model.encode(audio)
|
||||
audio_hat = model.decode(codes)
|
||||
```
|
||||
|
||||
You can also encode and reconstruct in a single call:
|
||||
|
||||
```python
|
||||
with torch.inference_mode():
|
||||
audio_hat, codes = model(audio)
|
||||
```
|
||||
|
||||
⚠️ Note that `codes` is a list of token sequences of variable lengths, each corresponding to a different temporal
|
||||
resolution.
|
||||
|
||||
```
|
||||
>>> [code.shape[1] for code in codes]
|
||||
[12, 24, 48]
|
||||
```
|
||||
|
||||
## Acknowledgements
|
||||
|
||||
Module definitions are adapted from the [Descript Audio Codec](https://github.com/descriptinc/descript-audio-codec).
|
||||
13
snac/config.json
Normal file
13
snac/config.json
Normal file
@@ -0,0 +1,13 @@
|
||||
{
|
||||
"sampling_rate": 24000,
|
||||
"encoder_dim": 48,
|
||||
"encoder_rates": [2, 4, 8, 8],
|
||||
"decoder_dim": 1024,
|
||||
"decoder_rates": [8, 8, 4, 2],
|
||||
"attn_window_size": null,
|
||||
"codebook_size": 4096,
|
||||
"codebook_dim": 8,
|
||||
"vq_strides": [4, 2, 1],
|
||||
"noise": true,
|
||||
"depthwise": true
|
||||
}
|
||||
3
snac/pytorch_model.bin
Normal file
3
snac/pytorch_model.bin
Normal file
@@ -0,0 +1,3 @@
|
||||
version https://git-lfs.github.com/spec/v1
|
||||
oid sha256:4b8164cc6606bfa627f1a784734c1e539891518f1191ed9194fe1e3b9b4bff40
|
||||
size 79488254
|
||||
Reference in New Issue
Block a user