checkpoint_9120/array_metadatas/process_0

{"array_metadatas": [{"array_metadata": {"param_name": "params.conv_in.bias", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.conv_in.kernel", "write_shape": [3, 3, 4, 320], "chunk_shape": [3, 3, 4, 320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.conv_norm_out.bias", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.conv_norm_out.scale", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.conv_out.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.conv_out.kernel", "write_shape": [3, 3, 320, 4], "chunk_shape": [3, 3, 320, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_0.attentions_0.norm.bias", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_0.attentions_0.norm.scale", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_0.attentions_0.proj_in.bias", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_0.attentions_0.proj_in.kernel", "write_shape": [1, 1, 320, 320], "chunk_shape": [1, 1, 320, 320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_0.attentions_0.proj_out.bias", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_0.attentions_0.proj_out.kernel", "write_shape": [1, 1, 320, 320], "chunk_shape": [1, 1, 320, 320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_0.attentions_0.transformer_blocks_0.attn1.to_k.kernel", "write_shape": [320, 320], "chunk_shape": [320, 320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_0.attentions_0.transformer_blocks_0.attn1.to_out_0.bias", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_0.attentions_0.transformer_blocks_0.attn1.to_out_0.kernel", "write_shape": [320, 320], "chunk_shape": [320, 320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_0.attentions_0.transformer_blocks_0.attn1.to_q.kernel", "write_shape": [320, 320], "chunk_shape": [320, 320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_0.attentions_0.transformer_blocks_0.attn1.to_v.kernel", "write_shape": [320, 320], "chunk_shape": [320, 320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_0.attentions_0.transformer_blocks_0.attn2.to_k.kernel", "write_shape": [768, 320], "chunk_shape": [768, 320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_0.attentions_0.transformer_blocks_0.attn2.to_out_0.bias", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_0.attentions_0.transformer_blocks_0.attn2.to_out_0.kernel", "write_shape": [320, 320], "chunk_shape": [320, 320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_0.attentions_0.transformer_blocks_0.attn2.to_q.kernel", "write_shape": [320, 320], "chunk_shape": [320, 320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_0.attentions_0.transformer_blocks_0.attn2.to_v.kernel", "write_shape": [768, 320], "chunk_shape": [768, 320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_0.attentions_0.transformer_blocks_0.ff.net_0.proj.bias", "write_shape": [2560], "chunk_shape": [2560], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_0.attentions_0.transformer_blocks_0.ff.net_0.proj.kernel", "write_shape": [320, 2560], "chunk_shape": [320, 2560], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_0.attentions_0.transformer_blocks_0.ff.net_2.bias", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_0.attentions_0.transformer_blocks_0.ff.net_2.kernel", "write_shape": [1280, 320], "chunk_shape": [1280, 320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_0.attentions_0.transformer_blocks_0.norm1.bias", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_0.attentions_0.transformer_blocks_0.norm1.scale", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_0.attentions_0.transformer_blocks_0.norm2.bias", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_0.attentions_0.transformer_blocks_0.norm2.scale", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_0.attentions_0.transformer_blocks_0.norm3.bias", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_0.attentions_0.transformer_blocks_0.norm3.scale", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_0.attentions_1.norm.bias", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_0.attentions_1.norm.scale", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_0.attentions_1.proj_in.bias", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_0.attentions_1.proj_in.kernel", "write_shape": [1, 1, 320, 320], "chunk_shape": [1, 1, 320, 320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_0.attentions_1.proj_out.bias", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_0.attentions_1.proj_out.kernel", "write_shape": [1, 1, 320, 320], "chunk_shape": [1, 1, 320, 320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_0.attentions_1.transformer_blocks_0.attn1.to_k.kernel", "write_shape": [320, 320], "chunk_shape": [320, 320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_0.attentions_1.transformer_blocks_0.attn1.to_out_0.bias", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_0.attentions_1.transformer_blocks_0.attn1.to_out_0.kernel", "write_shape": [320, 320], "chunk_shape": [320, 320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_0.attentions_1.transformer_blocks_0.attn1.to_q.kernel", "write_shape": [320, 320], "chunk_shape": [320, 320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_0.attentions_1.transformer_blocks_0.attn1.to_v.kernel", "write_shape": [320, 320], "chunk_shape": [320, 320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_0.attentions_1.transformer_blocks_0.attn2.to_k.kernel", "write_shape": [768, 320], "chunk_shape": [768, 320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_0.attentions_1.transformer_blocks_0.attn2.to_out_0.bias", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_0.attentions_1.transformer_blocks_0.attn2.to_out_0.kernel", "write_shape": [320, 320], "chunk_shape": [320, 320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_0.attentions_1.transformer_blocks_0.attn2.to_q.kernel", "write_shape": [320, 320], "chunk_shape": [320, 320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_0.attentions_1.transformer_blocks_0.attn2.to_v.kernel", "write_shape": [768, 320], "chunk_shape": [768, 320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_0.attentions_1.transformer_blocks_0.ff.net_0.proj.bias", "write_shape": [2560], "chunk_shape": [2560], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_0.attentions_1.transformer_blocks_0.ff.net_0.proj.kernel", "write_shape": [320, 2560], "chunk_shape": [320, 2560], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_0.attentions_1.transformer_blocks_0.ff.net_2.bias", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_0.attentions_1.transformer_blocks_0.ff.net_2.kernel", "write_shape": [1280, 320], "chunk_shape": [1280, 320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_0.attentions_1.transformer_blocks_0.norm1.bias", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_0.attentions_1.transformer_blocks_0.norm1.scale", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_0.attentions_1.transformer_blocks_0.norm2.bias", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_0.attentions_1.transformer_blocks_0.norm2.scale", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_0.attentions_1.transformer_blocks_0.norm3.bias", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_0.attentions_1.transformer_blocks_0.norm3.scale", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_0.downsamplers_0.conv.bias", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_0.downsamplers_0.conv.kernel", "write_shape": [3, 3, 320, 320], "chunk_shape": [3, 3, 320, 320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_0.resnets_0.conv1.bias", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_0.resnets_0.conv1.kernel", "write_shape": [3, 3, 320, 320], "chunk_shape": [3, 3, 320, 320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_0.resnets_0.conv2.bias", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_0.resnets_0.conv2.kernel", "write_shape": [3, 3, 320, 320], "chunk_shape": [3, 3, 320, 320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_0.resnets_0.norm1.bias", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_0.resnets_0.norm1.scale", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_0.resnets_0.norm2.bias", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_0.resnets_0.norm2.scale", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_0.resnets_0.time_emb_proj.bias", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_0.resnets_0.time_emb_proj.kernel", "write_shape": [1280, 320], "chunk_shape": [1280, 320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_0.resnets_1.conv1.bias", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_0.resnets_1.conv1.kernel", "write_shape": [3, 3, 320, 320], "chunk_shape": [3, 3, 320, 320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_0.resnets_1.conv2.bias", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_0.resnets_1.conv2.kernel", "write_shape": [3, 3, 320, 320], "chunk_shape": [3, 3, 320, 320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_0.resnets_1.norm1.bias", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_0.resnets_1.norm1.scale", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_0.resnets_1.norm2.bias", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_0.resnets_1.norm2.scale", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_0.resnets_1.time_emb_proj.bias", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_0.resnets_1.time_emb_proj.kernel", "write_shape": [1280, 320], "chunk_shape": [1280, 320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_1.attentions_0.norm.bias", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_1.attentions_0.norm.scale", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_1.attentions_0.proj_in.bias", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_1.attentions_0.proj_in.kernel", "write_shape": [1, 1, 640, 640], "chunk_shape": [1, 1, 640, 640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_1.attentions_0.proj_out.bias", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_1.attentions_0.proj_out.kernel", "write_shape": [1, 1, 640, 640], "chunk_shape": [1, 1, 640, 640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_1.attentions_0.transformer_blocks_0.attn1.to_k.kernel", "write_shape": [640, 640], "chunk_shape": [640, 640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_1.attentions_0.transformer_blocks_0.attn1.to_out_0.bias", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_1.attentions_0.transformer_blocks_0.attn1.to_out_0.kernel", "write_shape": [640, 640], "chunk_shape": [640, 640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_1.attentions_0.transformer_blocks_0.attn1.to_q.kernel", "write_shape": [640, 640], "chunk_shape": [640, 640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_1.attentions_0.transformer_blocks_0.attn1.to_v.kernel", "write_shape": [640, 640], "chunk_shape": [640, 640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_1.attentions_0.transformer_blocks_0.attn2.to_k.kernel", "write_shape": [768, 640], "chunk_shape": [768, 640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_1.attentions_0.transformer_blocks_0.attn2.to_out_0.bias", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_1.attentions_0.transformer_blocks_0.attn2.to_out_0.kernel", "write_shape": [640, 640], "chunk_shape": [640, 640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_1.attentions_0.transformer_blocks_0.attn2.to_q.kernel", "write_shape": [640, 640], "chunk_shape": [640, 640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_1.attentions_0.transformer_blocks_0.attn2.to_v.kernel", "write_shape": [768, 640], "chunk_shape": [768, 640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_1.attentions_0.transformer_blocks_0.ff.net_0.proj.bias", "write_shape": [5120], "chunk_shape": [5120], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_1.attentions_0.transformer_blocks_0.ff.net_0.proj.kernel", "write_shape": [640, 5120], "chunk_shape": [640, 5120], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_1.attentions_0.transformer_blocks_0.ff.net_2.bias", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_1.attentions_0.transformer_blocks_0.ff.net_2.kernel", "write_shape": [2560, 640], "chunk_shape": [2560, 640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_1.attentions_0.transformer_blocks_0.norm1.bias", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_1.attentions_0.transformer_blocks_0.norm1.scale", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_1.attentions_0.transformer_blocks_0.norm2.bias", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_1.attentions_0.transformer_blocks_0.norm2.scale", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_1.attentions_0.transformer_blocks_0.norm3.bias", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_1.attentions_0.transformer_blocks_0.norm3.scale", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_1.attentions_1.norm.bias", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_1.attentions_1.norm.scale", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_1.attentions_1.proj_in.bias", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_1.attentions_1.proj_in.kernel", "write_shape": [1, 1, 640, 640], "chunk_shape": [1, 1, 640, 640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_1.attentions_1.proj_out.bias", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_1.attentions_1.proj_out.kernel", "write_shape": [1, 1, 640, 640], "chunk_shape": [1, 1, 640, 640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_1.attentions_1.transformer_blocks_0.attn1.to_k.kernel", "write_shape": [640, 640], "chunk_shape": [640, 640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_1.attentions_1.transformer_blocks_0.attn1.to_out_0.bias", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_1.attentions_1.transformer_blocks_0.attn1.to_out_0.kernel", "write_shape": [640, 640], "chunk_shape": [640, 640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_1.attentions_1.transformer_blocks_0.attn1.to_q.kernel", "write_shape": [640, 640], "chunk_shape": [640, 640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_1.attentions_1.transformer_blocks_0.attn1.to_v.kernel", "write_shape": [640, 640], "chunk_shape": [640, 640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_1.attentions_1.transformer_blocks_0.attn2.to_k.kernel", "write_shape": [768, 640], "chunk_shape": [768, 640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_1.attentions_1.transformer_blocks_0.attn2.to_out_0.bias", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_1.attentions_1.transformer_blocks_0.attn2.to_out_0.kernel", "write_shape": [640, 640], "chunk_shape": [640, 640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_1.attentions_1.transformer_blocks_0.attn2.to_q.kernel", "write_shape": [640, 640], "chunk_shape": [640, 640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_1.attentions_1.transformer_blocks_0.attn2.to_v.kernel", "write_shape": [768, 640], "chunk_shape": [768, 640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_1.attentions_1.transformer_blocks_0.ff.net_0.proj.bias", "write_shape": [5120], "chunk_shape": [5120], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_1.attentions_1.transformer_blocks_0.ff.net_0.proj.kernel", "write_shape": [640, 5120], "chunk_shape": [640, 5120], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_1.attentions_1.transformer_blocks_0.ff.net_2.bias", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_1.attentions_1.transformer_blocks_0.ff.net_2.kernel", "write_shape": [2560, 640], "chunk_shape": [2560, 640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_1.attentions_1.transformer_blocks_0.norm1.bias", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_1.attentions_1.transformer_blocks_0.norm1.scale", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_1.attentions_1.transformer_blocks_0.norm2.bias", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_1.attentions_1.transformer_blocks_0.norm2.scale", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_1.attentions_1.transformer_blocks_0.norm3.bias", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_1.attentions_1.transformer_blocks_0.norm3.scale", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_1.downsamplers_0.conv.bias", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_1.downsamplers_0.conv.kernel", "write_shape": [3, 3, 640, 640], "chunk_shape": [3, 3, 640, 640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_1.resnets_0.conv1.bias", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_1.resnets_0.conv1.kernel", "write_shape": [3, 3, 320, 640], "chunk_shape": [3, 3, 320, 640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_1.resnets_0.conv2.bias", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_1.resnets_0.conv2.kernel", "write_shape": [3, 3, 640, 640], "chunk_shape": [3, 3, 640, 640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_1.resnets_0.conv_shortcut.bias", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_1.resnets_0.conv_shortcut.kernel", "write_shape": [1, 1, 320, 640], "chunk_shape": [1, 1, 320, 640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_1.resnets_0.norm1.bias", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_1.resnets_0.norm1.scale", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_1.resnets_0.norm2.bias", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_1.resnets_0.norm2.scale", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_1.resnets_0.time_emb_proj.bias", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_1.resnets_0.time_emb_proj.kernel", "write_shape": [1280, 640], "chunk_shape": [1280, 640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_1.resnets_1.conv1.bias", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_1.resnets_1.conv1.kernel", "write_shape": [3, 3, 640, 640], "chunk_shape": [3, 3, 640, 640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_1.resnets_1.conv2.bias", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_1.resnets_1.conv2.kernel", "write_shape": [3, 3, 640, 640], "chunk_shape": [3, 3, 640, 640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_1.resnets_1.norm1.bias", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_1.resnets_1.norm1.scale", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_1.resnets_1.norm2.bias", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_1.resnets_1.norm2.scale", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_1.resnets_1.time_emb_proj.bias", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_1.resnets_1.time_emb_proj.kernel", "write_shape": [1280, 640], "chunk_shape": [1280, 640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_2.attentions_0.norm.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_2.attentions_0.norm.scale", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_2.attentions_0.proj_in.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_2.attentions_0.proj_in.kernel", "write_shape": [1, 1, 1280, 1280], "chunk_shape": [1, 1, 1280, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_2.attentions_0.proj_out.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_2.attentions_0.proj_out.kernel", "write_shape": [1, 1, 1280, 1280], "chunk_shape": [1, 1, 1280, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_2.attentions_0.transformer_blocks_0.attn1.to_k.kernel", "write_shape": [1280, 1280], "chunk_shape": [1280, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_2.attentions_0.transformer_blocks_0.attn1.to_out_0.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_2.attentions_0.transformer_blocks_0.attn1.to_out_0.kernel", "write_shape": [1280, 1280], "chunk_shape": [1280, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_2.attentions_0.transformer_blocks_0.attn1.to_q.kernel", "write_shape": [1280, 1280], "chunk_shape": [1280, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_2.attentions_0.transformer_blocks_0.attn1.to_v.kernel", "write_shape": [1280, 1280], "chunk_shape": [1280, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_2.attentions_0.transformer_blocks_0.attn2.to_k.kernel", "write_shape": [768, 1280], "chunk_shape": [768, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_2.attentions_0.transformer_blocks_0.attn2.to_out_0.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_2.attentions_0.transformer_blocks_0.attn2.to_out_0.kernel", "write_shape": [1280, 1280], "chunk_shape": [1280, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_2.attentions_0.transformer_blocks_0.attn2.to_q.kernel", "write_shape": [1280, 1280], "chunk_shape": [1280, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_2.attentions_0.transformer_blocks_0.attn2.to_v.kernel", "write_shape": [768, 1280], "chunk_shape": [768, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_2.attentions_0.transformer_blocks_0.ff.net_0.proj.bias", "write_shape": [10240], "chunk_shape": [10240], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_2.attentions_0.transformer_blocks_0.ff.net_0.proj.kernel", "write_shape": [1280, 10240], "chunk_shape": [1280, 10240], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_2.attentions_0.transformer_blocks_0.ff.net_2.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_2.attentions_0.transformer_blocks_0.ff.net_2.kernel", "write_shape": [5120, 1280], "chunk_shape": [5120, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_2.attentions_0.transformer_blocks_0.norm1.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_2.attentions_0.transformer_blocks_0.norm1.scale", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_2.attentions_0.transformer_blocks_0.norm2.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_2.attentions_0.transformer_blocks_0.norm2.scale", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_2.attentions_0.transformer_blocks_0.norm3.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_2.attentions_0.transformer_blocks_0.norm3.scale", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_2.attentions_1.norm.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_2.attentions_1.norm.scale", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_2.attentions_1.proj_in.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_2.attentions_1.proj_in.kernel", "write_shape": [1, 1, 1280, 1280], "chunk_shape": [1, 1, 1280, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_2.attentions_1.proj_out.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_2.attentions_1.proj_out.kernel", "write_shape": [1, 1, 1280, 1280], "chunk_shape": [1, 1, 1280, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_2.attentions_1.transformer_blocks_0.attn1.to_k.kernel", "write_shape": [1280, 1280], "chunk_shape": [1280, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_2.attentions_1.transformer_blocks_0.attn1.to_out_0.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_2.attentions_1.transformer_blocks_0.attn1.to_out_0.kernel", "write_shape": [1280, 1280], "chunk_shape": [1280, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_2.attentions_1.transformer_blocks_0.attn1.to_q.kernel", "write_shape": [1280, 1280], "chunk_shape": [1280, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_2.attentions_1.transformer_blocks_0.attn1.to_v.kernel", "write_shape": [1280, 1280], "chunk_shape": [1280, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_2.attentions_1.transformer_blocks_0.attn2.to_k.kernel", "write_shape": [768, 1280], "chunk_shape": [768, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_2.attentions_1.transformer_blocks_0.attn2.to_out_0.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_2.attentions_1.transformer_blocks_0.attn2.to_out_0.kernel", "write_shape": [1280, 1280], "chunk_shape": [1280, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_2.attentions_1.transformer_blocks_0.attn2.to_q.kernel", "write_shape": [1280, 1280], "chunk_shape": [1280, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_2.attentions_1.transformer_blocks_0.attn2.to_v.kernel", "write_shape": [768, 1280], "chunk_shape": [768, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_2.attentions_1.transformer_blocks_0.ff.net_0.proj.bias", "write_shape": [10240], "chunk_shape": [10240], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_2.attentions_1.transformer_blocks_0.ff.net_0.proj.kernel", "write_shape": [1280, 10240], "chunk_shape": [1280, 10240], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_2.attentions_1.transformer_blocks_0.ff.net_2.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_2.attentions_1.transformer_blocks_0.ff.net_2.kernel", "write_shape": [5120, 1280], "chunk_shape": [5120, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_2.attentions_1.transformer_blocks_0.norm1.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_2.attentions_1.transformer_blocks_0.norm1.scale", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_2.attentions_1.transformer_blocks_0.norm2.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_2.attentions_1.transformer_blocks_0.norm2.scale", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_2.attentions_1.transformer_blocks_0.norm3.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_2.attentions_1.transformer_blocks_0.norm3.scale", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_2.downsamplers_0.conv.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_2.downsamplers_0.conv.kernel", "write_shape": [3, 3, 1280, 1280], "chunk_shape": [3, 3, 1280, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_2.resnets_0.conv1.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_2.resnets_0.conv1.kernel", "write_shape": [3, 3, 640, 1280], "chunk_shape": [3, 3, 640, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_2.resnets_0.conv2.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_2.resnets_0.conv2.kernel", "write_shape": [3, 3, 1280, 1280], "chunk_shape": [3, 3, 1280, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_2.resnets_0.conv_shortcut.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_2.resnets_0.conv_shortcut.kernel", "write_shape": [1, 1, 640, 1280], "chunk_shape": [1, 1, 640, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_2.resnets_0.norm1.bias", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_2.resnets_0.norm1.scale", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_2.resnets_0.norm2.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_2.resnets_0.norm2.scale", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_2.resnets_0.time_emb_proj.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_2.resnets_0.time_emb_proj.kernel", "write_shape": [1280, 1280], "chunk_shape": [1280, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_2.resnets_1.conv1.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_2.resnets_1.conv1.kernel", "write_shape": [3, 3, 1280, 1280], "chunk_shape": [3, 3, 1280, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_2.resnets_1.conv2.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_2.resnets_1.conv2.kernel", "write_shape": [3, 3, 1280, 1280], "chunk_shape": [3, 3, 1280, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_2.resnets_1.norm1.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_2.resnets_1.norm1.scale", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_2.resnets_1.norm2.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_2.resnets_1.norm2.scale", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_2.resnets_1.time_emb_proj.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_2.resnets_1.time_emb_proj.kernel", "write_shape": [1280, 1280], "chunk_shape": [1280, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_3.resnets_0.conv1.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_3.resnets_0.conv1.kernel", "write_shape": [3, 3, 1280, 1280], "chunk_shape": [3, 3, 1280, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_3.resnets_0.conv2.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_3.resnets_0.conv2.kernel", "write_shape": [3, 3, 1280, 1280], "chunk_shape": [3, 3, 1280, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_3.resnets_0.norm1.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_3.resnets_0.norm1.scale", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_3.resnets_0.norm2.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_3.resnets_0.norm2.scale", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_3.resnets_0.time_emb_proj.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_3.resnets_0.time_emb_proj.kernel", "write_shape": [1280, 1280], "chunk_shape": [1280, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_3.resnets_1.conv1.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_3.resnets_1.conv1.kernel", "write_shape": [3, 3, 1280, 1280], "chunk_shape": [3, 3, 1280, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_3.resnets_1.conv2.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_3.resnets_1.conv2.kernel", "write_shape": [3, 3, 1280, 1280], "chunk_shape": [3, 3, 1280, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_3.resnets_1.norm1.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_3.resnets_1.norm1.scale", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_3.resnets_1.norm2.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_3.resnets_1.norm2.scale", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_3.resnets_1.time_emb_proj.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.down_blocks_3.resnets_1.time_emb_proj.kernel", "write_shape": [1280, 1280], "chunk_shape": [1280, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.mid_block.attentions_0.norm.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.mid_block.attentions_0.norm.scale", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.mid_block.attentions_0.proj_in.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.mid_block.attentions_0.proj_in.kernel", "write_shape": [1, 1, 1280, 1280], "chunk_shape": [1, 1, 1280, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.mid_block.attentions_0.proj_out.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.mid_block.attentions_0.proj_out.kernel", "write_shape": [1, 1, 1280, 1280], "chunk_shape": [1, 1, 1280, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.mid_block.attentions_0.transformer_blocks_0.attn1.to_k.kernel", "write_shape": [1280, 1280], "chunk_shape": [1280, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.mid_block.attentions_0.transformer_blocks_0.attn1.to_out_0.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.mid_block.attentions_0.transformer_blocks_0.attn1.to_out_0.kernel", "write_shape": [1280, 1280], "chunk_shape": [1280, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.mid_block.attentions_0.transformer_blocks_0.attn1.to_q.kernel", "write_shape": [1280, 1280], "chunk_shape": [1280, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.mid_block.attentions_0.transformer_blocks_0.attn1.to_v.kernel", "write_shape": [1280, 1280], "chunk_shape": [1280, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.mid_block.attentions_0.transformer_blocks_0.attn2.to_k.kernel", "write_shape": [768, 1280], "chunk_shape": [768, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.mid_block.attentions_0.transformer_blocks_0.attn2.to_out_0.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.mid_block.attentions_0.transformer_blocks_0.attn2.to_out_0.kernel", "write_shape": [1280, 1280], "chunk_shape": [1280, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.mid_block.attentions_0.transformer_blocks_0.attn2.to_q.kernel", "write_shape": [1280, 1280], "chunk_shape": [1280, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.mid_block.attentions_0.transformer_blocks_0.attn2.to_v.kernel", "write_shape": [768, 1280], "chunk_shape": [768, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.mid_block.attentions_0.transformer_blocks_0.ff.net_0.proj.bias", "write_shape": [10240], "chunk_shape": [10240], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.mid_block.attentions_0.transformer_blocks_0.ff.net_0.proj.kernel", "write_shape": [1280, 10240], "chunk_shape": [1280, 10240], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.mid_block.attentions_0.transformer_blocks_0.ff.net_2.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.mid_block.attentions_0.transformer_blocks_0.ff.net_2.kernel", "write_shape": [5120, 1280], "chunk_shape": [5120, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.mid_block.attentions_0.transformer_blocks_0.norm1.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.mid_block.attentions_0.transformer_blocks_0.norm1.scale", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.mid_block.attentions_0.transformer_blocks_0.norm2.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.mid_block.attentions_0.transformer_blocks_0.norm2.scale", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.mid_block.attentions_0.transformer_blocks_0.norm3.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.mid_block.attentions_0.transformer_blocks_0.norm3.scale", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.mid_block.resnets_0.conv1.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.mid_block.resnets_0.conv1.kernel", "write_shape": [3, 3, 1280, 1280], "chunk_shape": [3, 3, 1280, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.mid_block.resnets_0.conv2.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.mid_block.resnets_0.conv2.kernel", "write_shape": [3, 3, 1280, 1280], "chunk_shape": [3, 3, 1280, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.mid_block.resnets_0.norm1.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.mid_block.resnets_0.norm1.scale", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.mid_block.resnets_0.norm2.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.mid_block.resnets_0.norm2.scale", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.mid_block.resnets_0.time_emb_proj.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.mid_block.resnets_0.time_emb_proj.kernel", "write_shape": [1280, 1280], "chunk_shape": [1280, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.mid_block.resnets_1.conv1.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.mid_block.resnets_1.conv1.kernel", "write_shape": [3, 3, 1280, 1280], "chunk_shape": [3, 3, 1280, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.mid_block.resnets_1.conv2.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.mid_block.resnets_1.conv2.kernel", "write_shape": [3, 3, 1280, 1280], "chunk_shape": [3, 3, 1280, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.mid_block.resnets_1.norm1.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.mid_block.resnets_1.norm1.scale", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.mid_block.resnets_1.norm2.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.mid_block.resnets_1.norm2.scale", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.mid_block.resnets_1.time_emb_proj.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.mid_block.resnets_1.time_emb_proj.kernel", "write_shape": [1280, 1280], "chunk_shape": [1280, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.time_embedding.linear_1.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.time_embedding.linear_1.kernel", "write_shape": [320, 1280], "chunk_shape": [320, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.time_embedding.linear_2.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.time_embedding.linear_2.kernel", "write_shape": [1280, 1280], "chunk_shape": [1280, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_0.resnets_0.conv1.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_0.resnets_0.conv1.kernel", "write_shape": [3, 3, 2560, 1280], "chunk_shape": [3, 3, 2560, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_0.resnets_0.conv2.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_0.resnets_0.conv2.kernel", "write_shape": [3, 3, 1280, 1280], "chunk_shape": [3, 3, 1280, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_0.resnets_0.conv_shortcut.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_0.resnets_0.conv_shortcut.kernel", "write_shape": [1, 1, 2560, 1280], "chunk_shape": [1, 1, 2560, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_0.resnets_0.norm1.bias", "write_shape": [2560], "chunk_shape": [2560], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_0.resnets_0.norm1.scale", "write_shape": [2560], "chunk_shape": [2560], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_0.resnets_0.norm2.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_0.resnets_0.norm2.scale", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_0.resnets_0.time_emb_proj.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_0.resnets_0.time_emb_proj.kernel", "write_shape": [1280, 1280], "chunk_shape": [1280, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_0.resnets_1.conv1.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_0.resnets_1.conv1.kernel", "write_shape": [3, 3, 2560, 1280], "chunk_shape": [3, 3, 2560, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_0.resnets_1.conv2.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_0.resnets_1.conv2.kernel", "write_shape": [3, 3, 1280, 1280], "chunk_shape": [3, 3, 1280, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_0.resnets_1.conv_shortcut.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_0.resnets_1.conv_shortcut.kernel", "write_shape": [1, 1, 2560, 1280], "chunk_shape": [1, 1, 2560, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_0.resnets_1.norm1.bias", "write_shape": [2560], "chunk_shape": [2560], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_0.resnets_1.norm1.scale", "write_shape": [2560], "chunk_shape": [2560], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_0.resnets_1.norm2.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_0.resnets_1.norm2.scale", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_0.resnets_1.time_emb_proj.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_0.resnets_1.time_emb_proj.kernel", "write_shape": [1280, 1280], "chunk_shape": [1280, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_0.resnets_2.conv1.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_0.resnets_2.conv1.kernel", "write_shape": [3, 3, 2560, 1280], "chunk_shape": [3, 3, 2560, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_0.resnets_2.conv2.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_0.resnets_2.conv2.kernel", "write_shape": [3, 3, 1280, 1280], "chunk_shape": [3, 3, 1280, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_0.resnets_2.conv_shortcut.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_0.resnets_2.conv_shortcut.kernel", "write_shape": [1, 1, 2560, 1280], "chunk_shape": [1, 1, 2560, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_0.resnets_2.norm1.bias", "write_shape": [2560], "chunk_shape": [2560], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_0.resnets_2.norm1.scale", "write_shape": [2560], "chunk_shape": [2560], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_0.resnets_2.norm2.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_0.resnets_2.norm2.scale", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_0.resnets_2.time_emb_proj.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_0.resnets_2.time_emb_proj.kernel", "write_shape": [1280, 1280], "chunk_shape": [1280, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_0.upsamplers_0.conv.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_0.upsamplers_0.conv.kernel", "write_shape": [3, 3, 1280, 1280], "chunk_shape": [3, 3, 1280, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.attentions_0.norm.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.attentions_0.norm.scale", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.attentions_0.proj_in.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.attentions_0.proj_in.kernel", "write_shape": [1, 1, 1280, 1280], "chunk_shape": [1, 1, 1280, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.attentions_0.proj_out.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.attentions_0.proj_out.kernel", "write_shape": [1, 1, 1280, 1280], "chunk_shape": [1, 1, 1280, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.attentions_0.transformer_blocks_0.attn1.to_k.kernel", "write_shape": [1280, 1280], "chunk_shape": [1280, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.attentions_0.transformer_blocks_0.attn1.to_out_0.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.attentions_0.transformer_blocks_0.attn1.to_out_0.kernel", "write_shape": [1280, 1280], "chunk_shape": [1280, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.attentions_0.transformer_blocks_0.attn1.to_q.kernel", "write_shape": [1280, 1280], "chunk_shape": [1280, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.attentions_0.transformer_blocks_0.attn1.to_v.kernel", "write_shape": [1280, 1280], "chunk_shape": [1280, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.attentions_0.transformer_blocks_0.attn2.to_k.kernel", "write_shape": [768, 1280], "chunk_shape": [768, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.attentions_0.transformer_blocks_0.attn2.to_out_0.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.attentions_0.transformer_blocks_0.attn2.to_out_0.kernel", "write_shape": [1280, 1280], "chunk_shape": [1280, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.attentions_0.transformer_blocks_0.attn2.to_q.kernel", "write_shape": [1280, 1280], "chunk_shape": [1280, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.attentions_0.transformer_blocks_0.attn2.to_v.kernel", "write_shape": [768, 1280], "chunk_shape": [768, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.attentions_0.transformer_blocks_0.ff.net_0.proj.bias", "write_shape": [10240], "chunk_shape": [10240], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.attentions_0.transformer_blocks_0.ff.net_0.proj.kernel", "write_shape": [1280, 10240], "chunk_shape": [1280, 10240], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.attentions_0.transformer_blocks_0.ff.net_2.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.attentions_0.transformer_blocks_0.ff.net_2.kernel", "write_shape": [5120, 1280], "chunk_shape": [5120, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.attentions_0.transformer_blocks_0.norm1.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.attentions_0.transformer_blocks_0.norm1.scale", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.attentions_0.transformer_blocks_0.norm2.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.attentions_0.transformer_blocks_0.norm2.scale", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.attentions_0.transformer_blocks_0.norm3.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.attentions_0.transformer_blocks_0.norm3.scale", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.attentions_1.norm.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.attentions_1.norm.scale", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.attentions_1.proj_in.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.attentions_1.proj_in.kernel", "write_shape": [1, 1, 1280, 1280], "chunk_shape": [1, 1, 1280, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.attentions_1.proj_out.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.attentions_1.proj_out.kernel", "write_shape": [1, 1, 1280, 1280], "chunk_shape": [1, 1, 1280, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.attentions_1.transformer_blocks_0.attn1.to_k.kernel", "write_shape": [1280, 1280], "chunk_shape": [1280, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.attentions_1.transformer_blocks_0.attn1.to_out_0.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.attentions_1.transformer_blocks_0.attn1.to_out_0.kernel", "write_shape": [1280, 1280], "chunk_shape": [1280, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.attentions_1.transformer_blocks_0.attn1.to_q.kernel", "write_shape": [1280, 1280], "chunk_shape": [1280, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.attentions_1.transformer_blocks_0.attn1.to_v.kernel", "write_shape": [1280, 1280], "chunk_shape": [1280, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.attentions_1.transformer_blocks_0.attn2.to_k.kernel", "write_shape": [768, 1280], "chunk_shape": [768, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.attentions_1.transformer_blocks_0.attn2.to_out_0.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.attentions_1.transformer_blocks_0.attn2.to_out_0.kernel", "write_shape": [1280, 1280], "chunk_shape": [1280, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.attentions_1.transformer_blocks_0.attn2.to_q.kernel", "write_shape": [1280, 1280], "chunk_shape": [1280, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.attentions_1.transformer_blocks_0.attn2.to_v.kernel", "write_shape": [768, 1280], "chunk_shape": [768, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.attentions_1.transformer_blocks_0.ff.net_0.proj.bias", "write_shape": [10240], "chunk_shape": [10240], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.attentions_1.transformer_blocks_0.ff.net_0.proj.kernel", "write_shape": [1280, 10240], "chunk_shape": [1280, 10240], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.attentions_1.transformer_blocks_0.ff.net_2.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.attentions_1.transformer_blocks_0.ff.net_2.kernel", "write_shape": [5120, 1280], "chunk_shape": [5120, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.attentions_1.transformer_blocks_0.norm1.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.attentions_1.transformer_blocks_0.norm1.scale", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.attentions_1.transformer_blocks_0.norm2.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.attentions_1.transformer_blocks_0.norm2.scale", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.attentions_1.transformer_blocks_0.norm3.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.attentions_1.transformer_blocks_0.norm3.scale", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.attentions_2.norm.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.attentions_2.norm.scale", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.attentions_2.proj_in.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.attentions_2.proj_in.kernel", "write_shape": [1, 1, 1280, 1280], "chunk_shape": [1, 1, 1280, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.attentions_2.proj_out.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.attentions_2.proj_out.kernel", "write_shape": [1, 1, 1280, 1280], "chunk_shape": [1, 1, 1280, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.attentions_2.transformer_blocks_0.attn1.to_k.kernel", "write_shape": [1280, 1280], "chunk_shape": [1280, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.attentions_2.transformer_blocks_0.attn1.to_out_0.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.attentions_2.transformer_blocks_0.attn1.to_out_0.kernel", "write_shape": [1280, 1280], "chunk_shape": [1280, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.attentions_2.transformer_blocks_0.attn1.to_q.kernel", "write_shape": [1280, 1280], "chunk_shape": [1280, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.attentions_2.transformer_blocks_0.attn1.to_v.kernel", "write_shape": [1280, 1280], "chunk_shape": [1280, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.attentions_2.transformer_blocks_0.attn2.to_k.kernel", "write_shape": [768, 1280], "chunk_shape": [768, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.attentions_2.transformer_blocks_0.attn2.to_out_0.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.attentions_2.transformer_blocks_0.attn2.to_out_0.kernel", "write_shape": [1280, 1280], "chunk_shape": [1280, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.attentions_2.transformer_blocks_0.attn2.to_q.kernel", "write_shape": [1280, 1280], "chunk_shape": [1280, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.attentions_2.transformer_blocks_0.attn2.to_v.kernel", "write_shape": [768, 1280], "chunk_shape": [768, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.attentions_2.transformer_blocks_0.ff.net_0.proj.bias", "write_shape": [10240], "chunk_shape": [10240], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.attentions_2.transformer_blocks_0.ff.net_0.proj.kernel", "write_shape": [1280, 10240], "chunk_shape": [1280, 10240], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.attentions_2.transformer_blocks_0.ff.net_2.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.attentions_2.transformer_blocks_0.ff.net_2.kernel", "write_shape": [5120, 1280], "chunk_shape": [5120, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.attentions_2.transformer_blocks_0.norm1.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.attentions_2.transformer_blocks_0.norm1.scale", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.attentions_2.transformer_blocks_0.norm2.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.attentions_2.transformer_blocks_0.norm2.scale", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.attentions_2.transformer_blocks_0.norm3.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.attentions_2.transformer_blocks_0.norm3.scale", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.resnets_0.conv1.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.resnets_0.conv1.kernel", "write_shape": [3, 3, 2560, 1280], "chunk_shape": [3, 3, 2560, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.resnets_0.conv2.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.resnets_0.conv2.kernel", "write_shape": [3, 3, 1280, 1280], "chunk_shape": [3, 3, 1280, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.resnets_0.conv_shortcut.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.resnets_0.conv_shortcut.kernel", "write_shape": [1, 1, 2560, 1280], "chunk_shape": [1, 1, 2560, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.resnets_0.norm1.bias", "write_shape": [2560], "chunk_shape": [2560], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.resnets_0.norm1.scale", "write_shape": [2560], "chunk_shape": [2560], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.resnets_0.norm2.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.resnets_0.norm2.scale", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.resnets_0.time_emb_proj.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.resnets_0.time_emb_proj.kernel", "write_shape": [1280, 1280], "chunk_shape": [1280, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.resnets_1.conv1.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.resnets_1.conv1.kernel", "write_shape": [3, 3, 2560, 1280], "chunk_shape": [3, 3, 2560, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.resnets_1.conv2.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.resnets_1.conv2.kernel", "write_shape": [3, 3, 1280, 1280], "chunk_shape": [3, 3, 1280, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.resnets_1.conv_shortcut.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.resnets_1.conv_shortcut.kernel", "write_shape": [1, 1, 2560, 1280], "chunk_shape": [1, 1, 2560, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.resnets_1.norm1.bias", "write_shape": [2560], "chunk_shape": [2560], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.resnets_1.norm1.scale", "write_shape": [2560], "chunk_shape": [2560], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.resnets_1.norm2.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.resnets_1.norm2.scale", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.resnets_1.time_emb_proj.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.resnets_1.time_emb_proj.kernel", "write_shape": [1280, 1280], "chunk_shape": [1280, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.resnets_2.conv1.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.resnets_2.conv1.kernel", "write_shape": [3, 3, 1920, 1280], "chunk_shape": [3, 3, 1920, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.resnets_2.conv2.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.resnets_2.conv2.kernel", "write_shape": [3, 3, 1280, 1280], "chunk_shape": [3, 3, 1280, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.resnets_2.conv_shortcut.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.resnets_2.conv_shortcut.kernel", "write_shape": [1, 1, 1920, 1280], "chunk_shape": [1, 1, 1920, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.resnets_2.norm1.bias", "write_shape": [1920], "chunk_shape": [1920], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.resnets_2.norm1.scale", "write_shape": [1920], "chunk_shape": [1920], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.resnets_2.norm2.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.resnets_2.norm2.scale", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.resnets_2.time_emb_proj.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.resnets_2.time_emb_proj.kernel", "write_shape": [1280, 1280], "chunk_shape": [1280, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.upsamplers_0.conv.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_1.upsamplers_0.conv.kernel", "write_shape": [3, 3, 1280, 1280], "chunk_shape": [3, 3, 1280, 1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.attentions_0.norm.bias", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.attentions_0.norm.scale", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.attentions_0.proj_in.bias", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.attentions_0.proj_in.kernel", "write_shape": [1, 1, 640, 640], "chunk_shape": [1, 1, 640, 640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.attentions_0.proj_out.bias", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.attentions_0.proj_out.kernel", "write_shape": [1, 1, 640, 640], "chunk_shape": [1, 1, 640, 640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.attentions_0.transformer_blocks_0.attn1.to_k.kernel", "write_shape": [640, 640], "chunk_shape": [640, 640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.attentions_0.transformer_blocks_0.attn1.to_out_0.bias", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.attentions_0.transformer_blocks_0.attn1.to_out_0.kernel", "write_shape": [640, 640], "chunk_shape": [640, 640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.attentions_0.transformer_blocks_0.attn1.to_q.kernel", "write_shape": [640, 640], "chunk_shape": [640, 640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.attentions_0.transformer_blocks_0.attn1.to_v.kernel", "write_shape": [640, 640], "chunk_shape": [640, 640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.attentions_0.transformer_blocks_0.attn2.to_k.kernel", "write_shape": [768, 640], "chunk_shape": [768, 640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.attentions_0.transformer_blocks_0.attn2.to_out_0.bias", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.attentions_0.transformer_blocks_0.attn2.to_out_0.kernel", "write_shape": [640, 640], "chunk_shape": [640, 640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.attentions_0.transformer_blocks_0.attn2.to_q.kernel", "write_shape": [640, 640], "chunk_shape": [640, 640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.attentions_0.transformer_blocks_0.attn2.to_v.kernel", "write_shape": [768, 640], "chunk_shape": [768, 640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.attentions_0.transformer_blocks_0.ff.net_0.proj.bias", "write_shape": [5120], "chunk_shape": [5120], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.attentions_0.transformer_blocks_0.ff.net_0.proj.kernel", "write_shape": [640, 5120], "chunk_shape": [640, 5120], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.attentions_0.transformer_blocks_0.ff.net_2.bias", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.attentions_0.transformer_blocks_0.ff.net_2.kernel", "write_shape": [2560, 640], "chunk_shape": [2560, 640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.attentions_0.transformer_blocks_0.norm1.bias", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.attentions_0.transformer_blocks_0.norm1.scale", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.attentions_0.transformer_blocks_0.norm2.bias", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.attentions_0.transformer_blocks_0.norm2.scale", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.attentions_0.transformer_blocks_0.norm3.bias", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.attentions_0.transformer_blocks_0.norm3.scale", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.attentions_1.norm.bias", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.attentions_1.norm.scale", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.attentions_1.proj_in.bias", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.attentions_1.proj_in.kernel", "write_shape": [1, 1, 640, 640], "chunk_shape": [1, 1, 640, 640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.attentions_1.proj_out.bias", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.attentions_1.proj_out.kernel", "write_shape": [1, 1, 640, 640], "chunk_shape": [1, 1, 640, 640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.attentions_1.transformer_blocks_0.attn1.to_k.kernel", "write_shape": [640, 640], "chunk_shape": [640, 640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.attentions_1.transformer_blocks_0.attn1.to_out_0.bias", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.attentions_1.transformer_blocks_0.attn1.to_out_0.kernel", "write_shape": [640, 640], "chunk_shape": [640, 640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.attentions_1.transformer_blocks_0.attn1.to_q.kernel", "write_shape": [640, 640], "chunk_shape": [640, 640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.attentions_1.transformer_blocks_0.attn1.to_v.kernel", "write_shape": [640, 640], "chunk_shape": [640, 640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.attentions_1.transformer_blocks_0.attn2.to_k.kernel", "write_shape": [768, 640], "chunk_shape": [768, 640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.attentions_1.transformer_blocks_0.attn2.to_out_0.bias", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.attentions_1.transformer_blocks_0.attn2.to_out_0.kernel", "write_shape": [640, 640], "chunk_shape": [640, 640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.attentions_1.transformer_blocks_0.attn2.to_q.kernel", "write_shape": [640, 640], "chunk_shape": [640, 640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.attentions_1.transformer_blocks_0.attn2.to_v.kernel", "write_shape": [768, 640], "chunk_shape": [768, 640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.attentions_1.transformer_blocks_0.ff.net_0.proj.bias", "write_shape": [5120], "chunk_shape": [5120], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.attentions_1.transformer_blocks_0.ff.net_0.proj.kernel", "write_shape": [640, 5120], "chunk_shape": [640, 5120], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.attentions_1.transformer_blocks_0.ff.net_2.bias", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.attentions_1.transformer_blocks_0.ff.net_2.kernel", "write_shape": [2560, 640], "chunk_shape": [2560, 640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.attentions_1.transformer_blocks_0.norm1.bias", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.attentions_1.transformer_blocks_0.norm1.scale", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.attentions_1.transformer_blocks_0.norm2.bias", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.attentions_1.transformer_blocks_0.norm2.scale", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.attentions_1.transformer_blocks_0.norm3.bias", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.attentions_1.transformer_blocks_0.norm3.scale", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.attentions_2.norm.bias", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.attentions_2.norm.scale", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.attentions_2.proj_in.bias", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.attentions_2.proj_in.kernel", "write_shape": [1, 1, 640, 640], "chunk_shape": [1, 1, 640, 640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.attentions_2.proj_out.bias", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.attentions_2.proj_out.kernel", "write_shape": [1, 1, 640, 640], "chunk_shape": [1, 1, 640, 640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.attentions_2.transformer_blocks_0.attn1.to_k.kernel", "write_shape": [640, 640], "chunk_shape": [640, 640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.attentions_2.transformer_blocks_0.attn1.to_out_0.bias", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.attentions_2.transformer_blocks_0.attn1.to_out_0.kernel", "write_shape": [640, 640], "chunk_shape": [640, 640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.attentions_2.transformer_blocks_0.attn1.to_q.kernel", "write_shape": [640, 640], "chunk_shape": [640, 640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.attentions_2.transformer_blocks_0.attn1.to_v.kernel", "write_shape": [640, 640], "chunk_shape": [640, 640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.attentions_2.transformer_blocks_0.attn2.to_k.kernel", "write_shape": [768, 640], "chunk_shape": [768, 640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.attentions_2.transformer_blocks_0.attn2.to_out_0.bias", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.attentions_2.transformer_blocks_0.attn2.to_out_0.kernel", "write_shape": [640, 640], "chunk_shape": [640, 640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.attentions_2.transformer_blocks_0.attn2.to_q.kernel", "write_shape": [640, 640], "chunk_shape": [640, 640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.attentions_2.transformer_blocks_0.attn2.to_v.kernel", "write_shape": [768, 640], "chunk_shape": [768, 640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.attentions_2.transformer_blocks_0.ff.net_0.proj.bias", "write_shape": [5120], "chunk_shape": [5120], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.attentions_2.transformer_blocks_0.ff.net_0.proj.kernel", "write_shape": [640, 5120], "chunk_shape": [640, 5120], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.attentions_2.transformer_blocks_0.ff.net_2.bias", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.attentions_2.transformer_blocks_0.ff.net_2.kernel", "write_shape": [2560, 640], "chunk_shape": [2560, 640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.attentions_2.transformer_blocks_0.norm1.bias", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.attentions_2.transformer_blocks_0.norm1.scale", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.attentions_2.transformer_blocks_0.norm2.bias", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.attentions_2.transformer_blocks_0.norm2.scale", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.attentions_2.transformer_blocks_0.norm3.bias", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.attentions_2.transformer_blocks_0.norm3.scale", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.resnets_0.conv1.bias", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.resnets_0.conv1.kernel", "write_shape": [3, 3, 1920, 640], "chunk_shape": [3, 3, 1920, 640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.resnets_0.conv2.bias", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.resnets_0.conv2.kernel", "write_shape": [3, 3, 640, 640], "chunk_shape": [3, 3, 640, 640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.resnets_0.conv_shortcut.bias", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.resnets_0.conv_shortcut.kernel", "write_shape": [1, 1, 1920, 640], "chunk_shape": [1, 1, 1920, 640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.resnets_0.norm1.bias", "write_shape": [1920], "chunk_shape": [1920], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.resnets_0.norm1.scale", "write_shape": [1920], "chunk_shape": [1920], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.resnets_0.norm2.bias", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.resnets_0.norm2.scale", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.resnets_0.time_emb_proj.bias", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.resnets_0.time_emb_proj.kernel", "write_shape": [1280, 640], "chunk_shape": [1280, 640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.resnets_1.conv1.bias", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.resnets_1.conv1.kernel", "write_shape": [3, 3, 1280, 640], "chunk_shape": [3, 3, 1280, 640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.resnets_1.conv2.bias", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.resnets_1.conv2.kernel", "write_shape": [3, 3, 640, 640], "chunk_shape": [3, 3, 640, 640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.resnets_1.conv_shortcut.bias", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.resnets_1.conv_shortcut.kernel", "write_shape": [1, 1, 1280, 640], "chunk_shape": [1, 1, 1280, 640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.resnets_1.norm1.bias", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.resnets_1.norm1.scale", "write_shape": [1280], "chunk_shape": [1280], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.resnets_1.norm2.bias", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.resnets_1.norm2.scale", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.resnets_1.time_emb_proj.bias", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.resnets_1.time_emb_proj.kernel", "write_shape": [1280, 640], "chunk_shape": [1280, 640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.resnets_2.conv1.bias", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.resnets_2.conv1.kernel", "write_shape": [3, 3, 960, 640], "chunk_shape": [3, 3, 960, 640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.resnets_2.conv2.bias", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.resnets_2.conv2.kernel", "write_shape": [3, 3, 640, 640], "chunk_shape": [3, 3, 640, 640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.resnets_2.conv_shortcut.bias", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.resnets_2.conv_shortcut.kernel", "write_shape": [1, 1, 960, 640], "chunk_shape": [1, 1, 960, 640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.resnets_2.norm1.bias", "write_shape": [960], "chunk_shape": [960], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.resnets_2.norm1.scale", "write_shape": [960], "chunk_shape": [960], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.resnets_2.norm2.bias", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.resnets_2.norm2.scale", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.resnets_2.time_emb_proj.bias", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.resnets_2.time_emb_proj.kernel", "write_shape": [1280, 640], "chunk_shape": [1280, 640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.upsamplers_0.conv.bias", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_2.upsamplers_0.conv.kernel", "write_shape": [3, 3, 640, 640], "chunk_shape": [3, 3, 640, 640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.attentions_0.norm.bias", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.attentions_0.norm.scale", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.attentions_0.proj_in.bias", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.attentions_0.proj_in.kernel", "write_shape": [1, 1, 320, 320], "chunk_shape": [1, 1, 320, 320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.attentions_0.proj_out.bias", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.attentions_0.proj_out.kernel", "write_shape": [1, 1, 320, 320], "chunk_shape": [1, 1, 320, 320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.attentions_0.transformer_blocks_0.attn1.to_k.kernel", "write_shape": [320, 320], "chunk_shape": [320, 320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.attentions_0.transformer_blocks_0.attn1.to_out_0.bias", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.attentions_0.transformer_blocks_0.attn1.to_out_0.kernel", "write_shape": [320, 320], "chunk_shape": [320, 320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.attentions_0.transformer_blocks_0.attn1.to_q.kernel", "write_shape": [320, 320], "chunk_shape": [320, 320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.attentions_0.transformer_blocks_0.attn1.to_v.kernel", "write_shape": [320, 320], "chunk_shape": [320, 320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.attentions_0.transformer_blocks_0.attn2.to_k.kernel", "write_shape": [768, 320], "chunk_shape": [768, 320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.attentions_0.transformer_blocks_0.attn2.to_out_0.bias", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.attentions_0.transformer_blocks_0.attn2.to_out_0.kernel", "write_shape": [320, 320], "chunk_shape": [320, 320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.attentions_0.transformer_blocks_0.attn2.to_q.kernel", "write_shape": [320, 320], "chunk_shape": [320, 320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.attentions_0.transformer_blocks_0.attn2.to_v.kernel", "write_shape": [768, 320], "chunk_shape": [768, 320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.attentions_0.transformer_blocks_0.ff.net_0.proj.bias", "write_shape": [2560], "chunk_shape": [2560], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.attentions_0.transformer_blocks_0.ff.net_0.proj.kernel", "write_shape": [320, 2560], "chunk_shape": [320, 2560], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.attentions_0.transformer_blocks_0.ff.net_2.bias", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.attentions_0.transformer_blocks_0.ff.net_2.kernel", "write_shape": [1280, 320], "chunk_shape": [1280, 320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.attentions_0.transformer_blocks_0.norm1.bias", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.attentions_0.transformer_blocks_0.norm1.scale", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.attentions_0.transformer_blocks_0.norm2.bias", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.attentions_0.transformer_blocks_0.norm2.scale", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.attentions_0.transformer_blocks_0.norm3.bias", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.attentions_0.transformer_blocks_0.norm3.scale", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.attentions_1.norm.bias", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.attentions_1.norm.scale", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.attentions_1.proj_in.bias", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.attentions_1.proj_in.kernel", "write_shape": [1, 1, 320, 320], "chunk_shape": [1, 1, 320, 320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.attentions_1.proj_out.bias", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.attentions_1.proj_out.kernel", "write_shape": [1, 1, 320, 320], "chunk_shape": [1, 1, 320, 320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.attentions_1.transformer_blocks_0.attn1.to_k.kernel", "write_shape": [320, 320], "chunk_shape": [320, 320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.attentions_1.transformer_blocks_0.attn1.to_out_0.bias", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.attentions_1.transformer_blocks_0.attn1.to_out_0.kernel", "write_shape": [320, 320], "chunk_shape": [320, 320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.attentions_1.transformer_blocks_0.attn1.to_q.kernel", "write_shape": [320, 320], "chunk_shape": [320, 320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.attentions_1.transformer_blocks_0.attn1.to_v.kernel", "write_shape": [320, 320], "chunk_shape": [320, 320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.attentions_1.transformer_blocks_0.attn2.to_k.kernel", "write_shape": [768, 320], "chunk_shape": [768, 320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.attentions_1.transformer_blocks_0.attn2.to_out_0.bias", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.attentions_1.transformer_blocks_0.attn2.to_out_0.kernel", "write_shape": [320, 320], "chunk_shape": [320, 320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.attentions_1.transformer_blocks_0.attn2.to_q.kernel", "write_shape": [320, 320], "chunk_shape": [320, 320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.attentions_1.transformer_blocks_0.attn2.to_v.kernel", "write_shape": [768, 320], "chunk_shape": [768, 320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.attentions_1.transformer_blocks_0.ff.net_0.proj.bias", "write_shape": [2560], "chunk_shape": [2560], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.attentions_1.transformer_blocks_0.ff.net_0.proj.kernel", "write_shape": [320, 2560], "chunk_shape": [320, 2560], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.attentions_1.transformer_blocks_0.ff.net_2.bias", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.attentions_1.transformer_blocks_0.ff.net_2.kernel", "write_shape": [1280, 320], "chunk_shape": [1280, 320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.attentions_1.transformer_blocks_0.norm1.bias", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.attentions_1.transformer_blocks_0.norm1.scale", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.attentions_1.transformer_blocks_0.norm2.bias", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.attentions_1.transformer_blocks_0.norm2.scale", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.attentions_1.transformer_blocks_0.norm3.bias", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.attentions_1.transformer_blocks_0.norm3.scale", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.attentions_2.norm.bias", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.attentions_2.norm.scale", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.attentions_2.proj_in.bias", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.attentions_2.proj_in.kernel", "write_shape": [1, 1, 320, 320], "chunk_shape": [1, 1, 320, 320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.attentions_2.proj_out.bias", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.attentions_2.proj_out.kernel", "write_shape": [1, 1, 320, 320], "chunk_shape": [1, 1, 320, 320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.attentions_2.transformer_blocks_0.attn1.to_k.kernel", "write_shape": [320, 320], "chunk_shape": [320, 320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.attentions_2.transformer_blocks_0.attn1.to_out_0.bias", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.attentions_2.transformer_blocks_0.attn1.to_out_0.kernel", "write_shape": [320, 320], "chunk_shape": [320, 320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.attentions_2.transformer_blocks_0.attn1.to_q.kernel", "write_shape": [320, 320], "chunk_shape": [320, 320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.attentions_2.transformer_blocks_0.attn1.to_v.kernel", "write_shape": [320, 320], "chunk_shape": [320, 320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.attentions_2.transformer_blocks_0.attn2.to_k.kernel", "write_shape": [768, 320], "chunk_shape": [768, 320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.attentions_2.transformer_blocks_0.attn2.to_out_0.bias", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.attentions_2.transformer_blocks_0.attn2.to_out_0.kernel", "write_shape": [320, 320], "chunk_shape": [320, 320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.attentions_2.transformer_blocks_0.attn2.to_q.kernel", "write_shape": [320, 320], "chunk_shape": [320, 320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.attentions_2.transformer_blocks_0.attn2.to_v.kernel", "write_shape": [768, 320], "chunk_shape": [768, 320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.attentions_2.transformer_blocks_0.ff.net_0.proj.bias", "write_shape": [2560], "chunk_shape": [2560], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.attentions_2.transformer_blocks_0.ff.net_0.proj.kernel", "write_shape": [320, 2560], "chunk_shape": [320, 2560], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.attentions_2.transformer_blocks_0.ff.net_2.bias", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.attentions_2.transformer_blocks_0.ff.net_2.kernel", "write_shape": [1280, 320], "chunk_shape": [1280, 320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.attentions_2.transformer_blocks_0.norm1.bias", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.attentions_2.transformer_blocks_0.norm1.scale", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.attentions_2.transformer_blocks_0.norm2.bias", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.attentions_2.transformer_blocks_0.norm2.scale", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.attentions_2.transformer_blocks_0.norm3.bias", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.attentions_2.transformer_blocks_0.norm3.scale", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.resnets_0.conv1.bias", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.resnets_0.conv1.kernel", "write_shape": [3, 3, 960, 320], "chunk_shape": [3, 3, 960, 320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.resnets_0.conv2.bias", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.resnets_0.conv2.kernel", "write_shape": [3, 3, 320, 320], "chunk_shape": [3, 3, 320, 320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.resnets_0.conv_shortcut.bias", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.resnets_0.conv_shortcut.kernel", "write_shape": [1, 1, 960, 320], "chunk_shape": [1, 1, 960, 320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.resnets_0.norm1.bias", "write_shape": [960], "chunk_shape": [960], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.resnets_0.norm1.scale", "write_shape": [960], "chunk_shape": [960], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.resnets_0.norm2.bias", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.resnets_0.norm2.scale", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.resnets_0.time_emb_proj.bias", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.resnets_0.time_emb_proj.kernel", "write_shape": [1280, 320], "chunk_shape": [1280, 320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.resnets_1.conv1.bias", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.resnets_1.conv1.kernel", "write_shape": [3, 3, 640, 320], "chunk_shape": [3, 3, 640, 320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.resnets_1.conv2.bias", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.resnets_1.conv2.kernel", "write_shape": [3, 3, 320, 320], "chunk_shape": [3, 3, 320, 320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.resnets_1.conv_shortcut.bias", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.resnets_1.conv_shortcut.kernel", "write_shape": [1, 1, 640, 320], "chunk_shape": [1, 1, 640, 320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.resnets_1.norm1.bias", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.resnets_1.norm1.scale", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.resnets_1.norm2.bias", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.resnets_1.norm2.scale", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.resnets_1.time_emb_proj.bias", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.resnets_1.time_emb_proj.kernel", "write_shape": [1280, 320], "chunk_shape": [1280, 320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.resnets_2.conv1.bias", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.resnets_2.conv1.kernel", "write_shape": [3, 3, 640, 320], "chunk_shape": [3, 3, 640, 320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.resnets_2.conv2.bias", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.resnets_2.conv2.kernel", "write_shape": [3, 3, 320, 320], "chunk_shape": [3, 3, 320, 320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.resnets_2.conv_shortcut.bias", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.resnets_2.conv_shortcut.kernel", "write_shape": [1, 1, 640, 320], "chunk_shape": [1, 1, 640, 320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.resnets_2.norm1.bias", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.resnets_2.norm1.scale", "write_shape": [640], "chunk_shape": [640], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.resnets_2.norm2.bias", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.resnets_2.norm2.scale", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.resnets_2.time_emb_proj.bias", "write_shape": [320], "chunk_shape": [320], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.up_blocks_3.resnets_2.time_emb_proj.kernel", "write_shape": [1280, 320], "chunk_shape": [1280, 320], "ext_metadata": null}}]}