{ | |
"_class_name": "AutoencoderKL", | |
"_commit_hash": null, | |
"_diffusers_version": "0.23.0", | |
"_name_or_path": "/home/ubuntu/.cache/huggingface/hub/models--playgroundai--playground-v2-512px-base/snapshots/fe6a2b202ba7e6144b52da383fdd1e0a1e6f9628/vae", | |
"act_fn": "silu", | |
"block_out_channels": [ | |
128, | |
256, | |
512, | |
512 | |
], | |
"down_block_types": [ | |
"DownEncoderBlock2D", | |
"DownEncoderBlock2D", | |
"DownEncoderBlock2D", | |
"DownEncoderBlock2D" | |
], | |
"force_upcast": true, | |
"in_channels": 3, | |
"latent_channels": 4, | |
"layers_per_block": 2, | |
"neuron": { | |
"auto_cast": "matmul", | |
"auto_cast_type": "bf16", | |
"compiler_type": "neuronx-cc", | |
"compiler_version": "2.11.0.34+c5231f848", | |
"dynamic_batch_size": false, | |
"input_names": [ | |
"sample" | |
], | |
"model_type": "vae-encoder", | |
"output_attentions": false, | |
"output_hidden_states": false, | |
"output_names": [ | |
"latent_sample" | |
], | |
"static_batch_size": 1, | |
"static_height": 512, | |
"static_num_beams": 1, | |
"static_num_channels": 3, | |
"static_width": 512 | |
}, | |
"norm_num_groups": 32, | |
"out_channels": 3, | |
"sample_size": 1024, | |
"scaling_factor": 0.13025, | |
"task": "semantic-segmentation", | |
"transformers_version": null, | |
"up_block_types": [ | |
"UpDecoderBlock2D", | |
"UpDecoderBlock2D", | |
"UpDecoderBlock2D", | |
"UpDecoderBlock2D" | |
] | |
} | |