Hugging Face
Models
Datasets
Spaces
Community
Docs
Enterprise
Pricing
Log In
Sign Up
ASLP-lab
/
DiffRhythm-vae
like
42
Safetensors
Chinese
English
DiffRhythm
music
art
diffusion
arxiv:
2503.01183
License:
stable-audio-community
Model card
Files
Files and versions
xet
Community
3
0c3bcd9
DiffRhythm-vae
1.25 GB
3 contributors
History:
4 commits
ing0
Not exported with torchscript
0c3bcd9
10 months ago
src
readme
10 months ago
.gitattributes
Safe
1.56 kB
readme
10 months ago
LICENSE.md
Safe
11.9 kB
readme
10 months ago
README.md
3.44 kB
readme
10 months ago
vae_config.json
Safe
3.61 kB
Not exported with torchscript
10 months ago
vae_model.pt
pickle
Detected Pickle imports (86)
"__torch__.stable_audio_tools.models.bottleneck.VAEBottleneck"
,
"__torch__.torch.nn.modules.container.Sequential"
,
"__torch__.torch.nn.modules.container.___torch_mangle_15.Sequential"
,
"__torch__.torch.nn.modules.container.___torch_mangle_57.Sequential"
,
"__torch__.torch.nn.modules.container.___torch_mangle_37.Sequential"
,
"__torch__.torch.nn.modules.container.___torch_mangle_44.Sequential"
,
"__torch__.stable_audio_tools.models.autoencoders.OobleckDecoder"
,
"__torch__.torch.nn.modules.container.___torch_mangle_55.Sequential"
,
"__torch__.torch.nn.modules.conv.___torch_mangle_46.Conv1d"
,
"__torch__.torch.nn.modules.container.___torch_mangle_28.Sequential"
,
"__torch__.torch.nn.modules.conv.___torch_mangle_30.Conv1d"
,
"__torch__.stable_audio_tools.models.blocks.SnakeBeta"
,
"__torch__.stable_audio_tools.models.autoencoders.___torch_mangle_22.ResidualUnit"
,
"torch._utils._rebuild_tensor_v2"
,
"__torch__.stable_audio_tools.models.autoencoders.___torch_mangle_29.ResidualUnit"
,
"__torch__.stable_audio_tools.models.autoencoders.___torch_mangle_35.ResidualUnit"
,
"__torch__.torch.nn.modules.conv.___torch_mangle_2.Conv1d"
,
"__torch__.torch.nn.modules.conv.___torch_mangle_36.Conv1d"
,
"__torch__.torch.nn.modules.container.___torch_mangle_53.Sequential"
,
"__torch__.stable_audio_tools.models.autoencoders.ResidualUnit"
,
"__torch__.torch.nn.modules.container.___torch_mangle_18.Sequential"
,
"__torch__.torch.nn.modules.container.___torch_mangle_24.Sequential"
,
"__torch__.stable_audio_tools.models.autoencoders.___torch_mangle_48.ResidualUnit"
,
"__torch__.torch.nn.modules.conv.___torch_mangle_54.Conv1d"
,
"__torch__.torch.nn.modules.container.___torch_mangle_11.Sequential"
,
"__torch__.stable_audio_tools.models.autoencoders.DecoderBlock"
,
"__torch__.torch.nn.modules.container.___torch_mangle_21.Sequential"
,
"__torch__.stable_audio_tools.models.autoencoders.___torch_mangle_61.DecoderBlock"
,
"__torch__.torch.nn.modules.conv.___torch_mangle_5.Conv1d"
,
"collections.OrderedDict"
,
"__torch__.torch.nn.modules.container.___torch_mangle_66.Sequential"
,
"__torch__.stable_audio_tools.models.autoencoders.___torch_mangle_32.ResidualUnit"
,
"__torch__.stable_audio_tools.models.autoencoders.___torch_mangle_51.EncoderBlock"
,
"__torch__.torch.nn.modules.conv.___torch_mangle_62.ConvTranspose1d"
,
"__torch__.torch.nn.modules.conv.___torch_mangle_39.Conv1d"
,
"__torch__.torch.nn.modules.conv.___torch_mangle_23.Conv1d"
,
"__torch__.torch.nn.modules.container.___torch_mangle_63.Sequential"
,
"__torch__.torch.nn.modules.container.___torch_mangle_69.Sequential"
,
"__torch__.torch.nn.modules.conv.___torch_mangle_49.Conv1d"
,
"__torch__.torch.nn.modules.conv.___torch_mangle_33.Conv1d"
,
"__torch__.torch.nn.modules.conv.___torch_mangle_20.Conv1d"
,
"__torch__.torch.nn.modules.conv.___torch_mangle_13.Conv1d"
,
"__torch__.stable_audio_tools.models.autoencoders.___torch_mangle_19.ResidualUnit"
,
"__torch__.torch.nn.modules.container.___torch_mangle_9.Sequential"
,
"__torch__.stable_audio_tools.models.autoencoders.___torch_mangle_4.ResidualUnit"
,
"__torch__.stable_audio_tools.models.autoencoders.___torch_mangle_67.DecoderBlock"
,
"__torch__.torch.nn.modules.container.___torch_mangle_6.Sequential"
,
"__torch__.torch.nn.modules.conv.___torch_mangle_52.Conv1d"
,
"__torch__.stable_audio_tools.models.autoencoders.___torch_mangle_64.DecoderBlock"
,
"__torch__.torch.nn.modules.conv.ConvTranspose1d"
,
"__torch__.stable_audio_tools.models.autoencoders.___torch_mangle_7.ResidualUnit"
,
"__torch__.torch.nn.modules.conv.___torch_mangle_26.Conv1d"
,
"__torch__.stable_audio_tools.models.autoencoders.AudioAutoencoder"
,
"__torch__.stable_audio_tools.models.autoencoders.___torch_mangle_45.ResidualUnit"
,
"__torch__.torch.nn.modules.container.___torch_mangle_50.Sequential"
,
"__torch__.torch.nn.modules.conv.___torch_mangle_27.Conv1d"
,
"torch.FloatStorage"
,
"__torch__.torch.nn.modules.conv.___torch_mangle_56.ConvTranspose1d"
,
"__torch__.stable_audio_tools.models.autoencoders.___torch_mangle_38.EncoderBlock"
,
"__torch__.torch.nn.modules.linear.Identity"
,
"__torch__.torch.nn.modules.conv.___torch_mangle_68.Conv1d"
,
"__torch__.torch.nn.modules.container.___torch_mangle_47.Sequential"
,
"__torch__.torch.nn.modules.conv.___torch_mangle_0.Conv1d"
,
"__torch__.torch.nn.modules.container.___torch_mangle_31.Sequential"
,
"torch.jit._pickle.build_intlist"
,
"__torch__.torch.nn.modules.conv.___torch_mangle_59.ConvTranspose1d"
,
"__torch__.torch.nn.modules.container.___torch_mangle_34.Sequential"
,
"__torch__.torch.nn.modules.conv.___torch_mangle_8.Conv1d"
,
"__torch__.torch.nn.modules.conv.___torch_mangle_65.ConvTranspose1d"
,
"__torch__.torch.nn.modules.conv.___torch_mangle_14.Conv1d"
,
"__torch__.torch.nn.modules.conv.___torch_mangle_10.Conv1d"
,
"__torch__.torch.nn.modules.conv.Conv1d"
,
"__torch__.torch.nn.modules.container.___torch_mangle_3.Sequential"
,
"__torch__.torch.nn.modules.container.___torch_mangle_41.Sequential"
,
"__torch__.stable_audio_tools.models.autoencoders.___torch_mangle_58.DecoderBlock"
,
"__torch__.torch.nn.modules.container.___torch_mangle_60.Sequential"
,
"__torch__.stable_audio_tools.models.autoencoders.EncoderBlock"
,
"__torch__.torch.nn.modules.conv.___torch_mangle_40.Conv1d"
,
"__torch__.torch.nn.modules.conv.___torch_mangle_1.Conv1d"
,
"__torch__.stable_audio_tools.models.autoencoders.___torch_mangle_42.ResidualUnit"
,
"__torch__.stable_audio_tools.models.autoencoders.___torch_mangle_12.EncoderBlock"
,
"__torch__.stable_audio_tools.models.autoencoders.___torch_mangle_16.ResidualUnit"
,
"__torch__.torch.nn.modules.conv.___torch_mangle_17.Conv1d"
,
"__torch__.torch.nn.modules.conv.___torch_mangle_43.Conv1d"
,
"__torch__.stable_audio_tools.models.autoencoders.___torch_mangle_25.EncoderBlock"
,
"__torch__.stable_audio_tools.models.autoencoders.OobleckEncoder"
How to fix it?
625 MB
xet
vae
11 months ago
vae_model.safetensors
624 MB
xet
Not exported with torchscript
10 months ago