Component	Keys prefix	Description
Audio VAE	`audio_vae.*`	Encoder / decoder VAE operating on mel-spectrograms (BF16)
Vocoder	`vocoder.vocoder.*`	HiFi-GAN style neural vocoder (BF16)
BWE Generator	`vocoder.bwe_generator.*`	Bandwidth extension generator (BF16)
Mel STFT	`vocoder.mel_stft.*`	Mel filterbank + STFT forward/inverse basis (BF16)

All weights are stored in BFloat16.

File

File	Contents
`dramabox-audiovae-vocoder.safetensors`	audio_vae + vocoder (merged)

Usage

python

from safetensors import safe_open

tensors = {}
with safe_open("dramabox-audiovae-vocoder.safetensors", framework="pt", device="cpu") as f:
    for key in f.keys():
        tensors[key] = f.get_tensor(key)

print(list(tensors.keys())[:5])

Source

Extracted from the original ResembleAI/Dramabox checkpoint. Please refer to the original repository for licensing details.

zuhri025/dramabox-audio-vae-vocoder

Dramabox — Audio VAE + Vocoder

File

Usage

Source

zuhri025/
dramabox-audio-vae-vocoder