point_map_vae/config.json

{
  "_class_name": "PMapAutoencoderKLTemporalDecoder",
  "_diffusers_version": "0.31.0",
  "_name_or_path": "/dockerdata/workspace/pmap_vae_hybrid_synthetic_image_320_640_train_fp32_l_patch/checkpoint-40000/",
  "dec_block_out_channels": [
    128,
    256,
    512,
    512
  ],
  "dec_layers_per_block": 2,
  "enc_block_out_channels": [
    128,
    256,
    512,
    512
  ],
  "enc_down_block_types": [
    "DownEncoderBlock2D",
    "DownEncoderBlock2D",
    "DownEncoderBlock2D",
    "DownEncoderBlock2D"
  ],
  "enc_layers_per_block": 2,
  "in_channels": 4,
  "latent_channels": 4,
  "mid_block_add_attention": true,
  "offset_scale_factor": 0.1
}