Pretrained Vocos model with a 48kHz sampling rate, as opposed to 24kHz of the official.
1from vocos import Vocos
2vocos = Vocos.from_pretrained("kittn/vocos-mel-48khz-alpha1")
Here's a summary from the official repo [
link]:
For more details and other variants, check out the repo link above.
1=================================================================
2Layer (type:depth-idx) Param #
3=================================================================
4Vocos --
5├─MelSpectrogramFeatures: 1-1 --
6│ └─MelSpectrogram: 2-1 --
7│ │ └─Spectrogram: 3-1 --
8│ │ └─MelScale: 3-2 --
9├─VocosBackbone: 1-2 --
10│ └─Conv1d: 2-2 918,528
11│ └─LayerNorm: 2-3 2,048
12│ └─ModuleList: 2-4 --
13│ │ └─ConvNeXtBlock: 3-3 4,208,640
14│ │ └─ConvNeXtBlock: 3-4 4,208,640
15│ │ └─ConvNeXtBlock: 3-5 4,208,640
16│ │ └─ConvNeXtBlock: 3-6 4,208,640
17│ │ └─ConvNeXtBlock: 3-7 4,208,640
18│ │ └─ConvNeXtBlock: 3-8 4,208,640
19│ │ └─ConvNeXtBlock: 3-9 4,208,640
20│ │ └─ConvNeXtBlock: 3-10 4,208,640
21│ └─LayerNorm: 2-5 2,048
22├─ISTFTHead: 1-3 --
23│ └─Linear: 2-6 2,101,250
24│ └─ISTFT: 2-7 --
25=================================================================
26Total params: 36,692,994
27Trainable params: 36,692,994
28Non-trainable params: 0
29=================================================================