Hugging Face's logo

gatilin
/
lance_wan22_vae

PyTorch
Wan2.2
video-vae
lance
extracted-component
unified-multimodal
Model card Files
xet
 Community
lance_wan22_vae / tensor_mapping.json
gatilin's picture
gatilin
Publish pinned Lance Wan2.2 VAE with strict native component validation
d52c11d verified
Raw History Blame Contribute Delete
31.2 kB
{
"encoder.conv1.weight": {
"shape": [
160,
12,
3,
3,
3
],
"dtype": "torch.float32",
"bytes": 207360
},
"encoder.conv1.bias": {
"shape": [
160
],
"dtype": "torch.float32",
"bytes": 640
},
"encoder.downsamples.0.downsamples.0.residual.0.gamma": {
"shape": [
160,
1,
1,
1
],
"dtype": "torch.float32",
"bytes": 640
},
"encoder.downsamples.0.downsamples.0.residual.2.weight": {
"shape": [
160,
160,
3,
3,
3
],
"dtype": "torch.float32",
"bytes": 2764800
},
"encoder.downsamples.0.downsamples.0.residual.2.bias": {
"shape": [
160
],
"dtype": "torch.float32",
"bytes": 640
},
"encoder.downsamples.0.downsamples.0.residual.3.gamma": {
"shape": [
160,
1,
1,
1
],
"dtype": "torch.float32",
"bytes": 640
},
"encoder.downsamples.0.downsamples.0.residual.6.weight": {
"shape": [
160,
160,
3,
3,
3
],
"dtype": "torch.float32",
"bytes": 2764800
},
"encoder.downsamples.0.downsamples.0.residual.6.bias": {
"shape": [
160
],
"dtype": "torch.float32",
"bytes": 640
},
"encoder.downsamples.0.downsamples.1.residual.0.gamma": {
"shape": [
160,
1,
1,
1
],
"dtype": "torch.float32",
"bytes": 640
},
"encoder.downsamples.0.downsamples.1.residual.2.weight": {
"shape": [
160,
160,
3,
3,
3
],
"dtype": "torch.float32",
"bytes": 2764800
},
"encoder.downsamples.0.downsamples.1.residual.2.bias": {
"shape": [
160
],
"dtype": "torch.float32",
"bytes": 640
},
"encoder.downsamples.0.downsamples.1.residual.3.gamma": {
"shape": [
160,
1,
1,
1
],
"dtype": "torch.float32",
"bytes": 640
},
"encoder.downsamples.0.downsamples.1.residual.6.weight": {
"shape": [
160,
160,
3,
3,
3
],
"dtype": "torch.float32",
"bytes": 2764800
},
"encoder.downsamples.0.downsamples.1.residual.6.bias": {
"shape": [
160
],
"dtype": "torch.float32",
"bytes": 640
},
"encoder.downsamples.0.downsamples.2.resample.1.weight": {
"shape": [
160,
160,
3,
3
],
"dtype": "torch.float32",
"bytes": 921600
},
"encoder.downsamples.0.downsamples.2.resample.1.bias": {
"shape": [
160
],
"dtype": "torch.float32",
"bytes": 640
},
"encoder.downsamples.1.downsamples.0.residual.0.gamma": {
"shape": [
160,
1,
1,
1
],
"dtype": "torch.float32",
"bytes": 640
},
"encoder.downsamples.1.downsamples.0.residual.2.weight": {
"shape": [
320,
160,
3,
3,
3
],
"dtype": "torch.float32",
"bytes": 5529600
},
"encoder.downsamples.1.downsamples.0.residual.2.bias": {
"shape": [
320
],
"dtype": "torch.float32",
"bytes": 1280
},
"encoder.downsamples.1.downsamples.0.residual.3.gamma": {
"shape": [
320,
1,
1,
1
],
"dtype": "torch.float32",
"bytes": 1280
},
"encoder.downsamples.1.downsamples.0.residual.6.weight": {
"shape": [
320,
320,
3,
3,
3
],
"dtype": "torch.float32",
"bytes": 11059200
},
"encoder.downsamples.1.downsamples.0.residual.6.bias": {
"shape": [
320
],
"dtype": "torch.float32",
"bytes": 1280
},
"encoder.downsamples.1.downsamples.0.shortcut.weight": {
"shape": [
320,
160,
1,
1,
1
],
"dtype": "torch.float32",
"bytes": 204800
},
"encoder.downsamples.1.downsamples.0.shortcut.bias": {
"shape": [
320
],
"dtype": "torch.float32",
"bytes": 1280
},
"encoder.downsamples.1.downsamples.1.residual.0.gamma": {
"shape": [
320,
1,
1,
1
],
"dtype": "torch.float32",
"bytes": 1280
},
"encoder.downsamples.1.downsamples.1.residual.2.weight": {
"shape": [
320,
320,
3,
3,
3
],
"dtype": "torch.float32",
"bytes": 11059200
},
"encoder.downsamples.1.downsamples.1.residual.2.bias": {
"shape": [
320
],
"dtype": "torch.float32",
"bytes": 1280
},
"encoder.downsamples.1.downsamples.1.residual.3.gamma": {
"shape": [
320,
1,
1,
1
],
"dtype": "torch.float32",
"bytes": 1280
},
"encoder.downsamples.1.downsamples.1.residual.6.weight": {
"shape": [
320,
320,
3,
3,
3
],
"dtype": "torch.float32",
"bytes": 11059200
},
"encoder.downsamples.1.downsamples.1.residual.6.bias": {
"shape": [
320
],
"dtype": "torch.float32",
"bytes": 1280
},
"encoder.downsamples.1.downsamples.2.resample.1.weight": {
"shape": [
320,
320,
3,
3
],
"dtype": "torch.float32",
"bytes": 3686400
},
"encoder.downsamples.1.downsamples.2.resample.1.bias": {
"shape": [
320
],
"dtype": "torch.float32",
"bytes": 1280
},
"encoder.downsamples.1.downsamples.2.time_conv.weight": {
"shape": [
320,
320,
3,
1,
1
],
"dtype": "torch.float32",
"bytes": 1228800
},
"encoder.downsamples.1.downsamples.2.time_conv.bias": {
"shape": [
320
],
"dtype": "torch.float32",
"bytes": 1280
},
"encoder.downsamples.2.downsamples.0.residual.0.gamma": {
"shape": [
320,
1,
1,
1
],
"dtype": "torch.float32",
"bytes": 1280
},
"encoder.downsamples.2.downsamples.0.residual.2.weight": {
"shape": [
640,
320,
3,
3,
3
],
"dtype": "torch.float32",
"bytes": 22118400
},
"encoder.downsamples.2.downsamples.0.residual.2.bias": {
"shape": [
640
],
"dtype": "torch.float32",
"bytes": 2560
},
"encoder.downsamples.2.downsamples.0.residual.3.gamma": {
"shape": [
640,
1,
1,
1
],
"dtype": "torch.float32",
"bytes": 2560
},
"encoder.downsamples.2.downsamples.0.residual.6.weight": {
"shape": [
640,
640,
3,
3,
3
],
"dtype": "torch.float32",
"bytes": 44236800
},
"encoder.downsamples.2.downsamples.0.residual.6.bias": {
"shape": [
640
],
"dtype": "torch.float32",
"bytes": 2560
},
"encoder.downsamples.2.downsamples.0.shortcut.weight": {
"shape": [
640,
320,
1,
1,
1
],
"dtype": "torch.float32",
"bytes": 819200
},
"encoder.downsamples.2.downsamples.0.shortcut.bias": {
"shape": [
640
],
"dtype": "torch.float32",
"bytes": 2560
},
"encoder.downsamples.2.downsamples.1.residual.0.gamma": {
"shape": [
640,
1,
1,
1
],
"dtype": "torch.float32",
"bytes": 2560
},
"encoder.downsamples.2.downsamples.1.residual.2.weight": {
"shape": [
640,
640,
3,
3,
3
],
"dtype": "torch.float32",
"bytes": 44236800
},
"encoder.downsamples.2.downsamples.1.residual.2.bias": {
"shape": [
640
],
"dtype": "torch.float32",
"bytes": 2560
},
"encoder.downsamples.2.downsamples.1.residual.3.gamma": {
"shape": [
640,
1,
1,
1
],
"dtype": "torch.float32",
"bytes": 2560
},
"encoder.downsamples.2.downsamples.1.residual.6.weight": {
"shape": [
640,
640,
3,
3,
3
],
"dtype": "torch.float32",
"bytes": 44236800
},
"encoder.downsamples.2.downsamples.1.residual.6.bias": {
"shape": [
640
],
"dtype": "torch.float32",
"bytes": 2560
},
"encoder.downsamples.2.downsamples.2.resample.1.weight": {
"shape": [
640,
640,
3,
3
],
"dtype": "torch.float32",
"bytes": 14745600
},
"encoder.downsamples.2.downsamples.2.resample.1.bias": {
"shape": [
640
],
"dtype": "torch.float32",
"bytes": 2560
},
"encoder.downsamples.2.downsamples.2.time_conv.weight": {
"shape": [
640,
640,
3,
1,
1
],
"dtype": "torch.float32",
"bytes": 4915200
},
"encoder.downsamples.2.downsamples.2.time_conv.bias": {
"shape": [
640
],
"dtype": "torch.float32",
"bytes": 2560
},
"encoder.downsamples.3.downsamples.0.residual.0.gamma": {
"shape": [
640,
1,
1,
1
],
"dtype": "torch.float32",
"bytes": 2560
},
"encoder.downsamples.3.downsamples.0.residual.2.weight": {
"shape": [
640,
640,
3,
3,
3
],
"dtype": "torch.float32",
"bytes": 44236800
},
"encoder.downsamples.3.downsamples.0.residual.2.bias": {
"shape": [
640
],
"dtype": "torch.float32",
"bytes": 2560
},
"encoder.downsamples.3.downsamples.0.residual.3.gamma": {
"shape": [
640,
1,
1,
1
],
"dtype": "torch.float32",
"bytes": 2560
},
"encoder.downsamples.3.downsamples.0.residual.6.weight": {
"shape": [
640,
640,
3,
3,
3
],
"dtype": "torch.float32",
"bytes": 44236800
},
"encoder.downsamples.3.downsamples.0.residual.6.bias": {
"shape": [
640
],
"dtype": "torch.float32",
"bytes": 2560
},
"encoder.downsamples.3.downsamples.1.residual.0.gamma": {
"shape": [
640,
1,
1,
1
],
"dtype": "torch.float32",
"bytes": 2560
},
"encoder.downsamples.3.downsamples.1.residual.2.weight": {
"shape": [
640,
640,
3,
3,
3
],
"dtype": "torch.float32",
"bytes": 44236800
},
"encoder.downsamples.3.downsamples.1.residual.2.bias": {
"shape": [
640
],
"dtype": "torch.float32",
"bytes": 2560
},
"encoder.downsamples.3.downsamples.1.residual.3.gamma": {
"shape": [
640,
1,
1,
1
],
"dtype": "torch.float32",
"bytes": 2560
},
"encoder.downsamples.3.downsamples.1.residual.6.weight": {
"shape": [
640,
640,
3,
3,
3
],
"dtype": "torch.float32",
"bytes": 44236800
},
"encoder.downsamples.3.downsamples.1.residual.6.bias": {
"shape": [
640
],
"dtype": "torch.float32",
"bytes": 2560
},
"encoder.middle.0.residual.0.gamma": {
"shape": [
640,
1,
1,
1
],
"dtype": "torch.float32",
"bytes": 2560
},
"encoder.middle.0.residual.2.weight": {
"shape": [
640,
640,
3,
3,
3
],
"dtype": "torch.float32",
"bytes": 44236800
},
"encoder.middle.0.residual.2.bias": {
"shape": [
640
],
"dtype": "torch.float32",
"bytes": 2560
},
"encoder.middle.0.residual.3.gamma": {
"shape": [
640,
1,
1,
1
],
"dtype": "torch.float32",
"bytes": 2560
},
"encoder.middle.0.residual.6.weight": {
"shape": [
640,
640,
3,
3,
3
],
"dtype": "torch.float32",
"bytes": 44236800
},
"encoder.middle.0.residual.6.bias": {
"shape": [
640
],
"dtype": "torch.float32",
"bytes": 2560
},
"encoder.middle.1.norm.gamma": {
"shape": [
640,
1,
1
],
"dtype": "torch.float32",
"bytes": 2560
},
"encoder.middle.1.to_qkv.weight": {
"shape": [
1920,
640,
1,
1
],
"dtype": "torch.float32",
"bytes": 4915200
},
"encoder.middle.1.to_qkv.bias": {
"shape": [
1920
],
"dtype": "torch.float32",
"bytes": 7680
},
"encoder.middle.1.proj.weight": {
"shape": [
640,
640,
1,
1
],
"dtype": "torch.float32",
"bytes": 1638400
},
"encoder.middle.1.proj.bias": {
"shape": [
640
],
"dtype": "torch.float32",
"bytes": 2560
},
"encoder.middle.2.residual.0.gamma": {
"shape": [
640,
1,
1,
1
],
"dtype": "torch.float32",
"bytes": 2560
},
"encoder.middle.2.residual.2.weight": {
"shape": [
640,
640,
3,
3,
3
],
"dtype": "torch.float32",
"bytes": 44236800
},
"encoder.middle.2.residual.2.bias": {
"shape": [
640
],
"dtype": "torch.float32",
"bytes": 2560
},
"encoder.middle.2.residual.3.gamma": {
"shape": [
640,
1,
1,
1
],
"dtype": "torch.float32",
"bytes": 2560
},
"encoder.middle.2.residual.6.weight": {
"shape": [
640,
640,
3,
3,
3
],
"dtype": "torch.float32",
"bytes": 44236800
},
"encoder.middle.2.residual.6.bias": {
"shape": [
640
],
"dtype": "torch.float32",
"bytes": 2560
},
"encoder.head.0.gamma": {
"shape": [
640,
1,
1,
1
],
"dtype": "torch.float32",
"bytes": 2560
},
"encoder.head.2.weight": {
"shape": [
96,
640,
3,
3,
3
],
"dtype": "torch.float32",
"bytes": 6635520
},
"encoder.head.2.bias": {
"shape": [
96
],
"dtype": "torch.float32",
"bytes": 384
},
"conv1.weight": {
"shape": [
96,
96,
1,
1,
1
],
"dtype": "torch.float32",
"bytes": 36864
},
"conv1.bias": {
"shape": [
96
],
"dtype": "torch.float32",
"bytes": 384
},
"conv2.weight": {
"shape": [
48,
48,
1,
1,
1
],
"dtype": "torch.float32",
"bytes": 9216
},
"conv2.bias": {
"shape": [
48
],
"dtype": "torch.float32",
"bytes": 192
},
"decoder.conv1.weight": {
"shape": [
1024,
48,
3,
3,
3
],
"dtype": "torch.float32",
"bytes": 5308416
},
"decoder.conv1.bias": {
"shape": [
1024
],
"dtype": "torch.float32",
"bytes": 4096
},
"decoder.middle.0.residual.0.gamma": {
"shape": [
1024,
1,
1,
1
],
"dtype": "torch.float32",
"bytes": 4096
},
"decoder.middle.0.residual.2.weight": {
"shape": [
1024,
1024,
3,
3,
3
],
"dtype": "torch.float32",
"bytes": 113246208
},
"decoder.middle.0.residual.2.bias": {
"shape": [
1024
],
"dtype": "torch.float32",
"bytes": 4096
},
"decoder.middle.0.residual.3.gamma": {
"shape": [
1024,
1,
1,
1
],
"dtype": "torch.float32",
"bytes": 4096
},
"decoder.middle.0.residual.6.weight": {
"shape": [
1024,
1024,
3,
3,
3
],
"dtype": "torch.float32",
"bytes": 113246208
},
"decoder.middle.0.residual.6.bias": {
"shape": [
1024
],
"dtype": "torch.float32",
"bytes": 4096
},
"decoder.middle.1.norm.gamma": {
"shape": [
1024,
1,
1
],
"dtype": "torch.float32",
"bytes": 4096
},
"decoder.middle.1.to_qkv.weight": {
"shape": [
3072,
1024,
1,
1
],
"dtype": "torch.float32",
"bytes": 12582912
},
"decoder.middle.1.to_qkv.bias": {
"shape": [
3072
],
"dtype": "torch.float32",
"bytes": 12288
},
"decoder.middle.1.proj.weight": {
"shape": [
1024,
1024,
1,
1
],
"dtype": "torch.float32",
"bytes": 4194304
},
"decoder.middle.1.proj.bias": {
"shape": [
1024
],
"dtype": "torch.float32",
"bytes": 4096
},
"decoder.middle.2.residual.0.gamma": {
"shape": [
1024,
1,
1,
1
],
"dtype": "torch.float32",
"bytes": 4096
},
"decoder.middle.2.residual.2.weight": {
"shape": [
1024,
1024,
3,
3,
3
],
"dtype": "torch.float32",
"bytes": 113246208
},
"decoder.middle.2.residual.2.bias": {
"shape": [
1024
],
"dtype": "torch.float32",
"bytes": 4096
},
"decoder.middle.2.residual.3.gamma": {
"shape": [
1024,
1,
1,
1
],
"dtype": "torch.float32",
"bytes": 4096
},
"decoder.middle.2.residual.6.weight": {
"shape": [
1024,
1024,
3,
3,
3
],
"dtype": "torch.float32",
"bytes": 113246208
},
"decoder.middle.2.residual.6.bias": {
"shape": [
1024
],
"dtype": "torch.float32",
"bytes": 4096
},
"decoder.upsamples.0.upsamples.0.residual.0.gamma": {
"shape": [
1024,
1,
1,
1
],
"dtype": "torch.float32",
"bytes": 4096
},
"decoder.upsamples.0.upsamples.0.residual.2.weight": {
"shape": [
1024,
1024,
3,
3,
3
],
"dtype": "torch.float32",
"bytes": 113246208
},
"decoder.upsamples.0.upsamples.0.residual.2.bias": {
"shape": [
1024
],
"dtype": "torch.float32",
"bytes": 4096
},
"decoder.upsamples.0.upsamples.0.residual.3.gamma": {
"shape": [
1024,
1,
1,
1
],
"dtype": "torch.float32",
"bytes": 4096
},
"decoder.upsamples.0.upsamples.0.residual.6.weight": {
"shape": [
1024,
1024,
3,
3,
3
],
"dtype": "torch.float32",
"bytes": 113246208
},
"decoder.upsamples.0.upsamples.0.residual.6.bias": {
"shape": [
1024
],
"dtype": "torch.float32",
"bytes": 4096
},
"decoder.upsamples.0.upsamples.1.residual.0.gamma": {
"shape": [
1024,
1,
1,
1
],
"dtype": "torch.float32",
"bytes": 4096
},
"decoder.upsamples.0.upsamples.1.residual.2.weight": {
"shape": [
1024,
1024,
3,
3,
3
],
"dtype": "torch.float32",
"bytes": 113246208
},
"decoder.upsamples.0.upsamples.1.residual.2.bias": {
"shape": [
1024
],
"dtype": "torch.float32",
"bytes": 4096
},
"decoder.upsamples.0.upsamples.1.residual.3.gamma": {
"shape": [
1024,
1,
1,
1
],
"dtype": "torch.float32",
"bytes": 4096
},
"decoder.upsamples.0.upsamples.1.residual.6.weight": {
"shape": [
1024,
1024,
3,
3,
3
],
"dtype": "torch.float32",
"bytes": 113246208
},
"decoder.upsamples.0.upsamples.1.residual.6.bias": {
"shape": [
1024
],
"dtype": "torch.float32",
"bytes": 4096
},
"decoder.upsamples.0.upsamples.2.residual.0.gamma": {
"shape": [
1024,
1,
1,
1
],
"dtype": "torch.float32",
"bytes": 4096
},
"decoder.upsamples.0.upsamples.2.residual.2.weight": {
"shape": [
1024,
1024,
3,
3,
3
],
"dtype": "torch.float32",
"bytes": 113246208
},
"decoder.upsamples.0.upsamples.2.residual.2.bias": {
"shape": [
1024
],
"dtype": "torch.float32",
"bytes": 4096
},
"decoder.upsamples.0.upsamples.2.residual.3.gamma": {
"shape": [
1024,
1,
1,
1
],
"dtype": "torch.float32",
"bytes": 4096
},
"decoder.upsamples.0.upsamples.2.residual.6.weight": {
"shape": [
1024,
1024,
3,
3,
3
],
"dtype": "torch.float32",
"bytes": 113246208
},
"decoder.upsamples.0.upsamples.2.residual.6.bias": {
"shape": [
1024
],
"dtype": "torch.float32",
"bytes": 4096
},
"decoder.upsamples.0.upsamples.3.resample.1.weight": {
"shape": [
1024,
1024,
3,
3
],
"dtype": "torch.float32",
"bytes": 37748736
},
"decoder.upsamples.0.upsamples.3.resample.1.bias": {
"shape": [
1024
],
"dtype": "torch.float32",
"bytes": 4096
},
"decoder.upsamples.0.upsamples.3.time_conv.weight": {
"shape": [
2048,
1024,
3,
1,
1
],
"dtype": "torch.float32",
"bytes": 25165824
},
"decoder.upsamples.0.upsamples.3.time_conv.bias": {
"shape": [
2048
],
"dtype": "torch.float32",
"bytes": 8192
},
"decoder.upsamples.1.upsamples.0.residual.0.gamma": {
"shape": [
1024,
1,
1,
1
],
"dtype": "torch.float32",
"bytes": 4096
},
"decoder.upsamples.1.upsamples.0.residual.2.weight": {
"shape": [
1024,
1024,
3,
3,
3
],
"dtype": "torch.float32",
"bytes": 113246208
},
"decoder.upsamples.1.upsamples.0.residual.2.bias": {
"shape": [
1024
],
"dtype": "torch.float32",
"bytes": 4096
},
"decoder.upsamples.1.upsamples.0.residual.3.gamma": {
"shape": [
1024,
1,
1,
1
],
"dtype": "torch.float32",
"bytes": 4096
},
"decoder.upsamples.1.upsamples.0.residual.6.weight": {
"shape": [
1024,
1024,
3,
3,
3
],
"dtype": "torch.float32",
"bytes": 113246208
},
"decoder.upsamples.1.upsamples.0.residual.6.bias": {
"shape": [
1024
],
"dtype": "torch.float32",
"bytes": 4096
},
"decoder.upsamples.1.upsamples.1.residual.0.gamma": {
"shape": [
1024,
1,
1,
1
],
"dtype": "torch.float32",
"bytes": 4096
},
"decoder.upsamples.1.upsamples.1.residual.2.weight": {
"shape": [
1024,
1024,
3,
3,
3
],
"dtype": "torch.float32",
"bytes": 113246208
},
"decoder.upsamples.1.upsamples.1.residual.2.bias": {
"shape": [
1024
],
"dtype": "torch.float32",
"bytes": 4096
},
"decoder.upsamples.1.upsamples.1.residual.3.gamma": {
"shape": [
1024,
1,
1,
1
],
"dtype": "torch.float32",
"bytes": 4096
},
"decoder.upsamples.1.upsamples.1.residual.6.weight": {
"shape": [
1024,
1024,
3,
3,
3
],
"dtype": "torch.float32",
"bytes": 113246208
},
"decoder.upsamples.1.upsamples.1.residual.6.bias": {
"shape": [
1024
],
"dtype": "torch.float32",
"bytes": 4096
},
"decoder.upsamples.1.upsamples.2.residual.0.gamma": {
"shape": [
1024,
1,
1,
1
],
"dtype": "torch.float32",
"bytes": 4096
},
"decoder.upsamples.1.upsamples.2.residual.2.weight": {
"shape": [
1024,
1024,
3,
3,
3
],
"dtype": "torch.float32",
"bytes": 113246208
},
"decoder.upsamples.1.upsamples.2.residual.2.bias": {
"shape": [
1024
],
"dtype": "torch.float32",
"bytes": 4096
},
"decoder.upsamples.1.upsamples.2.residual.3.gamma": {
"shape": [
1024,
1,
1,
1
],
"dtype": "torch.float32",
"bytes": 4096
},
"decoder.upsamples.1.upsamples.2.residual.6.weight": {
"shape": [
1024,
1024,
3,
3,
3
],
"dtype": "torch.float32",
"bytes": 113246208
},
"decoder.upsamples.1.upsamples.2.residual.6.bias": {
"shape": [
1024
],
"dtype": "torch.float32",
"bytes": 4096
},
"decoder.upsamples.1.upsamples.3.resample.1.weight": {
"shape": [
1024,
1024,
3,
3
],
"dtype": "torch.float32",
"bytes": 37748736
},
"decoder.upsamples.1.upsamples.3.resample.1.bias": {
"shape": [
1024
],
"dtype": "torch.float32",
"bytes": 4096
},
"decoder.upsamples.1.upsamples.3.time_conv.weight": {
"shape": [
2048,
1024,
3,
1,
1
],
"dtype": "torch.float32",
"bytes": 25165824
},
"decoder.upsamples.1.upsamples.3.time_conv.bias": {
"shape": [
2048
],
"dtype": "torch.float32",
"bytes": 8192
},
"decoder.upsamples.2.upsamples.0.residual.0.gamma": {
"shape": [
1024,
1,
1,
1
],
"dtype": "torch.float32",
"bytes": 4096
},
"decoder.upsamples.2.upsamples.0.residual.2.weight": {
"shape": [
512,
1024,
3,
3,
3
],
"dtype": "torch.float32",
"bytes": 56623104
},
"decoder.upsamples.2.upsamples.0.residual.2.bias": {
"shape": [
512
],
"dtype": "torch.float32",
"bytes": 2048
},
"decoder.upsamples.2.upsamples.0.residual.3.gamma": {
"shape": [
512,
1,
1,
1
],
"dtype": "torch.float32",
"bytes": 2048
},
"decoder.upsamples.2.upsamples.0.residual.6.weight": {
"shape": [
512,
512,
3,
3,
3
],
"dtype": "torch.float32",
"bytes": 28311552
},
"decoder.upsamples.2.upsamples.0.residual.6.bias": {
"shape": [
512
],
"dtype": "torch.float32",
"bytes": 2048
},
"decoder.upsamples.2.upsamples.0.shortcut.weight": {
"shape": [
512,
1024,
1,
1,
1
],
"dtype": "torch.float32",
"bytes": 2097152
},
"decoder.upsamples.2.upsamples.0.shortcut.bias": {
"shape": [
512
],
"dtype": "torch.float32",
"bytes": 2048
},
"decoder.upsamples.2.upsamples.1.residual.0.gamma": {
"shape": [
512,
1,
1,
1
],
"dtype": "torch.float32",
"bytes": 2048
},
"decoder.upsamples.2.upsamples.1.residual.2.weight": {
"shape": [
512,
512,
3,
3,
3
],
"dtype": "torch.float32",
"bytes": 28311552
},
"decoder.upsamples.2.upsamples.1.residual.2.bias": {
"shape": [
512
],
"dtype": "torch.float32",
"bytes": 2048
},
"decoder.upsamples.2.upsamples.1.residual.3.gamma": {
"shape": [
512,
1,
1,
1
],
"dtype": "torch.float32",
"bytes": 2048
},
"decoder.upsamples.2.upsamples.1.residual.6.weight": {
"shape": [
512,
512,
3,
3,
3
],
"dtype": "torch.float32",
"bytes": 28311552
},
"decoder.upsamples.2.upsamples.1.residual.6.bias": {
"shape": [
512
],
"dtype": "torch.float32",
"bytes": 2048
},
"decoder.upsamples.2.upsamples.2.residual.0.gamma": {
"shape": [
512,
1,
1,
1
],
"dtype": "torch.float32",
"bytes": 2048
},
"decoder.upsamples.2.upsamples.2.residual.2.weight": {
"shape": [
512,
512,
3,
3,
3
],
"dtype": "torch.float32",
"bytes": 28311552
},
"decoder.upsamples.2.upsamples.2.residual.2.bias": {
"shape": [
512
],
"dtype": "torch.float32",
"bytes": 2048
},
"decoder.upsamples.2.upsamples.2.residual.3.gamma": {
"shape": [
512,
1,
1,
1
],
"dtype": "torch.float32",
"bytes": 2048
},
"decoder.upsamples.2.upsamples.2.residual.6.weight": {
"shape": [
512,
512,
3,
3,
3
],
"dtype": "torch.float32",
"bytes": 28311552
},
"decoder.upsamples.2.upsamples.2.residual.6.bias": {
"shape": [
512
],
"dtype": "torch.float32",
"bytes": 2048
},
"decoder.upsamples.2.upsamples.3.resample.1.weight": {
"shape": [
512,
512,
3,
3
],
"dtype": "torch.float32",
"bytes": 9437184
},
"decoder.upsamples.2.upsamples.3.resample.1.bias": {
"shape": [
512
],
"dtype": "torch.float32",
"bytes": 2048
},
"decoder.upsamples.3.upsamples.0.residual.0.gamma": {
"shape": [
512,
1,
1,
1
],
"dtype": "torch.float32",
"bytes": 2048
},
"decoder.upsamples.3.upsamples.0.residual.2.weight": {
"shape": [
256,
512,
3,
3,
3
],
"dtype": "torch.float32",
"bytes": 14155776
},
"decoder.upsamples.3.upsamples.0.residual.2.bias": {
"shape": [
256
],
"dtype": "torch.float32",
"bytes": 1024
},
"decoder.upsamples.3.upsamples.0.residual.3.gamma": {
"shape": [
256,
1,
1,
1
],
"dtype": "torch.float32",
"bytes": 1024
},
"decoder.upsamples.3.upsamples.0.residual.6.weight": {
"shape": [
256,
256,
3,
3,
3
],
"dtype": "torch.float32",
"bytes": 7077888
},
"decoder.upsamples.3.upsamples.0.residual.6.bias": {
"shape": [
256
],
"dtype": "torch.float32",
"bytes": 1024
},
"decoder.upsamples.3.upsamples.0.shortcut.weight": {
"shape": [
256,
512,
1,
1,
1
],
"dtype": "torch.float32",
"bytes": 524288
},
"decoder.upsamples.3.upsamples.0.shortcut.bias": {
"shape": [
256
],
"dtype": "torch.float32",
"bytes": 1024
},
"decoder.upsamples.3.upsamples.1.residual.0.gamma": {
"shape": [
256,
1,
1,
1
],
"dtype": "torch.float32",
"bytes": 1024
},
"decoder.upsamples.3.upsamples.1.residual.2.weight": {
"shape": [
256,
256,
3,
3,
3
],
"dtype": "torch.float32",
"bytes": 7077888
},
"decoder.upsamples.3.upsamples.1.residual.2.bias": {
"shape": [
256
],
"dtype": "torch.float32",
"bytes": 1024
},
"decoder.upsamples.3.upsamples.1.residual.3.gamma": {
"shape": [
256,
1,
1,
1
],
"dtype": "torch.float32",
"bytes": 1024
},
"decoder.upsamples.3.upsamples.1.residual.6.weight": {
"shape": [
256,
256,
3,
3,
3
],
"dtype": "torch.float32",
"bytes": 7077888
},
"decoder.upsamples.3.upsamples.1.residual.6.bias": {
"shape": [
256
],
"dtype": "torch.float32",
"bytes": 1024
},
"decoder.upsamples.3.upsamples.2.residual.0.gamma": {
"shape": [
256,
1,
1,
1
],
"dtype": "torch.float32",
"bytes": 1024
},
"decoder.upsamples.3.upsamples.2.residual.2.weight": {
"shape": [
256,
256,
3,
3,
3
],
"dtype": "torch.float32",
"bytes": 7077888
},
"decoder.upsamples.3.upsamples.2.residual.2.bias": {
"shape": [
256
],
"dtype": "torch.float32",
"bytes": 1024
},
"decoder.upsamples.3.upsamples.2.residual.3.gamma": {
"shape": [
256,
1,
1,
1
],
"dtype": "torch.float32",
"bytes": 1024
},
"decoder.upsamples.3.upsamples.2.residual.6.weight": {
"shape": [
256,
256,
3,
3,
3
],
"dtype": "torch.float32",
"bytes": 7077888
},
"decoder.upsamples.3.upsamples.2.residual.6.bias": {
"shape": [
256
],
"dtype": "torch.float32",
"bytes": 1024
},
"decoder.head.0.gamma": {
"shape": [
256,
1,
1,
1
],
"dtype": "torch.float32",
"bytes": 1024
},
"decoder.head.2.weight": {
"shape": [
12,
256,
3,
3,
3
],
"dtype": "torch.float32",
"bytes": 331776
},
"decoder.head.2.bias": {
"shape": [
12
],
"dtype": "torch.float32",
"bytes": 48
}
}