vae/config.json

{
  "_class_name": "WFVAEModel",
  "_diffusers_version": "0.28.0",
  "base_channels": 128,
  "decoder_energy_flow_hidden_size": 128,
  "decoder_num_resblocks": 2,
  "dropout": 0.0,
  "encoder_energy_flow_hidden_size": 64,
  "encoder_num_resblocks": 2,
  "latent_dim": 8,
  "use_attention": true,
  "norm_type": "layernorm",
  "t_interpolation": "trilinear",
  "connect_res_layer_num": 2
}