{ | |
"_class_name": "VQModel", | |
"_diffusers_version": "0.4.0.dev0", | |
"act_fn": "swish", | |
"block_out_channels": [ | |
128, | |
128, | |
256, | |
256, | |
512 | |
], | |
"down_block_types": [ | |
"DownEncoderBlock2D", | |
"DownEncoderBlock2D", | |
"DownEncoderBlock2D", | |
"DownEncoderBlock2D", | |
"AttnDownEncoderBlock2D" | |
], | |
"final_decoder_activation": false, | |
"final_encoder_activation": false, | |
"in_channels": 3, | |
"latent_channels": 256, | |
"layers_per_block": 2, | |
"num_vq_embeddings": 1024, | |
"out_channels": 3, | |
"sample_size": 32, | |
"up_block_types": [ | |
"AttnUpDecoderBlock2D", | |
"UpDecoderBlock2D", | |
"UpDecoderBlock2D", | |
"UpDecoderBlock2D", | |
"UpDecoderBlock2D" | |
] | |
} | |