[
  "decoder.conv_in.bias",
  "decoder.conv_in.weight",
  "decoder.conv_out.bias",
  "decoder.conv_out.weight",
  "decoder.mid.attn_1.k.bias",
  "decoder.mid.attn_1.k.weight",
  "decoder.mid.attn_1.norm.bias",
  "decoder.mid.attn_1.norm.weight",
  "decoder.mid.attn_1.proj_out.bias",
  "decoder.mid.attn_1.proj_out.weight",
  "decoder.mid.attn_1.q.bias",
  "decoder.mid.attn_1.q.weight",
  "decoder.mid.attn_1.v.bias",
  "decoder.mid.attn_1.v.weight",
  "decoder.mid.block_1.conv1.bias",
  "decoder.mid.block_1.conv1.weight",
  "decoder.mid.block_1.conv2.bias",
  "decoder.mid.block_1.conv2.weight",
  "decoder.mid.block_1.norm1.bias",
  "decoder.mid.block_1.norm1.weight",
  "decoder.mid.block_1.norm2.bias",
  "decoder.mid.block_1.norm2.weight",
  "decoder.mid.block_2.conv1.bias",
  "decoder.mid.block_2.conv1.weight",
  "decoder.mid.block_2.conv2.bias",
  "decoder.mid.block_2.conv2.weight",
  "decoder.mid.block_2.norm1.bias",
  "decoder.mid.block_2.norm1.weight",
  "decoder.mid.block_2.norm2.bias",
  "decoder.mid.block_2.norm2.weight",
  "decoder.norm_out.bias",
  "decoder.norm_out.weight",
  "decoder.up.0.block.0.conv1.bias",
  "decoder.up.0.block.0.conv1.weight",
  "decoder.up.0.block.0.conv2.bias",
  "decoder.up.0.block.0.conv2.weight",
  "decoder.up.0.block.0.nin_shortcut.bias",
  "decoder.up.0.block.0.nin_shortcut.weight",
  "decoder.up.0.block.0.norm1.bias",
  "decoder.up.0.block.0.norm1.weight",
  "decoder.up.0.block.0.norm2.bias",
  "decoder.up.0.block.0.norm2.weight",
  "decoder.up.0.block.1.conv1.bias",
  "decoder.up.0.block.1.conv1.weight",
  "decoder.up.0.block.1.conv2.bias",
  "decoder.up.0.block.1.conv2.weight",
  "decoder.up.0.block.1.norm1.bias",
  "decoder.up.0.block.1.norm1.weight",
  "decoder.up.0.block.1.norm2.bias",
  "decoder.up.0.block.1.norm2.weight",
  "decoder.up.0.block.2.conv1.bias",
  "decoder.up.0.block.2.conv1.weight",
  "decoder.up.0.block.2.conv2.bias",
  "decoder.up.0.block.2.conv2.weight",
  "decoder.up.0.block.2.norm1.bias",
  "decoder.up.0.block.2.norm1.weight",
  "decoder.up.0.block.2.norm2.bias",
  "decoder.up.0.block.2.norm2.weight",
  "decoder.up.1.block.0.conv1.bias",
  "decoder.up.1.block.0.conv1.weight",
  "decoder.up.1.block.0.conv2.bias",
  "decoder.up.1.block.0.conv2.weight",
  "decoder.up.1.block.0.nin_shortcut.bias",
  "decoder.up.1.block.0.nin_shortcut.weight",
  "decoder.up.1.block.0.norm1.bias",
  "decoder.up.1.block.0.norm1.weight",
  "decoder.up.1.block.0.norm2.bias",
  "decoder.up.1.block.0.norm2.weight",
  "decoder.up.1.block.1.conv1.bias",
  "decoder.up.1.block.1.conv1.weight",
  "decoder.up.1.block.1.conv2.bias",
  "decoder.up.1.block.1.conv2.weight",
  "decoder.up.1.block.1.norm1.bias",
  "decoder.up.1.block.1.norm1.weight",
  "decoder.up.1.block.1.norm2.bias",
  "decoder.up.1.block.1.norm2.weight",
  "decoder.up.1.block.2.conv1.bias",
  "decoder.up.1.block.2.conv1.weight",
  "decoder.up.1.block.2.conv2.bias",
  "decoder.up.1.block.2.conv2.weight",
  "decoder.up.1.block.2.norm1.bias",
  "decoder.up.1.block.2.norm1.weight",
  "decoder.up.1.block.2.norm2.bias",
  "decoder.up.1.block.2.norm2.weight",
  "decoder.up.1.upsample.conv.bias",
  "decoder.up.1.upsample.conv.weight",
  "decoder.up.2.block.0.conv1.bias",
  "decoder.up.2.block.0.conv1.weight",
  "decoder.up.2.block.0.conv2.bias",
  "decoder.up.2.block.0.conv2.weight",
  "decoder.up.2.block.0.norm1.bias",
  "decoder.up.2.block.0.norm1.weight",
  "decoder.up.2.block.0.norm2.bias",
  "decoder.up.2.block.0.norm2.weight",
  "decoder.up.2.block.1.conv1.bias",
  "decoder.up.2.block.1.conv1.weight",
  "decoder.up.2.block.1.conv2.bias",
  "decoder.up.2.block.1.conv2.weight",
  "decoder.up.2.block.1.norm1.bias",
  "decoder.up.2.block.1.norm1.weight",
  "decoder.up.2.block.1.norm2.bias",
  "decoder.up.2.block.1.norm2.weight",
  "decoder.up.2.block.2.conv1.bias",
  "decoder.up.2.block.2.conv1.weight",
  "decoder.up.2.block.2.conv2.bias",
  "decoder.up.2.block.2.conv2.weight",
  "decoder.up.2.block.2.norm1.bias",
  "decoder.up.2.block.2.norm1.weight",
  "decoder.up.2.block.2.norm2.bias",
  "decoder.up.2.block.2.norm2.weight",
  "decoder.up.2.upsample.conv.bias",
  "decoder.up.2.upsample.conv.weight",
  "decoder.up.3.block.0.conv1.bias",
  "decoder.up.3.block.0.conv1.weight",
  "decoder.up.3.block.0.conv2.bias",
  "decoder.up.3.block.0.conv2.weight",
  "decoder.up.3.block.0.norm1.bias",
  "decoder.up.3.block.0.norm1.weight",
  "decoder.up.3.block.0.norm2.bias",
  "decoder.up.3.block.0.norm2.weight",
  "decoder.up.3.block.1.conv1.bias",
  "decoder.up.3.block.1.conv1.weight",
  "decoder.up.3.block.1.conv2.bias",
  "decoder.up.3.block.1.conv2.weight",
  "decoder.up.3.block.1.norm1.bias",
  "decoder.up.3.block.1.norm1.weight",
  "decoder.up.3.block.1.norm2.bias",
  "decoder.up.3.block.1.norm2.weight",
  "decoder.up.3.block.2.conv1.bias",
  "decoder.up.3.block.2.conv1.weight",
  "decoder.up.3.block.2.conv2.bias",
  "decoder.up.3.block.2.conv2.weight",
  "decoder.up.3.block.2.norm1.bias",
  "decoder.up.3.block.2.norm1.weight",
  "decoder.up.3.block.2.norm2.bias",
  "decoder.up.3.block.2.norm2.weight",
  "decoder.up.3.upsample.conv.bias",
  "decoder.up.3.upsample.conv.weight",
  "encoder.conv_in.bias",
  "encoder.conv_in.weight",
  "encoder.conv_out.bias",
  "encoder.conv_out.weight",
  "encoder.down.0.block.0.conv1.bias",
  "encoder.down.0.block.0.conv1.weight",
  "encoder.down.0.block.0.conv2.bias",
  "encoder.down.0.block.0.conv2.weight",
  "encoder.down.0.block.0.norm1.bias",
  "encoder.down.0.block.0.norm1.weight",
  "encoder.down.0.block.0.norm2.bias",
  "encoder.down.0.block.0.norm2.weight",
  "encoder.down.0.block.1.conv1.bias",
  "encoder.down.0.block.1.conv1.weight",
  "encoder.down.0.block.1.conv2.bias",
  "encoder.down.0.block.1.conv2.weight",
  "encoder.down.0.block.1.norm1.bias",
  "encoder.down.0.block.1.norm1.weight",
  "encoder.down.0.block.1.norm2.bias",
  "encoder.down.0.block.1.norm2.weight",
  "encoder.down.0.downsample.conv.bias",
  "encoder.down.0.downsample.conv.weight",
  "encoder.down.1.block.0.conv1.bias",
  "encoder.down.1.block.0.conv1.weight",
  "encoder.down.1.block.0.conv2.bias",
  "encoder.down.1.block.0.conv2.weight",
  "encoder.down.1.block.0.nin_shortcut.bias",
  "encoder.down.1.block.0.nin_shortcut.weight",
  "encoder.down.1.block.0.norm1.bias",
  "encoder.down.1.block.0.norm1.weight",
  "encoder.down.1.block.0.norm2.bias",
  "encoder.down.1.block.0.norm2.weight",
  "encoder.down.1.block.1.conv1.bias",
  "encoder.down.1.block.1.conv1.weight",
  "encoder.down.1.block.1.conv2.bias",
  "encoder.down.1.block.1.conv2.weight",
  "encoder.down.1.block.1.norm1.bias",
  "encoder.down.1.block.1.norm1.weight",
  "encoder.down.1.block.1.norm2.bias",
  "encoder.down.1.block.1.norm2.weight",
  "encoder.down.1.downsample.conv.bias",
  "encoder.down.1.downsample.conv.weight",
  "encoder.down.2.block.0.conv1.bias",
  "encoder.down.2.block.0.conv1.weight",
  "encoder.down.2.block.0.conv2.bias",
  "encoder.down.2.block.0.conv2.weight",
  "encoder.down.2.block.0.nin_shortcut.bias",
  "encoder.down.2.block.0.nin_shortcut.weight",
  "encoder.down.2.block.0.norm1.bias",
  "encoder.down.2.block.0.norm1.weight",
  "encoder.down.2.block.0.norm2.bias",
  "encoder.down.2.block.0.norm2.weight",
  "encoder.down.2.block.1.conv1.bias",
  "encoder.down.2.block.1.conv1.weight",
  "encoder.down.2.block.1.conv2.bias",
  "encoder.down.2.block.1.conv2.weight",
  "encoder.down.2.block.1.norm1.bias",
  "encoder.down.2.block.1.norm1.weight",
  "encoder.down.2.block.1.norm2.bias",
  "encoder.down.2.block.1.norm2.weight",
  "encoder.down.2.downsample.conv.bias",
  "encoder.down.2.downsample.conv.weight",
  "encoder.down.3.block.0.conv1.bias",
  "encoder.down.3.block.0.conv1.weight",
  "encoder.down.3.block.0.conv2.bias",
  "encoder.down.3.block.0.conv2.weight",
  "encoder.down.3.block.0.norm1.bias",
  "encoder.down.3.block.0.norm1.weight",
  "encoder.down.3.block.0.norm2.bias",
  "encoder.down.3.block.0.norm2.weight",
  "encoder.down.3.block.1.conv1.bias",
  "encoder.down.3.block.1.conv1.weight",
  "encoder.down.3.block.1.conv2.bias",
  "encoder.down.3.block.1.conv2.weight",
  "encoder.down.3.block.1.norm1.bias",
  "encoder.down.3.block.1.norm1.weight",
  "encoder.down.3.block.1.norm2.bias",
  "encoder.down.3.block.1.norm2.weight",
  "encoder.mid.attn_1.k.bias",
  "encoder.mid.attn_1.k.weight",
  "encoder.mid.attn_1.norm.bias",
  "encoder.mid.attn_1.norm.weight",
  "encoder.mid.attn_1.proj_out.bias",
  "encoder.mid.attn_1.proj_out.weight",
  "encoder.mid.attn_1.q.bias",
  "encoder.mid.attn_1.q.weight",
  "encoder.mid.attn_1.v.bias",
  "encoder.mid.attn_1.v.weight",
  "encoder.mid.block_1.conv1.bias",
  "encoder.mid.block_1.conv1.weight",
  "encoder.mid.block_1.conv2.bias",
  "encoder.mid.block_1.conv2.weight",
  "encoder.mid.block_1.norm1.bias",
  "encoder.mid.block_1.norm1.weight",
  "encoder.mid.block_1.norm2.bias",
  "encoder.mid.block_1.norm2.weight",
  "encoder.mid.block_2.conv1.bias",
  "encoder.mid.block_2.conv1.weight",
  "encoder.mid.block_2.conv2.bias",
  "encoder.mid.block_2.conv2.weight",
  "encoder.mid.block_2.norm1.bias",
  "encoder.mid.block_2.norm1.weight",
  "encoder.mid.block_2.norm2.bias",
  "encoder.mid.block_2.norm2.weight",
  "encoder.norm_out.bias",
  "encoder.norm_out.weight",
  "vae.decoder.conv_in.bias",
  "vae.decoder.conv_in.weight",
  "vae.decoder.conv_out.bias",
  "vae.decoder.conv_out.weight",
  "vae.decoder.mid.attn_1.k.bias",
  "vae.decoder.mid.attn_1.k.weight",
  "vae.decoder.mid.attn_1.norm.bias",
  "vae.decoder.mid.attn_1.norm.weight",
  "vae.decoder.mid.attn_1.proj_out.bias",
  "vae.decoder.mid.attn_1.proj_out.weight",
  "vae.decoder.mid.attn_1.q.bias",
  "vae.decoder.mid.attn_1.q.weight",
  "vae.decoder.mid.attn_1.v.bias",
  "vae.decoder.mid.attn_1.v.weight",
  "vae.decoder.mid.block_1.conv1.bias",
  "vae.decoder.mid.block_1.conv1.weight",
  "vae.decoder.mid.block_1.conv2.bias",
  "vae.decoder.mid.block_1.conv2.weight",
  "vae.decoder.mid.block_1.norm1.bias",
  "vae.decoder.mid.block_1.norm1.weight",
  "vae.decoder.mid.block_1.norm2.bias",
  "vae.decoder.mid.block_1.norm2.weight",
  "vae.decoder.mid.block_2.conv1.bias",
  "vae.decoder.mid.block_2.conv1.weight",
  "vae.decoder.mid.block_2.conv2.bias",
  "vae.decoder.mid.block_2.conv2.weight",
  "vae.decoder.mid.block_2.norm1.bias",
  "vae.decoder.mid.block_2.norm1.weight",
  "vae.decoder.mid.block_2.norm2.bias",
  "vae.decoder.mid.block_2.norm2.weight",
  "vae.decoder.norm_out.bias",
  "vae.decoder.norm_out.weight",
  "vae.decoder.up.0.block.0.conv1.bias",
  "vae.decoder.up.0.block.0.conv1.weight",
  "vae.decoder.up.0.block.0.conv2.bias",
  "vae.decoder.up.0.block.0.conv2.weight",
  "vae.decoder.up.0.block.0.nin_shortcut.bias",
  "vae.decoder.up.0.block.0.nin_shortcut.weight",
  "vae.decoder.up.0.block.0.norm1.bias",
  "vae.decoder.up.0.block.0.norm1.weight",
  "vae.decoder.up.0.block.0.norm2.bias",
  "vae.decoder.up.0.block.0.norm2.weight",
  "vae.decoder.up.0.block.1.conv1.bias",
  "vae.decoder.up.0.block.1.conv1.weight",
  "vae.decoder.up.0.block.1.conv2.bias",
  "vae.decoder.up.0.block.1.conv2.weight",
  "vae.decoder.up.0.block.1.norm1.bias",
  "vae.decoder.up.0.block.1.norm1.weight",
  "vae.decoder.up.0.block.1.norm2.bias",
  "vae.decoder.up.0.block.1.norm2.weight",
  "vae.decoder.up.0.block.2.conv1.bias",
  "vae.decoder.up.0.block.2.conv1.weight",
  "vae.decoder.up.0.block.2.conv2.bias",
  "vae.decoder.up.0.block.2.conv2.weight",
  "vae.decoder.up.0.block.2.norm1.bias",
  "vae.decoder.up.0.block.2.norm1.weight",
  "vae.decoder.up.0.block.2.norm2.bias",
  "vae.decoder.up.0.block.2.norm2.weight",
  "vae.decoder.up.1.block.0.conv1.bias",
  "vae.decoder.up.1.block.0.conv1.weight",
  "vae.decoder.up.1.block.0.conv2.bias",
  "vae.decoder.up.1.block.0.conv2.weight",
  "vae.decoder.up.1.block.0.nin_shortcut.bias",
  "vae.decoder.up.1.block.0.nin_shortcut.weight",
  "vae.decoder.up.1.block.0.norm1.bias",
  "vae.decoder.up.1.block.0.norm1.weight",
  "vae.decoder.up.1.block.0.norm2.bias",
  "vae.decoder.up.1.block.0.norm2.weight",
  "vae.decoder.up.1.block.1.conv1.bias",
  "vae.decoder.up.1.block.1.conv1.weight",
  "vae.decoder.up.1.block.1.conv2.bias",
  "vae.decoder.up.1.block.1.conv2.weight",
  "vae.decoder.up.1.block.1.norm1.bias",
  "vae.decoder.up.1.block.1.norm1.weight",
  "vae.decoder.up.1.block.1.norm2.bias",
  "vae.decoder.up.1.block.1.norm2.weight",
  "vae.decoder.up.1.block.2.conv1.bias",
  "vae.decoder.up.1.block.2.conv1.weight",
  "vae.decoder.up.1.block.2.conv2.bias",
  "vae.decoder.up.1.block.2.conv2.weight",
  "vae.decoder.up.1.block.2.norm1.bias",
  "vae.decoder.up.1.block.2.norm1.weight",
  "vae.decoder.up.1.block.2.norm2.bias",
  "vae.decoder.up.1.block.2.norm2.weight",
  "vae.decoder.up.1.upsample.conv.bias",
  "vae.decoder.up.1.upsample.conv.weight",
  "vae.decoder.up.2.block.0.conv1.bias",
  "vae.decoder.up.2.block.0.conv1.weight",
  "vae.decoder.up.2.block.0.conv2.bias",
  "vae.decoder.up.2.block.0.conv2.weight",
  "vae.decoder.up.2.block.0.norm1.bias",
  "vae.decoder.up.2.block.0.norm1.weight",
  "vae.decoder.up.2.block.0.norm2.bias",
  "vae.decoder.up.2.block.0.norm2.weight",
  "vae.decoder.up.2.block.1.conv1.bias",
  "vae.decoder.up.2.block.1.conv1.weight",
  "vae.decoder.up.2.block.1.conv2.bias",
  "vae.decoder.up.2.block.1.conv2.weight",
  "vae.decoder.up.2.block.1.norm1.bias",
  "vae.decoder.up.2.block.1.norm1.weight",
  "vae.decoder.up.2.block.1.norm2.bias",
  "vae.decoder.up.2.block.1.norm2.weight",
  "vae.decoder.up.2.block.2.conv1.bias",
  "vae.decoder.up.2.block.2.conv1.weight",
  "vae.decoder.up.2.block.2.conv2.bias",
  "vae.decoder.up.2.block.2.conv2.weight",
  "vae.decoder.up.2.block.2.norm1.bias",
  "vae.decoder.up.2.block.2.norm1.weight",
  "vae.decoder.up.2.block.2.norm2.bias",
  "vae.decoder.up.2.block.2.norm2.weight",
  "vae.decoder.up.2.upsample.conv.bias",
  "vae.decoder.up.2.upsample.conv.weight",
  "vae.decoder.up.3.block.0.conv1.bias",
  "vae.decoder.up.3.block.0.conv1.weight",
  "vae.decoder.up.3.block.0.conv2.bias",
  "vae.decoder.up.3.block.0.conv2.weight",
  "vae.decoder.up.3.block.0.norm1.bias",
  "vae.decoder.up.3.block.0.norm1.weight",
  "vae.decoder.up.3.block.0.norm2.bias",
  "vae.decoder.up.3.block.0.norm2.weight",
  "vae.decoder.up.3.block.1.conv1.bias",
  "vae.decoder.up.3.block.1.conv1.weight",
  "vae.decoder.up.3.block.1.conv2.bias",
  "vae.decoder.up.3.block.1.conv2.weight",
  "vae.decoder.up.3.block.1.norm1.bias",
  "vae.decoder.up.3.block.1.norm1.weight",
  "vae.decoder.up.3.block.1.norm2.bias",
  "vae.decoder.up.3.block.1.norm2.weight",
  "vae.decoder.up.3.block.2.conv1.bias",
  "vae.decoder.up.3.block.2.conv1.weight",
  "vae.decoder.up.3.block.2.conv2.bias",
  "vae.decoder.up.3.block.2.conv2.weight",
  "vae.decoder.up.3.block.2.norm1.bias",
  "vae.decoder.up.3.block.2.norm1.weight",
  "vae.decoder.up.3.block.2.norm2.bias",
  "vae.decoder.up.3.block.2.norm2.weight",
  "vae.decoder.up.3.upsample.conv.bias",
  "vae.decoder.up.3.upsample.conv.weight",
  "vae.encoder.conv_in.bias",
  "vae.encoder.conv_in.weight",
  "vae.encoder.conv_out.bias",
  "vae.encoder.conv_out.weight",
  "vae.encoder.down.0.block.0.conv1.bias",
  "vae.encoder.down.0.block.0.conv1.weight",
  "vae.encoder.down.0.block.0.conv2.bias",
  "vae.encoder.down.0.block.0.conv2.weight",
  "vae.encoder.down.0.block.0.norm1.bias",
  "vae.encoder.down.0.block.0.norm1.weight",
  "vae.encoder.down.0.block.0.norm2.bias",
  "vae.encoder.down.0.block.0.norm2.weight",
  "vae.encoder.down.0.block.1.conv1.bias",
  "vae.encoder.down.0.block.1.conv1.weight",
  "vae.encoder.down.0.block.1.conv2.bias",
  "vae.encoder.down.0.block.1.conv2.weight",
  "vae.encoder.down.0.block.1.norm1.bias",
  "vae.encoder.down.0.block.1.norm1.weight",
  "vae.encoder.down.0.block.1.norm2.bias",
  "vae.encoder.down.0.block.1.norm2.weight",
  "vae.encoder.down.0.downsample.conv.bias",
  "vae.encoder.down.0.downsample.conv.weight",
  "vae.encoder.down.1.block.0.conv1.bias",
  "vae.encoder.down.1.block.0.conv1.weight",
  "vae.encoder.down.1.block.0.conv2.bias",
  "vae.encoder.down.1.block.0.conv2.weight",
  "vae.encoder.down.1.block.0.nin_shortcut.bias",
  "vae.encoder.down.1.block.0.nin_shortcut.weight",
  "vae.encoder.down.1.block.0.norm1.bias",
  "vae.encoder.down.1.block.0.norm1.weight",
  "vae.encoder.down.1.block.0.norm2.bias",
  "vae.encoder.down.1.block.0.norm2.weight",
  "vae.encoder.down.1.block.1.conv1.bias",
  "vae.encoder.down.1.block.1.conv1.weight",
  "vae.encoder.down.1.block.1.conv2.bias",
  "vae.encoder.down.1.block.1.conv2.weight",
  "vae.encoder.down.1.block.1.norm1.bias",
  "vae.encoder.down.1.block.1.norm1.weight",
  "vae.encoder.down.1.block.1.norm2.bias",
  "vae.encoder.down.1.block.1.norm2.weight",
  "vae.encoder.down.1.downsample.conv.bias",
  "vae.encoder.down.1.downsample.conv.weight",
  "vae.encoder.down.2.block.0.conv1.bias",
  "vae.encoder.down.2.block.0.conv1.weight",
  "vae.encoder.down.2.block.0.conv2.bias",
  "vae.encoder.down.2.block.0.conv2.weight",
  "vae.encoder.down.2.block.0.nin_shortcut.bias",
  "vae.encoder.down.2.block.0.nin_shortcut.weight",
  "vae.encoder.down.2.block.0.norm1.bias",
  "vae.encoder.down.2.block.0.norm1.weight",
  "vae.encoder.down.2.block.0.norm2.bias",
  "vae.encoder.down.2.block.0.norm2.weight",
  "vae.encoder.down.2.block.1.conv1.bias",
  "vae.encoder.down.2.block.1.conv1.weight",
  "vae.encoder.down.2.block.1.conv2.bias",
  "vae.encoder.down.2.block.1.conv2.weight",
  "vae.encoder.down.2.block.1.norm1.bias",
  "vae.encoder.down.2.block.1.norm1.weight",
  "vae.encoder.down.2.block.1.norm2.bias",
  "vae.encoder.down.2.block.1.norm2.weight",
  "vae.encoder.down.2.downsample.conv.bias",
  "vae.encoder.down.2.downsample.conv.weight",
  "vae.encoder.down.3.block.0.conv1.bias",
  "vae.encoder.down.3.block.0.conv1.weight",
  "vae.encoder.down.3.block.0.conv2.bias",
  "vae.encoder.down.3.block.0.conv2.weight",
  "vae.encoder.down.3.block.0.norm1.bias",
  "vae.encoder.down.3.block.0.norm1.weight",
  "vae.encoder.down.3.block.0.norm2.bias",
  "vae.encoder.down.3.block.0.norm2.weight",
  "vae.encoder.down.3.block.1.conv1.bias",
  "vae.encoder.down.3.block.1.conv1.weight",
  "vae.encoder.down.3.block.1.conv2.bias",
  "vae.encoder.down.3.block.1.conv2.weight",
  "vae.encoder.down.3.block.1.norm1.bias",
  "vae.encoder.down.3.block.1.norm1.weight",
  "vae.encoder.down.3.block.1.norm2.bias",
  "vae.encoder.down.3.block.1.norm2.weight",
  "vae.encoder.mid.attn_1.k.bias",
  "vae.encoder.mid.attn_1.k.weight",
  "vae.encoder.mid.attn_1.norm.bias",
  "vae.encoder.mid.attn_1.norm.weight",
  "vae.encoder.mid.attn_1.proj_out.bias",
  "vae.encoder.mid.attn_1.proj_out.weight",
  "vae.encoder.mid.attn_1.q.bias",
  "vae.encoder.mid.attn_1.q.weight",
  "vae.encoder.mid.attn_1.v.bias",
  "vae.encoder.mid.attn_1.v.weight",
  "vae.encoder.mid.block_1.conv1.bias",
  "vae.encoder.mid.block_1.conv1.weight",
  "vae.encoder.mid.block_1.conv2.bias",
  "vae.encoder.mid.block_1.conv2.weight",
  "vae.encoder.mid.block_1.norm1.bias",
  "vae.encoder.mid.block_1.norm1.weight",
  "vae.encoder.mid.block_1.norm2.bias",
  "vae.encoder.mid.block_1.norm2.weight",
  "vae.encoder.mid.block_2.conv1.bias",
  "vae.encoder.mid.block_2.conv1.weight",
  "vae.encoder.mid.block_2.conv2.bias",
  "vae.encoder.mid.block_2.conv2.weight",
  "vae.encoder.mid.block_2.norm1.bias",
  "vae.encoder.mid.block_2.norm1.weight",
  "vae.encoder.mid.block_2.norm2.bias",
  "vae.encoder.mid.block_2.norm2.weight",
  "vae.encoder.norm_out.bias",
  "vae.encoder.norm_out.weight"
]