upload snac_vocos_16khz_hop200_scale8421_1kh
Browse files- README.md +42 -3
- snac_vocos_16khz_hop200_scale8421_1kh.ckpt +3 -0
README.md
CHANGED
|
@@ -1,3 +1,42 @@
|
|
| 1 |
-
|
| 2 |
-
|
| 3 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# SNAC-Vocos
|
| 2 |
+
A trainer for [SNAC](https://github.com/hubertsiuzdak/snac) (Multi-Scale Neural Audio Codec) has replaced the decoder with Vocos.
|
| 3 |
+
|
| 4 |
+
## Installation
|
| 5 |
+
Suggested python>=3.9.
|
| 6 |
+
Clone the repository:
|
| 7 |
+
```
|
| 8 |
+
git clone https://github.com/hertz-pj/SNAC-Vocos
|
| 9 |
+
cd SNAC-Vocos
|
| 10 |
+
```
|
| 11 |
+
Install packages:
|
| 12 |
+
```
|
| 13 |
+
pip install -r requirements.txt
|
| 14 |
+
```
|
| 15 |
+
## Infer
|
| 16 |
+
Refer to the [infer.py](./infer.py) for inference instructions and usage examples.
|
| 17 |
+
|
| 18 |
+
## Available Models
|
| 19 |
+
| Model name | Huggingface | Corpus | Domain |
|
| 20 |
+
|:------------|:--------|:--------|:--------|
|
| 21 |
+
|snac_vocos_16khz_hop200_scale8421_1kh | [🤗](https://huggingface.co/hertz-pj/snac-vocos) | 1k hours | Speech(Mandarin/English) |
|
| 22 |
+
|
| 23 |
+
|
| 24 |
+
## Training
|
| 25 |
+
1、Prepare a filelist of audio files for the training and validation set, e.g. [train.list](./data/train.list).
|
| 26 |
+
2、Fill a config file, e.g. [snac_vocos.yaml](./config/snac_vocos_nq4_scale8421_16khz.yaml). The main parameters to pay attention to are batch_size, filelist_path, save_dir, and device.
|
| 27 |
+
3、Start training
|
| 28 |
+
```
|
| 29 |
+
python train.py fit --config ./configs/snac_vocos.yaml
|
| 30 |
+
```
|
| 31 |
+
|
| 32 |
+
## TODO
|
| 33 |
+
- [x] Release code
|
| 34 |
+
- [x] Release a checkpoint trained with 1k hours of speech(Mandarin/English).
|
| 35 |
+
- [ ] Demo page.
|
| 36 |
+
|
| 37 |
+
|
| 38 |
+
## Acknowledgements
|
| 39 |
+
This implementation uses parts of the code from the following Github repos:
|
| 40 |
+
- [SNAC](https://github.com/hubertsiuzdak/snac)
|
| 41 |
+
- [WavTokenizer](https://github.com/jishengpeng/WavTokenizer/)
|
| 42 |
+
|
snac_vocos_16khz_hop200_scale8421_1kh.ckpt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:c8fa34b981f1d1f62a25801f86aeb041d5f548096ae4ec1c92761f749ed90d40
|
| 3 |
+
size 1710208559
|