depth_vae/config.json

{
  "_class_name": "AutoencoderTiny",
  "_diffusers_version": "0.32.0",
  "_name_or_path": "/alluxio/training/experiments/zhenqing/diffsplat/tinyvae-ckpt-wconf-016000",
  "act_fn": "relu",
  "block_out_channels": [
    64,
    64,
    64,
    64
  ],
  "decoder_block_out_channels": [
    64,
    64,
    64,
    64
  ],
  "encoder_block_out_channels": [
    64,
    64,
    64,
    64
  ],
  "force_upcast": false,
  "in_channels": 3,
  "latent_channels": 4,
  "latent_magnitude": 3,
  "latent_shift": 0.5,
  "num_decoder_blocks": [
    3,
    3,
    3,
    1
  ],
  "num_encoder_blocks": [
    1,
    3,
    3,
    3
  ],
  "out_channels": 4,
  "scaling_factor": 1.0,
  "shift_factor": 0.0,
  "upsample_fn": "nearest",
  "upsampling_scaling_factor": 2
}