{
  "inputs": {
    "mapping_reference_checkpoint": "/home/rezo/piper-wikipedia-approved-20260830/runs/mana-warmstart-pilot3-runtimecheck/lightning_logs/version_0/checkpoints/last.ckpt",
    "mapping_reference_checkpoint_sha256": "720e6f1deb9b665fa3a64b849578e67b22e5cd5457589492748c59b5af629239",
    "mapping_reference_onnx": "/home/rezo/piper-wikipedia-approved-20260830/runs/mana-warmstart-pilot3-runtimecheck/model.onnx",
    "mapping_reference_onnx_sha256": "b23dde224885d42fdddec8aa886e24fc8fa9cde7f188622e2771eb0ff2d4c2be",
    "production_onnx": "/home/rezo/piper-wikipedia-approved-20260830/models/rizeh-onnx-recovery/epoch4-rizehpizeh.onnx",
    "production_onnx_sha256": "dc8532f0c57a40ed076e466f380376577e8bbcd13d1fc19c81f69d461d26dadc",
    "template_checkpoint": "/home/rezo/piper-wikipedia-approved-20260830/models/mana/ckpt/epoch=6012-step=4203520.ckpt",
    "template_checkpoint_sha256": "ec2c289b8f4a32b0a7f1f491a98beaa24f8fab8a13de537495f1d649c8818980"
  },
  "output_checkpoint": "/home/rezo/piper-wikipedia-approved-20260830/models/rizeh-onnx-recovery/rizehpizeh-epoch4-recovered-trainable.ckpt",
  "output_checkpoint_sha256": "331c1e8c95e86f88f2a11a6eb14475da609a58891a014cf3cf1e3af414210347",
  "recovered_by_role": {
    "dec": 45,
    "dp": 114,
    "enc_p": 111,
    "flow": 80
  },
  "recovered_inference_tensors": 350,
  "schema_version": "gooya.rizeh-onnx-recovery/v1",
  "special_transform_verification": {
    "initializer": "onnx::Exp_8946",
    "max_abs_error": 0.0,
    "transform": "checkpoint_value = -onnx_initializer"
  },
  "status": "recovered_pending_roundtrip_parity",
  "tensors": [
    {
      "initializer": "sid",
      "shape": [
        256,
        192
      ],
      "storage": "direct",
      "target": "enc_p.emb.weight",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.attn_layers.0.emb_rel_k",
      "shape": [
        1,
        9,
        96
      ],
      "storage": "direct",
      "target": "enc_p.encoder.attn_layers.0.emb_rel_k",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.attn_layers.0.emb_rel_v",
      "shape": [
        1,
        9,
        96
      ],
      "storage": "direct",
      "target": "enc_p.encoder.attn_layers.0.emb_rel_v",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.attn_layers.0.conv_q.weight",
      "shape": [
        192,
        192,
        1
      ],
      "storage": "direct",
      "target": "enc_p.encoder.attn_layers.0.conv_q.weight",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.attn_layers.0.conv_q.bias",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "enc_p.encoder.attn_layers.0.conv_q.bias",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.attn_layers.0.conv_k.weight",
      "shape": [
        192,
        192,
        1
      ],
      "storage": "direct",
      "target": "enc_p.encoder.attn_layers.0.conv_k.weight",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.attn_layers.0.conv_k.bias",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "enc_p.encoder.attn_layers.0.conv_k.bias",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.attn_layers.0.conv_v.weight",
      "shape": [
        192,
        192,
        1
      ],
      "storage": "direct",
      "target": "enc_p.encoder.attn_layers.0.conv_v.weight",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.attn_layers.0.conv_v.bias",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "enc_p.encoder.attn_layers.0.conv_v.bias",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.attn_layers.0.conv_o.weight",
      "shape": [
        192,
        192,
        1
      ],
      "storage": "direct",
      "target": "enc_p.encoder.attn_layers.0.conv_o.weight",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.attn_layers.0.conv_o.bias",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "enc_p.encoder.attn_layers.0.conv_o.bias",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.attn_layers.1.emb_rel_k",
      "shape": [
        1,
        9,
        96
      ],
      "storage": "direct",
      "target": "enc_p.encoder.attn_layers.1.emb_rel_k",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.attn_layers.1.emb_rel_v",
      "shape": [
        1,
        9,
        96
      ],
      "storage": "direct",
      "target": "enc_p.encoder.attn_layers.1.emb_rel_v",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.attn_layers.1.conv_q.weight",
      "shape": [
        192,
        192,
        1
      ],
      "storage": "direct",
      "target": "enc_p.encoder.attn_layers.1.conv_q.weight",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.attn_layers.1.conv_q.bias",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "enc_p.encoder.attn_layers.1.conv_q.bias",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.attn_layers.1.conv_k.weight",
      "shape": [
        192,
        192,
        1
      ],
      "storage": "direct",
      "target": "enc_p.encoder.attn_layers.1.conv_k.weight",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.attn_layers.1.conv_k.bias",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "enc_p.encoder.attn_layers.1.conv_k.bias",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.attn_layers.1.conv_v.weight",
      "shape": [
        192,
        192,
        1
      ],
      "storage": "direct",
      "target": "enc_p.encoder.attn_layers.1.conv_v.weight",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.attn_layers.1.conv_v.bias",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "enc_p.encoder.attn_layers.1.conv_v.bias",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.attn_layers.1.conv_o.weight",
      "shape": [
        192,
        192,
        1
      ],
      "storage": "direct",
      "target": "enc_p.encoder.attn_layers.1.conv_o.weight",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.attn_layers.1.conv_o.bias",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "enc_p.encoder.attn_layers.1.conv_o.bias",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.attn_layers.2.emb_rel_k",
      "shape": [
        1,
        9,
        96
      ],
      "storage": "direct",
      "target": "enc_p.encoder.attn_layers.2.emb_rel_k",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.attn_layers.2.emb_rel_v",
      "shape": [
        1,
        9,
        96
      ],
      "storage": "direct",
      "target": "enc_p.encoder.attn_layers.2.emb_rel_v",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.attn_layers.2.conv_q.weight",
      "shape": [
        192,
        192,
        1
      ],
      "storage": "direct",
      "target": "enc_p.encoder.attn_layers.2.conv_q.weight",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.attn_layers.2.conv_q.bias",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "enc_p.encoder.attn_layers.2.conv_q.bias",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.attn_layers.2.conv_k.weight",
      "shape": [
        192,
        192,
        1
      ],
      "storage": "direct",
      "target": "enc_p.encoder.attn_layers.2.conv_k.weight",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.attn_layers.2.conv_k.bias",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "enc_p.encoder.attn_layers.2.conv_k.bias",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.attn_layers.2.conv_v.weight",
      "shape": [
        192,
        192,
        1
      ],
      "storage": "direct",
      "target": "enc_p.encoder.attn_layers.2.conv_v.weight",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.attn_layers.2.conv_v.bias",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "enc_p.encoder.attn_layers.2.conv_v.bias",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.attn_layers.2.conv_o.weight",
      "shape": [
        192,
        192,
        1
      ],
      "storage": "direct",
      "target": "enc_p.encoder.attn_layers.2.conv_o.weight",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.attn_layers.2.conv_o.bias",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "enc_p.encoder.attn_layers.2.conv_o.bias",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.attn_layers.3.emb_rel_k",
      "shape": [
        1,
        9,
        96
      ],
      "storage": "direct",
      "target": "enc_p.encoder.attn_layers.3.emb_rel_k",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.attn_layers.3.emb_rel_v",
      "shape": [
        1,
        9,
        96
      ],
      "storage": "direct",
      "target": "enc_p.encoder.attn_layers.3.emb_rel_v",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.attn_layers.3.conv_q.weight",
      "shape": [
        192,
        192,
        1
      ],
      "storage": "direct",
      "target": "enc_p.encoder.attn_layers.3.conv_q.weight",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.attn_layers.3.conv_q.bias",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "enc_p.encoder.attn_layers.3.conv_q.bias",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.attn_layers.3.conv_k.weight",
      "shape": [
        192,
        192,
        1
      ],
      "storage": "direct",
      "target": "enc_p.encoder.attn_layers.3.conv_k.weight",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.attn_layers.3.conv_k.bias",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "enc_p.encoder.attn_layers.3.conv_k.bias",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.attn_layers.3.conv_v.weight",
      "shape": [
        192,
        192,
        1
      ],
      "storage": "direct",
      "target": "enc_p.encoder.attn_layers.3.conv_v.weight",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.attn_layers.3.conv_v.bias",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "enc_p.encoder.attn_layers.3.conv_v.bias",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.attn_layers.3.conv_o.weight",
      "shape": [
        192,
        192,
        1
      ],
      "storage": "direct",
      "target": "enc_p.encoder.attn_layers.3.conv_o.weight",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.attn_layers.3.conv_o.bias",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "enc_p.encoder.attn_layers.3.conv_o.bias",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.attn_layers.4.emb_rel_k",
      "shape": [
        1,
        9,
        96
      ],
      "storage": "direct",
      "target": "enc_p.encoder.attn_layers.4.emb_rel_k",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.attn_layers.4.emb_rel_v",
      "shape": [
        1,
        9,
        96
      ],
      "storage": "direct",
      "target": "enc_p.encoder.attn_layers.4.emb_rel_v",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.attn_layers.4.conv_q.weight",
      "shape": [
        192,
        192,
        1
      ],
      "storage": "direct",
      "target": "enc_p.encoder.attn_layers.4.conv_q.weight",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.attn_layers.4.conv_q.bias",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "enc_p.encoder.attn_layers.4.conv_q.bias",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.attn_layers.4.conv_k.weight",
      "shape": [
        192,
        192,
        1
      ],
      "storage": "direct",
      "target": "enc_p.encoder.attn_layers.4.conv_k.weight",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.attn_layers.4.conv_k.bias",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "enc_p.encoder.attn_layers.4.conv_k.bias",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.attn_layers.4.conv_v.weight",
      "shape": [
        192,
        192,
        1
      ],
      "storage": "direct",
      "target": "enc_p.encoder.attn_layers.4.conv_v.weight",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.attn_layers.4.conv_v.bias",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "enc_p.encoder.attn_layers.4.conv_v.bias",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.attn_layers.4.conv_o.weight",
      "shape": [
        192,
        192,
        1
      ],
      "storage": "direct",
      "target": "enc_p.encoder.attn_layers.4.conv_o.weight",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.attn_layers.4.conv_o.bias",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "enc_p.encoder.attn_layers.4.conv_o.bias",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.attn_layers.5.emb_rel_k",
      "shape": [
        1,
        9,
        96
      ],
      "storage": "direct",
      "target": "enc_p.encoder.attn_layers.5.emb_rel_k",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.attn_layers.5.emb_rel_v",
      "shape": [
        1,
        9,
        96
      ],
      "storage": "direct",
      "target": "enc_p.encoder.attn_layers.5.emb_rel_v",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.attn_layers.5.conv_q.weight",
      "shape": [
        192,
        192,
        1
      ],
      "storage": "direct",
      "target": "enc_p.encoder.attn_layers.5.conv_q.weight",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.attn_layers.5.conv_q.bias",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "enc_p.encoder.attn_layers.5.conv_q.bias",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.attn_layers.5.conv_k.weight",
      "shape": [
        192,
        192,
        1
      ],
      "storage": "direct",
      "target": "enc_p.encoder.attn_layers.5.conv_k.weight",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.attn_layers.5.conv_k.bias",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "enc_p.encoder.attn_layers.5.conv_k.bias",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.attn_layers.5.conv_v.weight",
      "shape": [
        192,
        192,
        1
      ],
      "storage": "direct",
      "target": "enc_p.encoder.attn_layers.5.conv_v.weight",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.attn_layers.5.conv_v.bias",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "enc_p.encoder.attn_layers.5.conv_v.bias",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.attn_layers.5.conv_o.weight",
      "shape": [
        192,
        192,
        1
      ],
      "storage": "direct",
      "target": "enc_p.encoder.attn_layers.5.conv_o.weight",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.attn_layers.5.conv_o.bias",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "enc_p.encoder.attn_layers.5.conv_o.bias",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.norm_layers_1.0.gamma",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "enc_p.encoder.norm_layers_1.0.gamma",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.norm_layers_1.0.beta",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "enc_p.encoder.norm_layers_1.0.beta",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.norm_layers_1.1.gamma",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "enc_p.encoder.norm_layers_1.1.gamma",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.norm_layers_1.1.beta",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "enc_p.encoder.norm_layers_1.1.beta",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.norm_layers_1.2.gamma",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "enc_p.encoder.norm_layers_1.2.gamma",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.norm_layers_1.2.beta",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "enc_p.encoder.norm_layers_1.2.beta",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.norm_layers_1.3.gamma",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "enc_p.encoder.norm_layers_1.3.gamma",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.norm_layers_1.3.beta",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "enc_p.encoder.norm_layers_1.3.beta",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.norm_layers_1.4.gamma",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "enc_p.encoder.norm_layers_1.4.gamma",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.norm_layers_1.4.beta",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "enc_p.encoder.norm_layers_1.4.beta",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.norm_layers_1.5.gamma",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "enc_p.encoder.norm_layers_1.5.gamma",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.norm_layers_1.5.beta",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "enc_p.encoder.norm_layers_1.5.beta",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.ffn_layers.0.conv_1.weight",
      "shape": [
        768,
        192,
        3
      ],
      "storage": "direct",
      "target": "enc_p.encoder.ffn_layers.0.conv_1.weight",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.ffn_layers.0.conv_1.bias",
      "shape": [
        768
      ],
      "storage": "direct",
      "target": "enc_p.encoder.ffn_layers.0.conv_1.bias",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.ffn_layers.0.conv_2.weight",
      "shape": [
        192,
        768,
        3
      ],
      "storage": "direct",
      "target": "enc_p.encoder.ffn_layers.0.conv_2.weight",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.ffn_layers.0.conv_2.bias",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "enc_p.encoder.ffn_layers.0.conv_2.bias",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.ffn_layers.1.conv_1.weight",
      "shape": [
        768,
        192,
        3
      ],
      "storage": "direct",
      "target": "enc_p.encoder.ffn_layers.1.conv_1.weight",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.ffn_layers.1.conv_1.bias",
      "shape": [
        768
      ],
      "storage": "direct",
      "target": "enc_p.encoder.ffn_layers.1.conv_1.bias",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.ffn_layers.1.conv_2.weight",
      "shape": [
        192,
        768,
        3
      ],
      "storage": "direct",
      "target": "enc_p.encoder.ffn_layers.1.conv_2.weight",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.ffn_layers.1.conv_2.bias",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "enc_p.encoder.ffn_layers.1.conv_2.bias",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.ffn_layers.2.conv_1.weight",
      "shape": [
        768,
        192,
        3
      ],
      "storage": "direct",
      "target": "enc_p.encoder.ffn_layers.2.conv_1.weight",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.ffn_layers.2.conv_1.bias",
      "shape": [
        768
      ],
      "storage": "direct",
      "target": "enc_p.encoder.ffn_layers.2.conv_1.bias",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.ffn_layers.2.conv_2.weight",
      "shape": [
        192,
        768,
        3
      ],
      "storage": "direct",
      "target": "enc_p.encoder.ffn_layers.2.conv_2.weight",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.ffn_layers.2.conv_2.bias",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "enc_p.encoder.ffn_layers.2.conv_2.bias",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.ffn_layers.3.conv_1.weight",
      "shape": [
        768,
        192,
        3
      ],
      "storage": "direct",
      "target": "enc_p.encoder.ffn_layers.3.conv_1.weight",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.ffn_layers.3.conv_1.bias",
      "shape": [
        768
      ],
      "storage": "direct",
      "target": "enc_p.encoder.ffn_layers.3.conv_1.bias",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.ffn_layers.3.conv_2.weight",
      "shape": [
        192,
        768,
        3
      ],
      "storage": "direct",
      "target": "enc_p.encoder.ffn_layers.3.conv_2.weight",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.ffn_layers.3.conv_2.bias",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "enc_p.encoder.ffn_layers.3.conv_2.bias",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.ffn_layers.4.conv_1.weight",
      "shape": [
        768,
        192,
        3
      ],
      "storage": "direct",
      "target": "enc_p.encoder.ffn_layers.4.conv_1.weight",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.ffn_layers.4.conv_1.bias",
      "shape": [
        768
      ],
      "storage": "direct",
      "target": "enc_p.encoder.ffn_layers.4.conv_1.bias",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.ffn_layers.4.conv_2.weight",
      "shape": [
        192,
        768,
        3
      ],
      "storage": "direct",
      "target": "enc_p.encoder.ffn_layers.4.conv_2.weight",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.ffn_layers.4.conv_2.bias",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "enc_p.encoder.ffn_layers.4.conv_2.bias",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.ffn_layers.5.conv_1.weight",
      "shape": [
        768,
        192,
        3
      ],
      "storage": "direct",
      "target": "enc_p.encoder.ffn_layers.5.conv_1.weight",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.ffn_layers.5.conv_1.bias",
      "shape": [
        768
      ],
      "storage": "direct",
      "target": "enc_p.encoder.ffn_layers.5.conv_1.bias",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.ffn_layers.5.conv_2.weight",
      "shape": [
        192,
        768,
        3
      ],
      "storage": "direct",
      "target": "enc_p.encoder.ffn_layers.5.conv_2.weight",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.ffn_layers.5.conv_2.bias",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "enc_p.encoder.ffn_layers.5.conv_2.bias",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.norm_layers_2.0.gamma",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "enc_p.encoder.norm_layers_2.0.gamma",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.norm_layers_2.0.beta",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "enc_p.encoder.norm_layers_2.0.beta",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.norm_layers_2.1.gamma",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "enc_p.encoder.norm_layers_2.1.gamma",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.norm_layers_2.1.beta",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "enc_p.encoder.norm_layers_2.1.beta",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.norm_layers_2.2.gamma",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "enc_p.encoder.norm_layers_2.2.gamma",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.norm_layers_2.2.beta",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "enc_p.encoder.norm_layers_2.2.beta",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.norm_layers_2.3.gamma",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "enc_p.encoder.norm_layers_2.3.gamma",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.norm_layers_2.3.beta",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "enc_p.encoder.norm_layers_2.3.beta",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.norm_layers_2.4.gamma",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "enc_p.encoder.norm_layers_2.4.gamma",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.norm_layers_2.4.beta",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "enc_p.encoder.norm_layers_2.4.beta",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.norm_layers_2.5.gamma",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "enc_p.encoder.norm_layers_2.5.gamma",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.encoder.norm_layers_2.5.beta",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "enc_p.encoder.norm_layers_2.5.beta",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.proj.weight",
      "shape": [
        384,
        192,
        1
      ],
      "storage": "direct",
      "target": "enc_p.proj.weight",
      "transform": "identity"
    },
    {
      "initializer": "enc_p.proj.bias",
      "shape": [
        384
      ],
      "storage": "direct",
      "target": "enc_p.proj.bias",
      "transform": "identity"
    },
    {
      "initializer": "dec.conv_pre.weight",
      "shape": [
        256,
        192,
        7
      ],
      "storage": "direct",
      "target": "dec.conv_pre.weight",
      "transform": "identity"
    },
    {
      "initializer": "dec.conv_pre.bias",
      "shape": [
        256
      ],
      "storage": "direct",
      "target": "dec.conv_pre.bias",
      "transform": "identity"
    },
    {
      "initializer": "dec.ups.0.bias",
      "shape": [
        128
      ],
      "storage": "direct",
      "target": "dec.ups.0.bias",
      "transform": "identity"
    },
    {
      "initializer": "dec.ups.0.weight",
      "shape": [
        256,
        128,
        16
      ],
      "storage": "weight_norm_reconstructed",
      "target": "dec.ups.0.weight",
      "transform": "identity"
    },
    {
      "initializer": "dec.ups.1.bias",
      "shape": [
        64
      ],
      "storage": "direct",
      "target": "dec.ups.1.bias",
      "transform": "identity"
    },
    {
      "initializer": "dec.ups.1.weight",
      "shape": [
        128,
        64,
        16
      ],
      "storage": "weight_norm_reconstructed",
      "target": "dec.ups.1.weight",
      "transform": "identity"
    },
    {
      "initializer": "dec.ups.2.bias",
      "shape": [
        32
      ],
      "storage": "direct",
      "target": "dec.ups.2.bias",
      "transform": "identity"
    },
    {
      "initializer": "dec.ups.2.weight",
      "shape": [
        64,
        32,
        8
      ],
      "storage": "weight_norm_reconstructed",
      "target": "dec.ups.2.weight",
      "transform": "identity"
    },
    {
      "initializer": "dec.resblocks.0.convs.0.bias",
      "shape": [
        128
      ],
      "storage": "direct",
      "target": "dec.resblocks.0.convs.0.bias",
      "transform": "identity"
    },
    {
      "initializer": "dec.resblocks.0.convs.0.weight",
      "shape": [
        128,
        128,
        3
      ],
      "storage": "weight_norm_reconstructed",
      "target": "dec.resblocks.0.convs.0.weight",
      "transform": "identity"
    },
    {
      "initializer": "dec.resblocks.0.convs.1.bias",
      "shape": [
        128
      ],
      "storage": "direct",
      "target": "dec.resblocks.0.convs.1.bias",
      "transform": "identity"
    },
    {
      "initializer": "dec.resblocks.0.convs.1.weight",
      "shape": [
        128,
        128,
        3
      ],
      "storage": "weight_norm_reconstructed",
      "target": "dec.resblocks.0.convs.1.weight",
      "transform": "identity"
    },
    {
      "initializer": "dec.resblocks.1.convs.0.bias",
      "shape": [
        128
      ],
      "storage": "direct",
      "target": "dec.resblocks.1.convs.0.bias",
      "transform": "identity"
    },
    {
      "initializer": "dec.resblocks.1.convs.0.weight",
      "shape": [
        128,
        128,
        5
      ],
      "storage": "weight_norm_reconstructed",
      "target": "dec.resblocks.1.convs.0.weight",
      "transform": "identity"
    },
    {
      "initializer": "dec.resblocks.1.convs.1.bias",
      "shape": [
        128
      ],
      "storage": "direct",
      "target": "dec.resblocks.1.convs.1.bias",
      "transform": "identity"
    },
    {
      "initializer": "dec.resblocks.1.convs.1.weight",
      "shape": [
        128,
        128,
        5
      ],
      "storage": "weight_norm_reconstructed",
      "target": "dec.resblocks.1.convs.1.weight",
      "transform": "identity"
    },
    {
      "initializer": "dec.resblocks.2.convs.0.bias",
      "shape": [
        128
      ],
      "storage": "direct",
      "target": "dec.resblocks.2.convs.0.bias",
      "transform": "identity"
    },
    {
      "initializer": "dec.resblocks.2.convs.0.weight",
      "shape": [
        128,
        128,
        7
      ],
      "storage": "weight_norm_reconstructed",
      "target": "dec.resblocks.2.convs.0.weight",
      "transform": "identity"
    },
    {
      "initializer": "dec.resblocks.2.convs.1.bias",
      "shape": [
        128
      ],
      "storage": "direct",
      "target": "dec.resblocks.2.convs.1.bias",
      "transform": "identity"
    },
    {
      "initializer": "dec.resblocks.2.convs.1.weight",
      "shape": [
        128,
        128,
        7
      ],
      "storage": "weight_norm_reconstructed",
      "target": "dec.resblocks.2.convs.1.weight",
      "transform": "identity"
    },
    {
      "initializer": "dec.resblocks.3.convs.0.bias",
      "shape": [
        64
      ],
      "storage": "direct",
      "target": "dec.resblocks.3.convs.0.bias",
      "transform": "identity"
    },
    {
      "initializer": "dec.resblocks.3.convs.0.weight",
      "shape": [
        64,
        64,
        3
      ],
      "storage": "weight_norm_reconstructed",
      "target": "dec.resblocks.3.convs.0.weight",
      "transform": "identity"
    },
    {
      "initializer": "dec.resblocks.3.convs.1.bias",
      "shape": [
        64
      ],
      "storage": "direct",
      "target": "dec.resblocks.3.convs.1.bias",
      "transform": "identity"
    },
    {
      "initializer": "dec.resblocks.3.convs.1.weight",
      "shape": [
        64,
        64,
        3
      ],
      "storage": "weight_norm_reconstructed",
      "target": "dec.resblocks.3.convs.1.weight",
      "transform": "identity"
    },
    {
      "initializer": "dec.resblocks.4.convs.0.bias",
      "shape": [
        64
      ],
      "storage": "direct",
      "target": "dec.resblocks.4.convs.0.bias",
      "transform": "identity"
    },
    {
      "initializer": "dec.resblocks.4.convs.0.weight",
      "shape": [
        64,
        64,
        5
      ],
      "storage": "weight_norm_reconstructed",
      "target": "dec.resblocks.4.convs.0.weight",
      "transform": "identity"
    },
    {
      "initializer": "dec.resblocks.4.convs.1.bias",
      "shape": [
        64
      ],
      "storage": "direct",
      "target": "dec.resblocks.4.convs.1.bias",
      "transform": "identity"
    },
    {
      "initializer": "dec.resblocks.4.convs.1.weight",
      "shape": [
        64,
        64,
        5
      ],
      "storage": "weight_norm_reconstructed",
      "target": "dec.resblocks.4.convs.1.weight",
      "transform": "identity"
    },
    {
      "initializer": "dec.resblocks.5.convs.0.bias",
      "shape": [
        64
      ],
      "storage": "direct",
      "target": "dec.resblocks.5.convs.0.bias",
      "transform": "identity"
    },
    {
      "initializer": "dec.resblocks.5.convs.0.weight",
      "shape": [
        64,
        64,
        7
      ],
      "storage": "weight_norm_reconstructed",
      "target": "dec.resblocks.5.convs.0.weight",
      "transform": "identity"
    },
    {
      "initializer": "dec.resblocks.5.convs.1.bias",
      "shape": [
        64
      ],
      "storage": "direct",
      "target": "dec.resblocks.5.convs.1.bias",
      "transform": "identity"
    },
    {
      "initializer": "dec.resblocks.5.convs.1.weight",
      "shape": [
        64,
        64,
        7
      ],
      "storage": "weight_norm_reconstructed",
      "target": "dec.resblocks.5.convs.1.weight",
      "transform": "identity"
    },
    {
      "initializer": "dec.resblocks.6.convs.0.bias",
      "shape": [
        32
      ],
      "storage": "direct",
      "target": "dec.resblocks.6.convs.0.bias",
      "transform": "identity"
    },
    {
      "initializer": "dec.resblocks.6.convs.0.weight",
      "shape": [
        32,
        32,
        3
      ],
      "storage": "weight_norm_reconstructed",
      "target": "dec.resblocks.6.convs.0.weight",
      "transform": "identity"
    },
    {
      "initializer": "dec.resblocks.6.convs.1.bias",
      "shape": [
        32
      ],
      "storage": "direct",
      "target": "dec.resblocks.6.convs.1.bias",
      "transform": "identity"
    },
    {
      "initializer": "dec.resblocks.6.convs.1.weight",
      "shape": [
        32,
        32,
        3
      ],
      "storage": "weight_norm_reconstructed",
      "target": "dec.resblocks.6.convs.1.weight",
      "transform": "identity"
    },
    {
      "initializer": "dec.resblocks.7.convs.0.bias",
      "shape": [
        32
      ],
      "storage": "direct",
      "target": "dec.resblocks.7.convs.0.bias",
      "transform": "identity"
    },
    {
      "initializer": "dec.resblocks.7.convs.0.weight",
      "shape": [
        32,
        32,
        5
      ],
      "storage": "weight_norm_reconstructed",
      "target": "dec.resblocks.7.convs.0.weight",
      "transform": "identity"
    },
    {
      "initializer": "dec.resblocks.7.convs.1.bias",
      "shape": [
        32
      ],
      "storage": "direct",
      "target": "dec.resblocks.7.convs.1.bias",
      "transform": "identity"
    },
    {
      "initializer": "dec.resblocks.7.convs.1.weight",
      "shape": [
        32,
        32,
        5
      ],
      "storage": "weight_norm_reconstructed",
      "target": "dec.resblocks.7.convs.1.weight",
      "transform": "identity"
    },
    {
      "initializer": "dec.resblocks.8.convs.0.bias",
      "shape": [
        32
      ],
      "storage": "direct",
      "target": "dec.resblocks.8.convs.0.bias",
      "transform": "identity"
    },
    {
      "initializer": "dec.resblocks.8.convs.0.weight",
      "shape": [
        32,
        32,
        7
      ],
      "storage": "weight_norm_reconstructed",
      "target": "dec.resblocks.8.convs.0.weight",
      "transform": "identity"
    },
    {
      "initializer": "dec.resblocks.8.convs.1.bias",
      "shape": [
        32
      ],
      "storage": "direct",
      "target": "dec.resblocks.8.convs.1.bias",
      "transform": "identity"
    },
    {
      "initializer": "dec.resblocks.8.convs.1.weight",
      "shape": [
        32,
        32,
        7
      ],
      "storage": "weight_norm_reconstructed",
      "target": "dec.resblocks.8.convs.1.weight",
      "transform": "identity"
    },
    {
      "initializer": "dec.conv_post.weight",
      "shape": [
        1,
        32,
        7
      ],
      "storage": "direct",
      "target": "dec.conv_post.weight",
      "transform": "identity"
    },
    {
      "initializer": "flow.flows.0.pre.weight",
      "shape": [
        192,
        96,
        1
      ],
      "storage": "direct",
      "target": "flow.flows.0.pre.weight",
      "transform": "identity"
    },
    {
      "initializer": "flow.flows.0.pre.bias",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "flow.flows.0.pre.bias",
      "transform": "identity"
    },
    {
      "initializer": "flow.flows.0.enc.in_layers.0.bias",
      "shape": [
        384
      ],
      "storage": "direct",
      "target": "flow.flows.0.enc.in_layers.0.bias",
      "transform": "identity"
    },
    {
      "initializer": "flow.flows.0.enc.in_layers.0.weight",
      "shape": [
        384,
        192,
        5
      ],
      "storage": "weight_norm_reconstructed",
      "target": "flow.flows.0.enc.in_layers.0.weight",
      "transform": "identity"
    },
    {
      "initializer": "flow.flows.0.enc.in_layers.1.bias",
      "shape": [
        384
      ],
      "storage": "direct",
      "target": "flow.flows.0.enc.in_layers.1.bias",
      "transform": "identity"
    },
    {
      "initializer": "flow.flows.0.enc.in_layers.1.weight",
      "shape": [
        384,
        192,
        5
      ],
      "storage": "weight_norm_reconstructed",
      "target": "flow.flows.0.enc.in_layers.1.weight",
      "transform": "identity"
    },
    {
      "initializer": "flow.flows.0.enc.in_layers.2.bias",
      "shape": [
        384
      ],
      "storage": "direct",
      "target": "flow.flows.0.enc.in_layers.2.bias",
      "transform": "identity"
    },
    {
      "initializer": "flow.flows.0.enc.in_layers.2.weight",
      "shape": [
        384,
        192,
        5
      ],
      "storage": "weight_norm_reconstructed",
      "target": "flow.flows.0.enc.in_layers.2.weight",
      "transform": "identity"
    },
    {
      "initializer": "flow.flows.0.enc.in_layers.3.bias",
      "shape": [
        384
      ],
      "storage": "direct",
      "target": "flow.flows.0.enc.in_layers.3.bias",
      "transform": "identity"
    },
    {
      "initializer": "flow.flows.0.enc.in_layers.3.weight",
      "shape": [
        384,
        192,
        5
      ],
      "storage": "weight_norm_reconstructed",
      "target": "flow.flows.0.enc.in_layers.3.weight",
      "transform": "identity"
    },
    {
      "initializer": "flow.flows.0.enc.res_skip_layers.0.bias",
      "shape": [
        384
      ],
      "storage": "direct",
      "target": "flow.flows.0.enc.res_skip_layers.0.bias",
      "transform": "identity"
    },
    {
      "initializer": "flow.flows.0.enc.res_skip_layers.0.weight",
      "shape": [
        384,
        192,
        1
      ],
      "storage": "weight_norm_reconstructed",
      "target": "flow.flows.0.enc.res_skip_layers.0.weight",
      "transform": "identity"
    },
    {
      "initializer": "flow.flows.0.enc.res_skip_layers.1.bias",
      "shape": [
        384
      ],
      "storage": "direct",
      "target": "flow.flows.0.enc.res_skip_layers.1.bias",
      "transform": "identity"
    },
    {
      "initializer": "flow.flows.0.enc.res_skip_layers.1.weight",
      "shape": [
        384,
        192,
        1
      ],
      "storage": "weight_norm_reconstructed",
      "target": "flow.flows.0.enc.res_skip_layers.1.weight",
      "transform": "identity"
    },
    {
      "initializer": "flow.flows.0.enc.res_skip_layers.2.bias",
      "shape": [
        384
      ],
      "storage": "direct",
      "target": "flow.flows.0.enc.res_skip_layers.2.bias",
      "transform": "identity"
    },
    {
      "initializer": "flow.flows.0.enc.res_skip_layers.2.weight",
      "shape": [
        384,
        192,
        1
      ],
      "storage": "weight_norm_reconstructed",
      "target": "flow.flows.0.enc.res_skip_layers.2.weight",
      "transform": "identity"
    },
    {
      "initializer": "flow.flows.0.enc.res_skip_layers.3.bias",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "flow.flows.0.enc.res_skip_layers.3.bias",
      "transform": "identity"
    },
    {
      "initializer": "flow.flows.0.enc.res_skip_layers.3.weight",
      "shape": [
        192,
        192,
        1
      ],
      "storage": "weight_norm_reconstructed",
      "target": "flow.flows.0.enc.res_skip_layers.3.weight",
      "transform": "identity"
    },
    {
      "initializer": "flow.flows.0.post.weight",
      "shape": [
        96,
        192,
        1
      ],
      "storage": "direct",
      "target": "flow.flows.0.post.weight",
      "transform": "identity"
    },
    {
      "initializer": "flow.flows.0.post.bias",
      "shape": [
        96
      ],
      "storage": "direct",
      "target": "flow.flows.0.post.bias",
      "transform": "identity"
    },
    {
      "initializer": "flow.flows.2.pre.weight",
      "shape": [
        192,
        96,
        1
      ],
      "storage": "direct",
      "target": "flow.flows.2.pre.weight",
      "transform": "identity"
    },
    {
      "initializer": "flow.flows.2.pre.bias",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "flow.flows.2.pre.bias",
      "transform": "identity"
    },
    {
      "initializer": "flow.flows.2.enc.in_layers.0.bias",
      "shape": [
        384
      ],
      "storage": "direct",
      "target": "flow.flows.2.enc.in_layers.0.bias",
      "transform": "identity"
    },
    {
      "initializer": "flow.flows.2.enc.in_layers.0.weight",
      "shape": [
        384,
        192,
        5
      ],
      "storage": "weight_norm_reconstructed",
      "target": "flow.flows.2.enc.in_layers.0.weight",
      "transform": "identity"
    },
    {
      "initializer": "flow.flows.2.enc.in_layers.1.bias",
      "shape": [
        384
      ],
      "storage": "direct",
      "target": "flow.flows.2.enc.in_layers.1.bias",
      "transform": "identity"
    },
    {
      "initializer": "flow.flows.2.enc.in_layers.1.weight",
      "shape": [
        384,
        192,
        5
      ],
      "storage": "weight_norm_reconstructed",
      "target": "flow.flows.2.enc.in_layers.1.weight",
      "transform": "identity"
    },
    {
      "initializer": "flow.flows.2.enc.in_layers.2.bias",
      "shape": [
        384
      ],
      "storage": "direct",
      "target": "flow.flows.2.enc.in_layers.2.bias",
      "transform": "identity"
    },
    {
      "initializer": "flow.flows.2.enc.in_layers.2.weight",
      "shape": [
        384,
        192,
        5
      ],
      "storage": "weight_norm_reconstructed",
      "target": "flow.flows.2.enc.in_layers.2.weight",
      "transform": "identity"
    },
    {
      "initializer": "flow.flows.2.enc.in_layers.3.bias",
      "shape": [
        384
      ],
      "storage": "direct",
      "target": "flow.flows.2.enc.in_layers.3.bias",
      "transform": "identity"
    },
    {
      "initializer": "flow.flows.2.enc.in_layers.3.weight",
      "shape": [
        384,
        192,
        5
      ],
      "storage": "weight_norm_reconstructed",
      "target": "flow.flows.2.enc.in_layers.3.weight",
      "transform": "identity"
    },
    {
      "initializer": "flow.flows.2.enc.res_skip_layers.0.bias",
      "shape": [
        384
      ],
      "storage": "direct",
      "target": "flow.flows.2.enc.res_skip_layers.0.bias",
      "transform": "identity"
    },
    {
      "initializer": "flow.flows.2.enc.res_skip_layers.0.weight",
      "shape": [
        384,
        192,
        1
      ],
      "storage": "weight_norm_reconstructed",
      "target": "flow.flows.2.enc.res_skip_layers.0.weight",
      "transform": "identity"
    },
    {
      "initializer": "flow.flows.2.enc.res_skip_layers.1.bias",
      "shape": [
        384
      ],
      "storage": "direct",
      "target": "flow.flows.2.enc.res_skip_layers.1.bias",
      "transform": "identity"
    },
    {
      "initializer": "flow.flows.2.enc.res_skip_layers.1.weight",
      "shape": [
        384,
        192,
        1
      ],
      "storage": "weight_norm_reconstructed",
      "target": "flow.flows.2.enc.res_skip_layers.1.weight",
      "transform": "identity"
    },
    {
      "initializer": "flow.flows.2.enc.res_skip_layers.2.bias",
      "shape": [
        384
      ],
      "storage": "direct",
      "target": "flow.flows.2.enc.res_skip_layers.2.bias",
      "transform": "identity"
    },
    {
      "initializer": "flow.flows.2.enc.res_skip_layers.2.weight",
      "shape": [
        384,
        192,
        1
      ],
      "storage": "weight_norm_reconstructed",
      "target": "flow.flows.2.enc.res_skip_layers.2.weight",
      "transform": "identity"
    },
    {
      "initializer": "flow.flows.2.enc.res_skip_layers.3.bias",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "flow.flows.2.enc.res_skip_layers.3.bias",
      "transform": "identity"
    },
    {
      "initializer": "flow.flows.2.enc.res_skip_layers.3.weight",
      "shape": [
        192,
        192,
        1
      ],
      "storage": "weight_norm_reconstructed",
      "target": "flow.flows.2.enc.res_skip_layers.3.weight",
      "transform": "identity"
    },
    {
      "initializer": "flow.flows.2.post.weight",
      "shape": [
        96,
        192,
        1
      ],
      "storage": "direct",
      "target": "flow.flows.2.post.weight",
      "transform": "identity"
    },
    {
      "initializer": "flow.flows.2.post.bias",
      "shape": [
        96
      ],
      "storage": "direct",
      "target": "flow.flows.2.post.bias",
      "transform": "identity"
    },
    {
      "initializer": "flow.flows.4.pre.weight",
      "shape": [
        192,
        96,
        1
      ],
      "storage": "direct",
      "target": "flow.flows.4.pre.weight",
      "transform": "identity"
    },
    {
      "initializer": "flow.flows.4.pre.bias",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "flow.flows.4.pre.bias",
      "transform": "identity"
    },
    {
      "initializer": "flow.flows.4.enc.in_layers.0.bias",
      "shape": [
        384
      ],
      "storage": "direct",
      "target": "flow.flows.4.enc.in_layers.0.bias",
      "transform": "identity"
    },
    {
      "initializer": "flow.flows.4.enc.in_layers.0.weight",
      "shape": [
        384,
        192,
        5
      ],
      "storage": "weight_norm_reconstructed",
      "target": "flow.flows.4.enc.in_layers.0.weight",
      "transform": "identity"
    },
    {
      "initializer": "flow.flows.4.enc.in_layers.1.bias",
      "shape": [
        384
      ],
      "storage": "direct",
      "target": "flow.flows.4.enc.in_layers.1.bias",
      "transform": "identity"
    },
    {
      "initializer": "flow.flows.4.enc.in_layers.1.weight",
      "shape": [
        384,
        192,
        5
      ],
      "storage": "weight_norm_reconstructed",
      "target": "flow.flows.4.enc.in_layers.1.weight",
      "transform": "identity"
    },
    {
      "initializer": "flow.flows.4.enc.in_layers.2.bias",
      "shape": [
        384
      ],
      "storage": "direct",
      "target": "flow.flows.4.enc.in_layers.2.bias",
      "transform": "identity"
    },
    {
      "initializer": "flow.flows.4.enc.in_layers.2.weight",
      "shape": [
        384,
        192,
        5
      ],
      "storage": "weight_norm_reconstructed",
      "target": "flow.flows.4.enc.in_layers.2.weight",
      "transform": "identity"
    },
    {
      "initializer": "flow.flows.4.enc.in_layers.3.bias",
      "shape": [
        384
      ],
      "storage": "direct",
      "target": "flow.flows.4.enc.in_layers.3.bias",
      "transform": "identity"
    },
    {
      "initializer": "flow.flows.4.enc.in_layers.3.weight",
      "shape": [
        384,
        192,
        5
      ],
      "storage": "weight_norm_reconstructed",
      "target": "flow.flows.4.enc.in_layers.3.weight",
      "transform": "identity"
    },
    {
      "initializer": "flow.flows.4.enc.res_skip_layers.0.bias",
      "shape": [
        384
      ],
      "storage": "direct",
      "target": "flow.flows.4.enc.res_skip_layers.0.bias",
      "transform": "identity"
    },
    {
      "initializer": "flow.flows.4.enc.res_skip_layers.0.weight",
      "shape": [
        384,
        192,
        1
      ],
      "storage": "weight_norm_reconstructed",
      "target": "flow.flows.4.enc.res_skip_layers.0.weight",
      "transform": "identity"
    },
    {
      "initializer": "flow.flows.4.enc.res_skip_layers.1.bias",
      "shape": [
        384
      ],
      "storage": "direct",
      "target": "flow.flows.4.enc.res_skip_layers.1.bias",
      "transform": "identity"
    },
    {
      "initializer": "flow.flows.4.enc.res_skip_layers.1.weight",
      "shape": [
        384,
        192,
        1
      ],
      "storage": "weight_norm_reconstructed",
      "target": "flow.flows.4.enc.res_skip_layers.1.weight",
      "transform": "identity"
    },
    {
      "initializer": "flow.flows.4.enc.res_skip_layers.2.bias",
      "shape": [
        384
      ],
      "storage": "direct",
      "target": "flow.flows.4.enc.res_skip_layers.2.bias",
      "transform": "identity"
    },
    {
      "initializer": "flow.flows.4.enc.res_skip_layers.2.weight",
      "shape": [
        384,
        192,
        1
      ],
      "storage": "weight_norm_reconstructed",
      "target": "flow.flows.4.enc.res_skip_layers.2.weight",
      "transform": "identity"
    },
    {
      "initializer": "flow.flows.4.enc.res_skip_layers.3.bias",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "flow.flows.4.enc.res_skip_layers.3.bias",
      "transform": "identity"
    },
    {
      "initializer": "flow.flows.4.enc.res_skip_layers.3.weight",
      "shape": [
        192,
        192,
        1
      ],
      "storage": "weight_norm_reconstructed",
      "target": "flow.flows.4.enc.res_skip_layers.3.weight",
      "transform": "identity"
    },
    {
      "initializer": "flow.flows.4.post.weight",
      "shape": [
        96,
        192,
        1
      ],
      "storage": "direct",
      "target": "flow.flows.4.post.weight",
      "transform": "identity"
    },
    {
      "initializer": "flow.flows.4.post.bias",
      "shape": [
        96
      ],
      "storage": "direct",
      "target": "flow.flows.4.post.bias",
      "transform": "identity"
    },
    {
      "initializer": "flow.flows.6.pre.weight",
      "shape": [
        192,
        96,
        1
      ],
      "storage": "direct",
      "target": "flow.flows.6.pre.weight",
      "transform": "identity"
    },
    {
      "initializer": "flow.flows.6.pre.bias",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "flow.flows.6.pre.bias",
      "transform": "identity"
    },
    {
      "initializer": "flow.flows.6.enc.in_layers.0.bias",
      "shape": [
        384
      ],
      "storage": "direct",
      "target": "flow.flows.6.enc.in_layers.0.bias",
      "transform": "identity"
    },
    {
      "initializer": "flow.flows.6.enc.in_layers.0.weight",
      "shape": [
        384,
        192,
        5
      ],
      "storage": "weight_norm_reconstructed",
      "target": "flow.flows.6.enc.in_layers.0.weight",
      "transform": "identity"
    },
    {
      "initializer": "flow.flows.6.enc.in_layers.1.bias",
      "shape": [
        384
      ],
      "storage": "direct",
      "target": "flow.flows.6.enc.in_layers.1.bias",
      "transform": "identity"
    },
    {
      "initializer": "flow.flows.6.enc.in_layers.1.weight",
      "shape": [
        384,
        192,
        5
      ],
      "storage": "weight_norm_reconstructed",
      "target": "flow.flows.6.enc.in_layers.1.weight",
      "transform": "identity"
    },
    {
      "initializer": "flow.flows.6.enc.in_layers.2.bias",
      "shape": [
        384
      ],
      "storage": "direct",
      "target": "flow.flows.6.enc.in_layers.2.bias",
      "transform": "identity"
    },
    {
      "initializer": "flow.flows.6.enc.in_layers.2.weight",
      "shape": [
        384,
        192,
        5
      ],
      "storage": "weight_norm_reconstructed",
      "target": "flow.flows.6.enc.in_layers.2.weight",
      "transform": "identity"
    },
    {
      "initializer": "flow.flows.6.enc.in_layers.3.bias",
      "shape": [
        384
      ],
      "storage": "direct",
      "target": "flow.flows.6.enc.in_layers.3.bias",
      "transform": "identity"
    },
    {
      "initializer": "flow.flows.6.enc.in_layers.3.weight",
      "shape": [
        384,
        192,
        5
      ],
      "storage": "weight_norm_reconstructed",
      "target": "flow.flows.6.enc.in_layers.3.weight",
      "transform": "identity"
    },
    {
      "initializer": "flow.flows.6.enc.res_skip_layers.0.bias",
      "shape": [
        384
      ],
      "storage": "direct",
      "target": "flow.flows.6.enc.res_skip_layers.0.bias",
      "transform": "identity"
    },
    {
      "initializer": "flow.flows.6.enc.res_skip_layers.0.weight",
      "shape": [
        384,
        192,
        1
      ],
      "storage": "weight_norm_reconstructed",
      "target": "flow.flows.6.enc.res_skip_layers.0.weight",
      "transform": "identity"
    },
    {
      "initializer": "flow.flows.6.enc.res_skip_layers.1.bias",
      "shape": [
        384
      ],
      "storage": "direct",
      "target": "flow.flows.6.enc.res_skip_layers.1.bias",
      "transform": "identity"
    },
    {
      "initializer": "flow.flows.6.enc.res_skip_layers.1.weight",
      "shape": [
        384,
        192,
        1
      ],
      "storage": "weight_norm_reconstructed",
      "target": "flow.flows.6.enc.res_skip_layers.1.weight",
      "transform": "identity"
    },
    {
      "initializer": "flow.flows.6.enc.res_skip_layers.2.bias",
      "shape": [
        384
      ],
      "storage": "direct",
      "target": "flow.flows.6.enc.res_skip_layers.2.bias",
      "transform": "identity"
    },
    {
      "initializer": "flow.flows.6.enc.res_skip_layers.2.weight",
      "shape": [
        384,
        192,
        1
      ],
      "storage": "weight_norm_reconstructed",
      "target": "flow.flows.6.enc.res_skip_layers.2.weight",
      "transform": "identity"
    },
    {
      "initializer": "flow.flows.6.enc.res_skip_layers.3.bias",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "flow.flows.6.enc.res_skip_layers.3.bias",
      "transform": "identity"
    },
    {
      "initializer": "flow.flows.6.enc.res_skip_layers.3.weight",
      "shape": [
        192,
        192,
        1
      ],
      "storage": "weight_norm_reconstructed",
      "target": "flow.flows.6.enc.res_skip_layers.3.weight",
      "transform": "identity"
    },
    {
      "initializer": "flow.flows.6.post.weight",
      "shape": [
        96,
        192,
        1
      ],
      "storage": "direct",
      "target": "flow.flows.6.post.weight",
      "transform": "identity"
    },
    {
      "initializer": "flow.flows.6.post.bias",
      "shape": [
        96
      ],
      "storage": "direct",
      "target": "flow.flows.6.post.bias",
      "transform": "identity"
    },
    {
      "initializer": "dp.flows.0.m",
      "shape": [
        2,
        1
      ],
      "storage": "direct",
      "target": "dp.flows.0.m",
      "transform": "identity"
    },
    {
      "initializer": "dp.flows.3.pre.weight",
      "shape": [
        192,
        1,
        1
      ],
      "storage": "direct",
      "target": "dp.flows.3.pre.weight",
      "transform": "identity"
    },
    {
      "initializer": "dp.flows.3.pre.bias",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "dp.flows.3.pre.bias",
      "transform": "identity"
    },
    {
      "initializer": "dp.flows.3.convs.convs_sep.0.weight",
      "shape": [
        192,
        1,
        3
      ],
      "storage": "direct",
      "target": "dp.flows.3.convs.convs_sep.0.weight",
      "transform": "identity"
    },
    {
      "initializer": "dp.flows.3.convs.convs_sep.0.bias",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "dp.flows.3.convs.convs_sep.0.bias",
      "transform": "identity"
    },
    {
      "initializer": "dp.flows.3.convs.convs_sep.1.weight",
      "shape": [
        192,
        1,
        3
      ],
      "storage": "direct",
      "target": "dp.flows.3.convs.convs_sep.1.weight",
      "transform": "identity"
    },
    {
      "initializer": "dp.flows.3.convs.convs_sep.1.bias",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "dp.flows.3.convs.convs_sep.1.bias",
      "transform": "identity"
    },
    {
      "initializer": "dp.flows.3.convs.convs_sep.2.weight",
      "shape": [
        192,
        1,
        3
      ],
      "storage": "direct",
      "target": "dp.flows.3.convs.convs_sep.2.weight",
      "transform": "identity"
    },
    {
      "initializer": "dp.flows.3.convs.convs_sep.2.bias",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "dp.flows.3.convs.convs_sep.2.bias",
      "transform": "identity"
    },
    {
      "initializer": "dp.flows.3.convs.convs_1x1.0.weight",
      "shape": [
        192,
        192,
        1
      ],
      "storage": "direct",
      "target": "dp.flows.3.convs.convs_1x1.0.weight",
      "transform": "identity"
    },
    {
      "initializer": "dp.flows.3.convs.convs_1x1.0.bias",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "dp.flows.3.convs.convs_1x1.0.bias",
      "transform": "identity"
    },
    {
      "initializer": "dp.flows.3.convs.convs_1x1.1.weight",
      "shape": [
        192,
        192,
        1
      ],
      "storage": "direct",
      "target": "dp.flows.3.convs.convs_1x1.1.weight",
      "transform": "identity"
    },
    {
      "initializer": "dp.flows.3.convs.convs_1x1.1.bias",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "dp.flows.3.convs.convs_1x1.1.bias",
      "transform": "identity"
    },
    {
      "initializer": "dp.flows.3.convs.convs_1x1.2.weight",
      "shape": [
        192,
        192,
        1
      ],
      "storage": "direct",
      "target": "dp.flows.3.convs.convs_1x1.2.weight",
      "transform": "identity"
    },
    {
      "initializer": "dp.flows.3.convs.convs_1x1.2.bias",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "dp.flows.3.convs.convs_1x1.2.bias",
      "transform": "identity"
    },
    {
      "initializer": "dp.flows.3.convs.norms_1.0.gamma",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "dp.flows.3.convs.norms_1.0.gamma",
      "transform": "identity"
    },
    {
      "initializer": "dp.flows.3.convs.norms_1.0.beta",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "dp.flows.3.convs.norms_1.0.beta",
      "transform": "identity"
    },
    {
      "initializer": "dp.flows.3.convs.norms_1.1.gamma",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "dp.flows.3.convs.norms_1.1.gamma",
      "transform": "identity"
    },
    {
      "initializer": "dp.flows.3.convs.norms_1.1.beta",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "dp.flows.3.convs.norms_1.1.beta",
      "transform": "identity"
    },
    {
      "initializer": "dp.flows.3.convs.norms_1.2.gamma",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "dp.flows.3.convs.norms_1.2.gamma",
      "transform": "identity"
    },
    {
      "initializer": "dp.flows.3.convs.norms_1.2.beta",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "dp.flows.3.convs.norms_1.2.beta",
      "transform": "identity"
    },
    {
      "initializer": "dp.flows.3.convs.norms_2.0.gamma",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "dp.flows.3.convs.norms_2.0.gamma",
      "transform": "identity"
    },
    {
      "initializer": "dp.flows.3.convs.norms_2.0.beta",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "dp.flows.3.convs.norms_2.0.beta",
      "transform": "identity"
    },
    {
      "initializer": "dp.flows.3.convs.norms_2.1.gamma",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "dp.flows.3.convs.norms_2.1.gamma",
      "transform": "identity"
    },
    {
      "initializer": "dp.flows.3.convs.norms_2.1.beta",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "dp.flows.3.convs.norms_2.1.beta",
      "transform": "identity"
    },
    {
      "initializer": "dp.flows.3.convs.norms_2.2.gamma",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "dp.flows.3.convs.norms_2.2.gamma",
      "transform": "identity"
    },
    {
      "initializer": "dp.flows.3.convs.norms_2.2.beta",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "dp.flows.3.convs.norms_2.2.beta",
      "transform": "identity"
    },
    {
      "initializer": "dp.flows.3.proj.weight",
      "shape": [
        29,
        192,
        1
      ],
      "storage": "direct",
      "target": "dp.flows.3.proj.weight",
      "transform": "identity"
    },
    {
      "initializer": "dp.flows.3.proj.bias",
      "shape": [
        29
      ],
      "storage": "direct",
      "target": "dp.flows.3.proj.bias",
      "transform": "identity"
    },
    {
      "initializer": "dp.flows.5.pre.weight",
      "shape": [
        192,
        1,
        1
      ],
      "storage": "direct",
      "target": "dp.flows.5.pre.weight",
      "transform": "identity"
    },
    {
      "initializer": "dp.flows.5.pre.bias",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "dp.flows.5.pre.bias",
      "transform": "identity"
    },
    {
      "initializer": "dp.flows.5.convs.convs_sep.0.weight",
      "shape": [
        192,
        1,
        3
      ],
      "storage": "direct",
      "target": "dp.flows.5.convs.convs_sep.0.weight",
      "transform": "identity"
    },
    {
      "initializer": "dp.flows.5.convs.convs_sep.0.bias",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "dp.flows.5.convs.convs_sep.0.bias",
      "transform": "identity"
    },
    {
      "initializer": "dp.flows.5.convs.convs_sep.1.weight",
      "shape": [
        192,
        1,
        3
      ],
      "storage": "direct",
      "target": "dp.flows.5.convs.convs_sep.1.weight",
      "transform": "identity"
    },
    {
      "initializer": "dp.flows.5.convs.convs_sep.1.bias",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "dp.flows.5.convs.convs_sep.1.bias",
      "transform": "identity"
    },
    {
      "initializer": "dp.flows.5.convs.convs_sep.2.weight",
      "shape": [
        192,
        1,
        3
      ],
      "storage": "direct",
      "target": "dp.flows.5.convs.convs_sep.2.weight",
      "transform": "identity"
    },
    {
      "initializer": "dp.flows.5.convs.convs_sep.2.bias",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "dp.flows.5.convs.convs_sep.2.bias",
      "transform": "identity"
    },
    {
      "initializer": "dp.flows.5.convs.convs_1x1.0.weight",
      "shape": [
        192,
        192,
        1
      ],
      "storage": "direct",
      "target": "dp.flows.5.convs.convs_1x1.0.weight",
      "transform": "identity"
    },
    {
      "initializer": "dp.flows.5.convs.convs_1x1.0.bias",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "dp.flows.5.convs.convs_1x1.0.bias",
      "transform": "identity"
    },
    {
      "initializer": "dp.flows.5.convs.convs_1x1.1.weight",
      "shape": [
        192,
        192,
        1
      ],
      "storage": "direct",
      "target": "dp.flows.5.convs.convs_1x1.1.weight",
      "transform": "identity"
    },
    {
      "initializer": "dp.flows.5.convs.convs_1x1.1.bias",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "dp.flows.5.convs.convs_1x1.1.bias",
      "transform": "identity"
    },
    {
      "initializer": "dp.flows.5.convs.convs_1x1.2.weight",
      "shape": [
        192,
        192,
        1
      ],
      "storage": "direct",
      "target": "dp.flows.5.convs.convs_1x1.2.weight",
      "transform": "identity"
    },
    {
      "initializer": "dp.flows.5.convs.convs_1x1.2.bias",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "dp.flows.5.convs.convs_1x1.2.bias",
      "transform": "identity"
    },
    {
      "initializer": "dp.flows.5.convs.norms_1.0.gamma",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "dp.flows.5.convs.norms_1.0.gamma",
      "transform": "identity"
    },
    {
      "initializer": "dp.flows.5.convs.norms_1.0.beta",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "dp.flows.5.convs.norms_1.0.beta",
      "transform": "identity"
    },
    {
      "initializer": "dp.flows.5.convs.norms_1.1.gamma",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "dp.flows.5.convs.norms_1.1.gamma",
      "transform": "identity"
    },
    {
      "initializer": "dp.flows.5.convs.norms_1.1.beta",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "dp.flows.5.convs.norms_1.1.beta",
      "transform": "identity"
    },
    {
      "initializer": "dp.flows.5.convs.norms_1.2.gamma",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "dp.flows.5.convs.norms_1.2.gamma",
      "transform": "identity"
    },
    {
      "initializer": "dp.flows.5.convs.norms_1.2.beta",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "dp.flows.5.convs.norms_1.2.beta",
      "transform": "identity"
    },
    {
      "initializer": "dp.flows.5.convs.norms_2.0.gamma",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "dp.flows.5.convs.norms_2.0.gamma",
      "transform": "identity"
    },
    {
      "initializer": "dp.flows.5.convs.norms_2.0.beta",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "dp.flows.5.convs.norms_2.0.beta",
      "transform": "identity"
    },
    {
      "initializer": "dp.flows.5.convs.norms_2.1.gamma",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "dp.flows.5.convs.norms_2.1.gamma",
      "transform": "identity"
    },
    {
      "initializer": "dp.flows.5.convs.norms_2.1.beta",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "dp.flows.5.convs.norms_2.1.beta",
      "transform": "identity"
    },
    {
      "initializer": "dp.flows.5.convs.norms_2.2.gamma",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "dp.flows.5.convs.norms_2.2.gamma",
      "transform": "identity"
    },
    {
      "initializer": "dp.flows.5.convs.norms_2.2.beta",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "dp.flows.5.convs.norms_2.2.beta",
      "transform": "identity"
    },
    {
      "initializer": "dp.flows.5.proj.weight",
      "shape": [
        29,
        192,
        1
      ],
      "storage": "direct",
      "target": "dp.flows.5.proj.weight",
      "transform": "identity"
    },
    {
      "initializer": "dp.flows.5.proj.bias",
      "shape": [
        29
      ],
      "storage": "direct",
      "target": "dp.flows.5.proj.bias",
      "transform": "identity"
    },
    {
      "initializer": "dp.flows.7.pre.weight",
      "shape": [
        192,
        1,
        1
      ],
      "storage": "direct",
      "target": "dp.flows.7.pre.weight",
      "transform": "identity"
    },
    {
      "initializer": "dp.flows.7.pre.bias",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "dp.flows.7.pre.bias",
      "transform": "identity"
    },
    {
      "initializer": "dp.flows.7.convs.convs_sep.0.weight",
      "shape": [
        192,
        1,
        3
      ],
      "storage": "direct",
      "target": "dp.flows.7.convs.convs_sep.0.weight",
      "transform": "identity"
    },
    {
      "initializer": "dp.flows.7.convs.convs_sep.0.bias",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "dp.flows.7.convs.convs_sep.0.bias",
      "transform": "identity"
    },
    {
      "initializer": "dp.flows.7.convs.convs_sep.1.weight",
      "shape": [
        192,
        1,
        3
      ],
      "storage": "direct",
      "target": "dp.flows.7.convs.convs_sep.1.weight",
      "transform": "identity"
    },
    {
      "initializer": "dp.flows.7.convs.convs_sep.1.bias",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "dp.flows.7.convs.convs_sep.1.bias",
      "transform": "identity"
    },
    {
      "initializer": "dp.flows.7.convs.convs_sep.2.weight",
      "shape": [
        192,
        1,
        3
      ],
      "storage": "direct",
      "target": "dp.flows.7.convs.convs_sep.2.weight",
      "transform": "identity"
    },
    {
      "initializer": "dp.flows.7.convs.convs_sep.2.bias",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "dp.flows.7.convs.convs_sep.2.bias",
      "transform": "identity"
    },
    {
      "initializer": "dp.flows.7.convs.convs_1x1.0.weight",
      "shape": [
        192,
        192,
        1
      ],
      "storage": "direct",
      "target": "dp.flows.7.convs.convs_1x1.0.weight",
      "transform": "identity"
    },
    {
      "initializer": "dp.flows.7.convs.convs_1x1.0.bias",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "dp.flows.7.convs.convs_1x1.0.bias",
      "transform": "identity"
    },
    {
      "initializer": "dp.flows.7.convs.convs_1x1.1.weight",
      "shape": [
        192,
        192,
        1
      ],
      "storage": "direct",
      "target": "dp.flows.7.convs.convs_1x1.1.weight",
      "transform": "identity"
    },
    {
      "initializer": "dp.flows.7.convs.convs_1x1.1.bias",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "dp.flows.7.convs.convs_1x1.1.bias",
      "transform": "identity"
    },
    {
      "initializer": "dp.flows.7.convs.convs_1x1.2.weight",
      "shape": [
        192,
        192,
        1
      ],
      "storage": "direct",
      "target": "dp.flows.7.convs.convs_1x1.2.weight",
      "transform": "identity"
    },
    {
      "initializer": "dp.flows.7.convs.convs_1x1.2.bias",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "dp.flows.7.convs.convs_1x1.2.bias",
      "transform": "identity"
    },
    {
      "initializer": "dp.flows.7.convs.norms_1.0.gamma",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "dp.flows.7.convs.norms_1.0.gamma",
      "transform": "identity"
    },
    {
      "initializer": "dp.flows.7.convs.norms_1.0.beta",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "dp.flows.7.convs.norms_1.0.beta",
      "transform": "identity"
    },
    {
      "initializer": "dp.flows.7.convs.norms_1.1.gamma",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "dp.flows.7.convs.norms_1.1.gamma",
      "transform": "identity"
    },
    {
      "initializer": "dp.flows.7.convs.norms_1.1.beta",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "dp.flows.7.convs.norms_1.1.beta",
      "transform": "identity"
    },
    {
      "initializer": "dp.flows.7.convs.norms_1.2.gamma",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "dp.flows.7.convs.norms_1.2.gamma",
      "transform": "identity"
    },
    {
      "initializer": "dp.flows.7.convs.norms_1.2.beta",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "dp.flows.7.convs.norms_1.2.beta",
      "transform": "identity"
    },
    {
      "initializer": "dp.flows.7.convs.norms_2.0.gamma",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "dp.flows.7.convs.norms_2.0.gamma",
      "transform": "identity"
    },
    {
      "initializer": "dp.flows.7.convs.norms_2.0.beta",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "dp.flows.7.convs.norms_2.0.beta",
      "transform": "identity"
    },
    {
      "initializer": "dp.flows.7.convs.norms_2.1.gamma",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "dp.flows.7.convs.norms_2.1.gamma",
      "transform": "identity"
    },
    {
      "initializer": "dp.flows.7.convs.norms_2.1.beta",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "dp.flows.7.convs.norms_2.1.beta",
      "transform": "identity"
    },
    {
      "initializer": "dp.flows.7.convs.norms_2.2.gamma",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "dp.flows.7.convs.norms_2.2.gamma",
      "transform": "identity"
    },
    {
      "initializer": "dp.flows.7.convs.norms_2.2.beta",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "dp.flows.7.convs.norms_2.2.beta",
      "transform": "identity"
    },
    {
      "initializer": "dp.flows.7.proj.weight",
      "shape": [
        29,
        192,
        1
      ],
      "storage": "direct",
      "target": "dp.flows.7.proj.weight",
      "transform": "identity"
    },
    {
      "initializer": "dp.flows.7.proj.bias",
      "shape": [
        29
      ],
      "storage": "direct",
      "target": "dp.flows.7.proj.bias",
      "transform": "identity"
    },
    {
      "initializer": "dp.pre.weight",
      "shape": [
        192,
        192,
        1
      ],
      "storage": "direct",
      "target": "dp.pre.weight",
      "transform": "identity"
    },
    {
      "initializer": "dp.pre.bias",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "dp.pre.bias",
      "transform": "identity"
    },
    {
      "initializer": "dp.proj.weight",
      "shape": [
        192,
        192,
        1
      ],
      "storage": "direct",
      "target": "dp.proj.weight",
      "transform": "identity"
    },
    {
      "initializer": "dp.proj.bias",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "dp.proj.bias",
      "transform": "identity"
    },
    {
      "initializer": "dp.convs.convs_sep.0.weight",
      "shape": [
        192,
        1,
        3
      ],
      "storage": "direct",
      "target": "dp.convs.convs_sep.0.weight",
      "transform": "identity"
    },
    {
      "initializer": "dp.convs.convs_sep.0.bias",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "dp.convs.convs_sep.0.bias",
      "transform": "identity"
    },
    {
      "initializer": "dp.convs.convs_sep.1.weight",
      "shape": [
        192,
        1,
        3
      ],
      "storage": "direct",
      "target": "dp.convs.convs_sep.1.weight",
      "transform": "identity"
    },
    {
      "initializer": "dp.convs.convs_sep.1.bias",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "dp.convs.convs_sep.1.bias",
      "transform": "identity"
    },
    {
      "initializer": "dp.convs.convs_sep.2.weight",
      "shape": [
        192,
        1,
        3
      ],
      "storage": "direct",
      "target": "dp.convs.convs_sep.2.weight",
      "transform": "identity"
    },
    {
      "initializer": "dp.convs.convs_sep.2.bias",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "dp.convs.convs_sep.2.bias",
      "transform": "identity"
    },
    {
      "initializer": "dp.convs.convs_1x1.0.weight",
      "shape": [
        192,
        192,
        1
      ],
      "storage": "direct",
      "target": "dp.convs.convs_1x1.0.weight",
      "transform": "identity"
    },
    {
      "initializer": "dp.convs.convs_1x1.0.bias",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "dp.convs.convs_1x1.0.bias",
      "transform": "identity"
    },
    {
      "initializer": "dp.convs.convs_1x1.1.weight",
      "shape": [
        192,
        192,
        1
      ],
      "storage": "direct",
      "target": "dp.convs.convs_1x1.1.weight",
      "transform": "identity"
    },
    {
      "initializer": "dp.convs.convs_1x1.1.bias",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "dp.convs.convs_1x1.1.bias",
      "transform": "identity"
    },
    {
      "initializer": "dp.convs.convs_1x1.2.weight",
      "shape": [
        192,
        192,
        1
      ],
      "storage": "direct",
      "target": "dp.convs.convs_1x1.2.weight",
      "transform": "identity"
    },
    {
      "initializer": "dp.convs.convs_1x1.2.bias",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "dp.convs.convs_1x1.2.bias",
      "transform": "identity"
    },
    {
      "initializer": "dp.convs.norms_1.0.gamma",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "dp.convs.norms_1.0.gamma",
      "transform": "identity"
    },
    {
      "initializer": "dp.convs.norms_1.0.beta",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "dp.convs.norms_1.0.beta",
      "transform": "identity"
    },
    {
      "initializer": "dp.convs.norms_1.1.gamma",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "dp.convs.norms_1.1.gamma",
      "transform": "identity"
    },
    {
      "initializer": "dp.convs.norms_1.1.beta",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "dp.convs.norms_1.1.beta",
      "transform": "identity"
    },
    {
      "initializer": "dp.convs.norms_1.2.gamma",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "dp.convs.norms_1.2.gamma",
      "transform": "identity"
    },
    {
      "initializer": "dp.convs.norms_1.2.beta",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "dp.convs.norms_1.2.beta",
      "transform": "identity"
    },
    {
      "initializer": "dp.convs.norms_2.0.gamma",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "dp.convs.norms_2.0.gamma",
      "transform": "identity"
    },
    {
      "initializer": "dp.convs.norms_2.0.beta",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "dp.convs.norms_2.0.beta",
      "transform": "identity"
    },
    {
      "initializer": "dp.convs.norms_2.1.gamma",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "dp.convs.norms_2.1.gamma",
      "transform": "identity"
    },
    {
      "initializer": "dp.convs.norms_2.1.beta",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "dp.convs.norms_2.1.beta",
      "transform": "identity"
    },
    {
      "initializer": "dp.convs.norms_2.2.gamma",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "dp.convs.norms_2.2.gamma",
      "transform": "identity"
    },
    {
      "initializer": "dp.convs.norms_2.2.beta",
      "shape": [
        192
      ],
      "storage": "direct",
      "target": "dp.convs.norms_2.2.beta",
      "transform": "identity"
    },
    {
      "initializer": "onnx::Exp_8793",
      "shape": [
        2,
        1
      ],
      "storage": "direct",
      "target": "dp.flows.0.logs",
      "transform": "negate"
    }
  ],
  "training_only_initialization": {
    "modules": [
      "model_g.enc_q",
      "model_g.dp training-only branches",
      "model_d"
    ],
    "source": "Mana parent checkpoint"
  }
}