File size: 1,572 Bytes
91f26ad |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 |
{
"_class_name": "AutoencoderDC",
"_diffusers_version": "0.32.0.dev0",
"attention_head_dim": 32,
"decoder_act_fns": [
"relu",
"relu",
"relu",
"silu",
"silu",
"silu",
"silu",
"silu"
],
"decoder_block_out_channels": [
128,
256,
512,
512,
1024,
1024,
2048,
2048
],
"decoder_block_types": [
"ResBlock",
"ResBlock",
"ResBlock",
"EfficientViTBlock",
"EfficientViTBlock",
"EfficientViTBlock",
"EfficientViTBlock",
"EfficientViTBlock"
],
"decoder_layers_per_block": [
0,
5,
10,
2,
2,
2,
2,
2
],
"decoder_norm_types": [
"batch_norm",
"batch_norm",
"batch_norm",
"rms_norm",
"rms_norm",
"rms_norm",
"rms_norm",
"rms_norm"
],
"decoder_qkv_multiscales": [
[],
[],
[],
[],
[],
[],
[],
[]
],
"downsample_block_type": "pixel_unshuffle",
"encoder_block_out_channels": [
128,
256,
512,
512,
1024,
1024,
2048,
2048
],
"encoder_block_types": [
"ResBlock",
"ResBlock",
"ResBlock",
"EfficientViTBlock",
"EfficientViTBlock",
"EfficientViTBlock",
"EfficientViTBlock",
"EfficientViTBlock"
],
"encoder_layers_per_block": [
0,
4,
8,
2,
2,
2,
2,
2
],
"encoder_qkv_multiscales": [
[],
[],
[],
[],
[],
[],
[],
[]
],
"in_channels": 3,
"latent_channels": 512,
"scaling_factor": 0.4883,
"upsample_block_type": "pixel_shuffle"
}
|