WFVAE2_18832_slim/config.json

{
  "_class_name": "WFVAE2Model",
  "_diffusers_version": "0.30.2",
  "base_channels": [
    128,
    256,
    512
  ],
  "decoder_energy_flow_size": 128,
  "decoder_num_resblocks": 8,
  "dropout": 0.0,
  "encoder_energy_flow_size": 128,
  "encoder_num_resblocks": 8,
  "input_type": "image",
  "latent_dim": 32,
  "layer_res_type": [
    "ResnetBlock2D",
    "ResnetBlock2D"
  ],
  "layer_type": [
    "hw",
    "hw"
  ],
  "mid_layers_type": [
    "ResnetBlock2D",
    "ResnetBlock2D"
  ],
  "norm_type": "rmsnorm",
  "scale": [
    0.6836,
    0.4297,
    1.6484,
    1.3672,
    1.4688,
    1.5547,
    0.9688,
    0.7266,
    1.2109,
    0.4844,
    0.7070,
    1.0391,
    0.8242,
    1.9219,
    1.3047,
    1.5156,
    1.4609,
    2.1562,
    0.8945,
    1.4062,
    0.7578,
    0.7695,
    1.1875,
    0.5938,
    0.8945,
    1.3438,
    1.0469,
    1.1406,
    1.7812,
    2.3594,
    1.1953,
    1.0391
  ],
  "shift": [
    1.0703,
    -0.3047,
    -0.0118,
    0.5430,
    -0.1562,
    -0.1060,
    0.0491,
    -0.2910,
    0.8047,
    0.1641,
    -1.0547,
    0.1514,
    -0.7969,
    -0.0098,
    0.5703,
    -0.0640,
    0.0308,
    0.1221,
    0.7148,
    0.0442,
    -0.2334,
    -0.1074,
    -0.1709,
    -0.9141,
    -0.4062,
    -0.6445,
    0.3867,
    0.5625,
    0.1377,
    -0.3457,
    -0.5156,
    1.7188
  ],
  "t_interpolation": "trilinear",
  "use_attention": true
}