checkpoint_100000.orbax/_METADATA

{"tree_metadata": {"('controlnet_cond_embedding', 'blocks_0', 'bias')": {"key_metadata": [{"key": "controlnet_cond_embedding", "key_type": 2}, {"key": "blocks_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('controlnet_cond_embedding', 'blocks_0', 'kernel')": {"key_metadata": [{"key": "controlnet_cond_embedding", "key_type": 2}, {"key": "blocks_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('controlnet_cond_embedding', 'blocks_1', 'bias')": {"key_metadata": [{"key": "controlnet_cond_embedding", "key_type": 2}, {"key": "blocks_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('controlnet_cond_embedding', 'blocks_1', 'kernel')": {"key_metadata": [{"key": "controlnet_cond_embedding", "key_type": 2}, {"key": "blocks_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('controlnet_cond_embedding', 'blocks_2', 'bias')": {"key_metadata": [{"key": "controlnet_cond_embedding", "key_type": 2}, {"key": "blocks_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('controlnet_cond_embedding', 'blocks_2', 'kernel')": {"key_metadata": [{"key": "controlnet_cond_embedding", "key_type": 2}, {"key": "blocks_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('controlnet_cond_embedding', 'blocks_3', 'bias')": {"key_metadata": [{"key": "controlnet_cond_embedding", "key_type": 2}, {"key": "blocks_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('controlnet_cond_embedding', 'blocks_3', 'kernel')": {"key_metadata": [{"key": "controlnet_cond_embedding", "key_type": 2}, {"key": "blocks_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('controlnet_cond_embedding', 'blocks_4', 'bias')": {"key_metadata": [{"key": "controlnet_cond_embedding", "key_type": 2}, {"key": "blocks_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('controlnet_cond_embedding', 'blocks_4', 'kernel')": {"key_metadata": [{"key": "controlnet_cond_embedding", "key_type": 2}, {"key": "blocks_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('controlnet_cond_embedding', 'blocks_5', 'bias')": {"key_metadata": [{"key": "controlnet_cond_embedding", "key_type": 2}, {"key": "blocks_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('controlnet_cond_embedding', 'blocks_5', 'kernel')": {"key_metadata": [{"key": "controlnet_cond_embedding", "key_type": 2}, {"key": "blocks_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('controlnet_cond_embedding', 'conv_in', 'bias')": {"key_metadata": [{"key": "controlnet_cond_embedding", "key_type": 2}, {"key": "conv_in", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('controlnet_cond_embedding', 'conv_in', 'kernel')": {"key_metadata": [{"key": "controlnet_cond_embedding", "key_type": 2}, {"key": "conv_in", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('controlnet_cond_embedding', 'conv_out', 'bias')": {"key_metadata": [{"key": "controlnet_cond_embedding", "key_type": 2}, {"key": "conv_out", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('controlnet_cond_embedding', 'conv_out', 'kernel')": {"key_metadata": [{"key": "controlnet_cond_embedding", "key_type": 2}, {"key": "conv_out", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('controlnet_down_blocks_0', 'bias')": {"key_metadata": [{"key": "controlnet_down_blocks_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('controlnet_down_blocks_0', 'kernel')": {"key_metadata": [{"key": "controlnet_down_blocks_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('controlnet_down_blocks_1', 'bias')": {"key_metadata": [{"key": "controlnet_down_blocks_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('controlnet_down_blocks_1', 'kernel')": {"key_metadata": [{"key": "controlnet_down_blocks_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('controlnet_down_blocks_10', 'bias')": {"key_metadata": [{"key": "controlnet_down_blocks_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('controlnet_down_blocks_10', 'kernel')": {"key_metadata": [{"key": "controlnet_down_blocks_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('controlnet_down_blocks_11', 'bias')": {"key_metadata": [{"key": "controlnet_down_blocks_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('controlnet_down_blocks_11', 'kernel')": {"key_metadata": [{"key": "controlnet_down_blocks_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('controlnet_down_blocks_2', 'bias')": {"key_metadata": [{"key": "controlnet_down_blocks_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('controlnet_down_blocks_2', 'kernel')": {"key_metadata": [{"key": "controlnet_down_blocks_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('controlnet_down_blocks_3', 'bias')": {"key_metadata": [{"key": "controlnet_down_blocks_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('controlnet_down_blocks_3', 'kernel')": {"key_metadata": [{"key": "controlnet_down_blocks_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('controlnet_down_blocks_4', 'bias')": {"key_metadata": [{"key": "controlnet_down_blocks_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('controlnet_down_blocks_4', 'kernel')": {"key_metadata": [{"key": "controlnet_down_blocks_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('controlnet_down_blocks_5', 'bias')": {"key_metadata": [{"key": "controlnet_down_blocks_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('controlnet_down_blocks_5', 'kernel')": {"key_metadata": [{"key": "controlnet_down_blocks_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('controlnet_down_blocks_6', 'bias')": {"key_metadata": [{"key": "controlnet_down_blocks_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('controlnet_down_blocks_6', 'kernel')": {"key_metadata": [{"key": "controlnet_down_blocks_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('controlnet_down_blocks_7', 'bias')": {"key_metadata": [{"key": "controlnet_down_blocks_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('controlnet_down_blocks_7', 'kernel')": {"key_metadata": [{"key": "controlnet_down_blocks_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('controlnet_down_blocks_8', 'bias')": {"key_metadata": [{"key": "controlnet_down_blocks_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('controlnet_down_blocks_8', 'kernel')": {"key_metadata": [{"key": "controlnet_down_blocks_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('controlnet_down_blocks_9', 'bias')": {"key_metadata": [{"key": "controlnet_down_blocks_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('controlnet_down_blocks_9', 'kernel')": {"key_metadata": [{"key": "controlnet_down_blocks_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('controlnet_mid_block', 'bias')": {"key_metadata": [{"key": "controlnet_mid_block", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('controlnet_mid_block', 'kernel')": {"key_metadata": [{"key": "controlnet_mid_block", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('conv_in', 'bias')": {"key_metadata": [{"key": "conv_in", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('conv_in', 'kernel')": {"key_metadata": [{"key": "conv_in", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_0', 'attentions_0', 'norm', 'bias')": {"key_metadata": [{"key": "down_blocks_0", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "norm", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_0', 'attentions_0', 'norm', 'scale')": {"key_metadata": [{"key": "down_blocks_0", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "norm", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_0', 'attentions_0', 'proj_in', 'bias')": {"key_metadata": [{"key": "down_blocks_0", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "proj_in", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_0', 'attentions_0', 'proj_in', 'kernel')": {"key_metadata": [{"key": "down_blocks_0", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "proj_in", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_0', 'attentions_0', 'proj_out', 'bias')": {"key_metadata": [{"key": "down_blocks_0", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "proj_out", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_0', 'attentions_0', 'proj_out', 'kernel')": {"key_metadata": [{"key": "down_blocks_0", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "proj_out", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_0', 'attentions_0', 'transformer_blocks_0', 'attn1', 'to_k', 'kernel')": {"key_metadata": [{"key": "down_blocks_0", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "attn1", "key_type": 2}, {"key": "to_k", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_0', 'attentions_0', 'transformer_blocks_0', 'attn1', 'to_out_0', 'bias')": {"key_metadata": [{"key": "down_blocks_0", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "attn1", "key_type": 2}, {"key": "to_out_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_0', 'attentions_0', 'transformer_blocks_0', 'attn1', 'to_out_0', 'kernel')": {"key_metadata": [{"key": "down_blocks_0", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "attn1", "key_type": 2}, {"key": "to_out_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_0', 'attentions_0', 'transformer_blocks_0', 'attn1', 'to_q', 'kernel')": {"key_metadata": [{"key": "down_blocks_0", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "attn1", "key_type": 2}, {"key": "to_q", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_0', 'attentions_0', 'transformer_blocks_0', 'attn1', 'to_v', 'kernel')": {"key_metadata": [{"key": "down_blocks_0", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "attn1", "key_type": 2}, {"key": "to_v", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_0', 'attentions_0', 'transformer_blocks_0', 'attn2', 'to_k', 'kernel')": {"key_metadata": [{"key": "down_blocks_0", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "attn2", "key_type": 2}, {"key": "to_k", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_0', 'attentions_0', 'transformer_blocks_0', 'attn2', 'to_out_0', 'bias')": {"key_metadata": [{"key": "down_blocks_0", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "attn2", "key_type": 2}, {"key": "to_out_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_0', 'attentions_0', 'transformer_blocks_0', 'attn2', 'to_out_0', 'kernel')": {"key_metadata": [{"key": "down_blocks_0", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "attn2", "key_type": 2}, {"key": "to_out_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_0', 'attentions_0', 'transformer_blocks_0', 'attn2', 'to_q', 'kernel')": {"key_metadata": [{"key": "down_blocks_0", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "attn2", "key_type": 2}, {"key": "to_q", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_0', 'attentions_0', 'transformer_blocks_0', 'attn2', 'to_v', 'kernel')": {"key_metadata": [{"key": "down_blocks_0", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "attn2", "key_type": 2}, {"key": "to_v", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_0', 'attentions_0', 'transformer_blocks_0', 'ff', 'net_0', 'proj', 'bias')": {"key_metadata": [{"key": "down_blocks_0", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "ff", "key_type": 2}, {"key": "net_0", "key_type": 2}, {"key": "proj", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_0', 'attentions_0', 'transformer_blocks_0', 'ff', 'net_0', 'proj', 'kernel')": {"key_metadata": [{"key": "down_blocks_0", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "ff", "key_type": 2}, {"key": "net_0", "key_type": 2}, {"key": "proj", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_0', 'attentions_0', 'transformer_blocks_0', 'ff', 'net_2', 'bias')": {"key_metadata": [{"key": "down_blocks_0", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "ff", "key_type": 2}, {"key": "net_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_0', 'attentions_0', 'transformer_blocks_0', 'ff', 'net_2', 'kernel')": {"key_metadata": [{"key": "down_blocks_0", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "ff", "key_type": 2}, {"key": "net_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_0', 'attentions_0', 'transformer_blocks_0', 'norm1', 'bias')": {"key_metadata": [{"key": "down_blocks_0", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "norm1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_0', 'attentions_0', 'transformer_blocks_0', 'norm1', 'scale')": {"key_metadata": [{"key": "down_blocks_0", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "norm1", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_0', 'attentions_0', 'transformer_blocks_0', 'norm2', 'bias')": {"key_metadata": [{"key": "down_blocks_0", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "norm2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_0', 'attentions_0', 'transformer_blocks_0', 'norm2', 'scale')": {"key_metadata": [{"key": "down_blocks_0", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "norm2", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_0', 'attentions_0', 'transformer_blocks_0', 'norm3', 'bias')": {"key_metadata": [{"key": "down_blocks_0", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "norm3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_0', 'attentions_0', 'transformer_blocks_0', 'norm3', 'scale')": {"key_metadata": [{"key": "down_blocks_0", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "norm3", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_0', 'attentions_1', 'norm', 'bias')": {"key_metadata": [{"key": "down_blocks_0", "key_type": 2}, {"key": "attentions_1", "key_type": 2}, {"key": "norm", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_0', 'attentions_1', 'norm', 'scale')": {"key_metadata": [{"key": "down_blocks_0", "key_type": 2}, {"key": "attentions_1", "key_type": 2}, {"key": "norm", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_0', 'attentions_1', 'proj_in', 'bias')": {"key_metadata": [{"key": "down_blocks_0", "key_type": 2}, {"key": "attentions_1", "key_type": 2}, {"key": "proj_in", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_0', 'attentions_1', 'proj_in', 'kernel')": {"key_metadata": [{"key": "down_blocks_0", "key_type": 2}, {"key": "attentions_1", "key_type": 2}, {"key": "proj_in", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_0', 'attentions_1', 'proj_out', 'bias')": {"key_metadata": [{"key": "down_blocks_0", "key_type": 2}, {"key": "attentions_1", "key_type": 2}, {"key": "proj_out", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_0', 'attentions_1', 'proj_out', 'kernel')": {"key_metadata": [{"key": "down_blocks_0", "key_type": 2}, {"key": "attentions_1", "key_type": 2}, {"key": "proj_out", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_0', 'attentions_1', 'transformer_blocks_0', 'attn1', 'to_k', 'kernel')": {"key_metadata": [{"key": "down_blocks_0", "key_type": 2}, {"key": "attentions_1", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "attn1", "key_type": 2}, {"key": "to_k", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_0', 'attentions_1', 'transformer_blocks_0', 'attn1', 'to_out_0', 'bias')": {"key_metadata": [{"key": "down_blocks_0", "key_type": 2}, {"key": "attentions_1", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "attn1", "key_type": 2}, {"key": "to_out_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_0', 'attentions_1', 'transformer_blocks_0', 'attn1', 'to_out_0', 'kernel')": {"key_metadata": [{"key": "down_blocks_0", "key_type": 2}, {"key": "attentions_1", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "attn1", "key_type": 2}, {"key": "to_out_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_0', 'attentions_1', 'transformer_blocks_0', 'attn1', 'to_q', 'kernel')": {"key_metadata": [{"key": "down_blocks_0", "key_type": 2}, {"key": "attentions_1", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "attn1", "key_type": 2}, {"key": "to_q", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_0', 'attentions_1', 'transformer_blocks_0', 'attn1', 'to_v', 'kernel')": {"key_metadata": [{"key": "down_blocks_0", "key_type": 2}, {"key": "attentions_1", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "attn1", "key_type": 2}, {"key": "to_v", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_0', 'attentions_1', 'transformer_blocks_0', 'attn2', 'to_k', 'kernel')": {"key_metadata": [{"key": "down_blocks_0", "key_type": 2}, {"key": "attentions_1", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "attn2", "key_type": 2}, {"key": "to_k", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_0', 'attentions_1', 'transformer_blocks_0', 'attn2', 'to_out_0', 'bias')": {"key_metadata": [{"key": "down_blocks_0", "key_type": 2}, {"key": "attentions_1", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "attn2", "key_type": 2}, {"key": "to_out_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_0', 'attentions_1', 'transformer_blocks_0', 'attn2', 'to_out_0', 'kernel')": {"key_metadata": [{"key": "down_blocks_0", "key_type": 2}, {"key": "attentions_1", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "attn2", "key_type": 2}, {"key": "to_out_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_0', 'attentions_1', 'transformer_blocks_0', 'attn2', 'to_q', 'kernel')": {"key_metadata": [{"key": "down_blocks_0", "key_type": 2}, {"key": "attentions_1", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "attn2", "key_type": 2}, {"key": "to_q", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_0', 'attentions_1', 'transformer_blocks_0', 'attn2', 'to_v', 'kernel')": {"key_metadata": [{"key": "down_blocks_0", "key_type": 2}, {"key": "attentions_1", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "attn2", "key_type": 2}, {"key": "to_v", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_0', 'attentions_1', 'transformer_blocks_0', 'ff', 'net_0', 'proj', 'bias')": {"key_metadata": [{"key": "down_blocks_0", "key_type": 2}, {"key": "attentions_1", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "ff", "key_type": 2}, {"key": "net_0", "key_type": 2}, {"key": "proj", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_0', 'attentions_1', 'transformer_blocks_0', 'ff', 'net_0', 'proj', 'kernel')": {"key_metadata": [{"key": "down_blocks_0", "key_type": 2}, {"key": "attentions_1", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "ff", "key_type": 2}, {"key": "net_0", "key_type": 2}, {"key": "proj", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_0', 'attentions_1', 'transformer_blocks_0', 'ff', 'net_2', 'bias')": {"key_metadata": [{"key": "down_blocks_0", "key_type": 2}, {"key": "attentions_1", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "ff", "key_type": 2}, {"key": "net_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_0', 'attentions_1', 'transformer_blocks_0', 'ff', 'net_2', 'kernel')": {"key_metadata": [{"key": "down_blocks_0", "key_type": 2}, {"key": "attentions_1", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "ff", "key_type": 2}, {"key": "net_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_0', 'attentions_1', 'transformer_blocks_0', 'norm1', 'bias')": {"key_metadata": [{"key": "down_blocks_0", "key_type": 2}, {"key": "attentions_1", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "norm1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_0', 'attentions_1', 'transformer_blocks_0', 'norm1', 'scale')": {"key_metadata": [{"key": "down_blocks_0", "key_type": 2}, {"key": "attentions_1", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "norm1", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_0', 'attentions_1', 'transformer_blocks_0', 'norm2', 'bias')": {"key_metadata": [{"key": "down_blocks_0", "key_type": 2}, {"key": "attentions_1", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "norm2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_0', 'attentions_1', 'transformer_blocks_0', 'norm2', 'scale')": {"key_metadata": [{"key": "down_blocks_0", "key_type": 2}, {"key": "attentions_1", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "norm2", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_0', 'attentions_1', 'transformer_blocks_0', 'norm3', 'bias')": {"key_metadata": [{"key": "down_blocks_0", "key_type": 2}, {"key": "attentions_1", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "norm3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_0', 'attentions_1', 'transformer_blocks_0', 'norm3', 'scale')": {"key_metadata": [{"key": "down_blocks_0", "key_type": 2}, {"key": "attentions_1", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "norm3", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_0', 'downsamplers_0', 'conv', 'bias')": {"key_metadata": [{"key": "down_blocks_0", "key_type": 2}, {"key": "downsamplers_0", "key_type": 2}, {"key": "conv", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_0', 'downsamplers_0', 'conv', 'kernel')": {"key_metadata": [{"key": "down_blocks_0", "key_type": 2}, {"key": "downsamplers_0", "key_type": 2}, {"key": "conv", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_0', 'resnets_0', 'conv1', 'bias')": {"key_metadata": [{"key": "down_blocks_0", "key_type": 2}, {"key": "resnets_0", "key_type": 2}, {"key": "conv1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_0', 'resnets_0', 'conv1', 'kernel')": {"key_metadata": [{"key": "down_blocks_0", "key_type": 2}, {"key": "resnets_0", "key_type": 2}, {"key": "conv1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_0', 'resnets_0', 'conv2', 'bias')": {"key_metadata": [{"key": "down_blocks_0", "key_type": 2}, {"key": "resnets_0", "key_type": 2}, {"key": "conv2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_0', 'resnets_0', 'conv2', 'kernel')": {"key_metadata": [{"key": "down_blocks_0", "key_type": 2}, {"key": "resnets_0", "key_type": 2}, {"key": "conv2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_0', 'resnets_0', 'norm1', 'bias')": {"key_metadata": [{"key": "down_blocks_0", "key_type": 2}, {"key": "resnets_0", "key_type": 2}, {"key": "norm1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_0', 'resnets_0', 'norm1', 'scale')": {"key_metadata": [{"key": "down_blocks_0", "key_type": 2}, {"key": "resnets_0", "key_type": 2}, {"key": "norm1", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_0', 'resnets_0', 'norm2', 'bias')": {"key_metadata": [{"key": "down_blocks_0", "key_type": 2}, {"key": "resnets_0", "key_type": 2}, {"key": "norm2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_0', 'resnets_0', 'norm2', 'scale')": {"key_metadata": [{"key": "down_blocks_0", "key_type": 2}, {"key": "resnets_0", "key_type": 2}, {"key": "norm2", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_0', 'resnets_0', 'time_emb_proj', 'bias')": {"key_metadata": [{"key": "down_blocks_0", "key_type": 2}, {"key": "resnets_0", "key_type": 2}, {"key": "time_emb_proj", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_0', 'resnets_0', 'time_emb_proj', 'kernel')": {"key_metadata": [{"key": "down_blocks_0", "key_type": 2}, {"key": "resnets_0", "key_type": 2}, {"key": "time_emb_proj", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_0', 'resnets_1', 'conv1', 'bias')": {"key_metadata": [{"key": "down_blocks_0", "key_type": 2}, {"key": "resnets_1", "key_type": 2}, {"key": "conv1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_0', 'resnets_1', 'conv1', 'kernel')": {"key_metadata": [{"key": "down_blocks_0", "key_type": 2}, {"key": "resnets_1", "key_type": 2}, {"key": "conv1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_0', 'resnets_1', 'conv2', 'bias')": {"key_metadata": [{"key": "down_blocks_0", "key_type": 2}, {"key": "resnets_1", "key_type": 2}, {"key": "conv2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_0', 'resnets_1', 'conv2', 'kernel')": {"key_metadata": [{"key": "down_blocks_0", "key_type": 2}, {"key": "resnets_1", "key_type": 2}, {"key": "conv2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_0', 'resnets_1', 'norm1', 'bias')": {"key_metadata": [{"key": "down_blocks_0", "key_type": 2}, {"key": "resnets_1", "key_type": 2}, {"key": "norm1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_0', 'resnets_1', 'norm1', 'scale')": {"key_metadata": [{"key": "down_blocks_0", "key_type": 2}, {"key": "resnets_1", "key_type": 2}, {"key": "norm1", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_0', 'resnets_1', 'norm2', 'bias')": {"key_metadata": [{"key": "down_blocks_0", "key_type": 2}, {"key": "resnets_1", "key_type": 2}, {"key": "norm2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_0', 'resnets_1', 'norm2', 'scale')": {"key_metadata": [{"key": "down_blocks_0", "key_type": 2}, {"key": "resnets_1", "key_type": 2}, {"key": "norm2", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_0', 'resnets_1', 'time_emb_proj', 'bias')": {"key_metadata": [{"key": "down_blocks_0", "key_type": 2}, {"key": "resnets_1", "key_type": 2}, {"key": "time_emb_proj", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_0', 'resnets_1', 'time_emb_proj', 'kernel')": {"key_metadata": [{"key": "down_blocks_0", "key_type": 2}, {"key": "resnets_1", "key_type": 2}, {"key": "time_emb_proj", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_1', 'attentions_0', 'norm', 'bias')": {"key_metadata": [{"key": "down_blocks_1", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "norm", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_1', 'attentions_0', 'norm', 'scale')": {"key_metadata": [{"key": "down_blocks_1", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "norm", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_1', 'attentions_0', 'proj_in', 'bias')": {"key_metadata": [{"key": "down_blocks_1", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "proj_in", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_1', 'attentions_0', 'proj_in', 'kernel')": {"key_metadata": [{"key": "down_blocks_1", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "proj_in", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_1', 'attentions_0', 'proj_out', 'bias')": {"key_metadata": [{"key": "down_blocks_1", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "proj_out", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_1', 'attentions_0', 'proj_out', 'kernel')": {"key_metadata": [{"key": "down_blocks_1", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "proj_out", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_1', 'attentions_0', 'transformer_blocks_0', 'attn1', 'to_k', 'kernel')": {"key_metadata": [{"key": "down_blocks_1", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "attn1", "key_type": 2}, {"key": "to_k", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_1', 'attentions_0', 'transformer_blocks_0', 'attn1', 'to_out_0', 'bias')": {"key_metadata": [{"key": "down_blocks_1", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "attn1", "key_type": 2}, {"key": "to_out_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_1', 'attentions_0', 'transformer_blocks_0', 'attn1', 'to_out_0', 'kernel')": {"key_metadata": [{"key": "down_blocks_1", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "attn1", "key_type": 2}, {"key": "to_out_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_1', 'attentions_0', 'transformer_blocks_0', 'attn1', 'to_q', 'kernel')": {"key_metadata": [{"key": "down_blocks_1", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "attn1", "key_type": 2}, {"key": "to_q", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_1', 'attentions_0', 'transformer_blocks_0', 'attn1', 'to_v', 'kernel')": {"key_metadata": [{"key": "down_blocks_1", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "attn1", "key_type": 2}, {"key": "to_v", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_1', 'attentions_0', 'transformer_blocks_0', 'attn2', 'to_k', 'kernel')": {"key_metadata": [{"key": "down_blocks_1", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "attn2", "key_type": 2}, {"key": "to_k", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_1', 'attentions_0', 'transformer_blocks_0', 'attn2', 'to_out_0', 'bias')": {"key_metadata": [{"key": "down_blocks_1", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "attn2", "key_type": 2}, {"key": "to_out_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_1', 'attentions_0', 'transformer_blocks_0', 'attn2', 'to_out_0', 'kernel')": {"key_metadata": [{"key": "down_blocks_1", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "attn2", "key_type": 2}, {"key": "to_out_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_1', 'attentions_0', 'transformer_blocks_0', 'attn2', 'to_q', 'kernel')": {"key_metadata": [{"key": "down_blocks_1", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "attn2", "key_type": 2}, {"key": "to_q", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_1', 'attentions_0', 'transformer_blocks_0', 'attn2', 'to_v', 'kernel')": {"key_metadata": [{"key": "down_blocks_1", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "attn2", "key_type": 2}, {"key": "to_v", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_1', 'attentions_0', 'transformer_blocks_0', 'ff', 'net_0', 'proj', 'bias')": {"key_metadata": [{"key": "down_blocks_1", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "ff", "key_type": 2}, {"key": "net_0", "key_type": 2}, {"key": "proj", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_1', 'attentions_0', 'transformer_blocks_0', 'ff', 'net_0', 'proj', 'kernel')": {"key_metadata": [{"key": "down_blocks_1", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "ff", "key_type": 2}, {"key": "net_0", "key_type": 2}, {"key": "proj", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_1', 'attentions_0', 'transformer_blocks_0', 'ff', 'net_2', 'bias')": {"key_metadata": [{"key": "down_blocks_1", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "ff", "key_type": 2}, {"key": "net_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_1', 'attentions_0', 'transformer_blocks_0', 'ff', 'net_2', 'kernel')": {"key_metadata": [{"key": "down_blocks_1", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "ff", "key_type": 2}, {"key": "net_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_1', 'attentions_0', 'transformer_blocks_0', 'norm1', 'bias')": {"key_metadata": [{"key": "down_blocks_1", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "norm1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_1', 'attentions_0', 'transformer_blocks_0', 'norm1', 'scale')": {"key_metadata": [{"key": "down_blocks_1", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "norm1", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_1', 'attentions_0', 'transformer_blocks_0', 'norm2', 'bias')": {"key_metadata": [{"key": "down_blocks_1", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "norm2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_1', 'attentions_0', 'transformer_blocks_0', 'norm2', 'scale')": {"key_metadata": [{"key": "down_blocks_1", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "norm2", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_1', 'attentions_0', 'transformer_blocks_0', 'norm3', 'bias')": {"key_metadata": [{"key": "down_blocks_1", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "norm3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_1', 'attentions_0', 'transformer_blocks_0', 'norm3', 'scale')": {"key_metadata": [{"key": "down_blocks_1", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "norm3", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_1', 'attentions_1', 'norm', 'bias')": {"key_metadata": [{"key": "down_blocks_1", "key_type": 2}, {"key": "attentions_1", "key_type": 2}, {"key": "norm", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_1', 'attentions_1', 'norm', 'scale')": {"key_metadata": [{"key": "down_blocks_1", "key_type": 2}, {"key": "attentions_1", "key_type": 2}, {"key": "norm", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_1', 'attentions_1', 'proj_in', 'bias')": {"key_metadata": [{"key": "down_blocks_1", "key_type": 2}, {"key": "attentions_1", "key_type": 2}, {"key": "proj_in", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_1', 'attentions_1', 'proj_in', 'kernel')": {"key_metadata": [{"key": "down_blocks_1", "key_type": 2}, {"key": "attentions_1", "key_type": 2}, {"key": "proj_in", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_1', 'attentions_1', 'proj_out', 'bias')": {"key_metadata": [{"key": "down_blocks_1", "key_type": 2}, {"key": "attentions_1", "key_type": 2}, {"key": "proj_out", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_1', 'attentions_1', 'proj_out', 'kernel')": {"key_metadata": [{"key": "down_blocks_1", "key_type": 2}, {"key": "attentions_1", "key_type": 2}, {"key": "proj_out", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_1', 'attentions_1', 'transformer_blocks_0', 'attn1', 'to_k', 'kernel')": {"key_metadata": [{"key": "down_blocks_1", "key_type": 2}, {"key": "attentions_1", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "attn1", "key_type": 2}, {"key": "to_k", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_1', 'attentions_1', 'transformer_blocks_0', 'attn1', 'to_out_0', 'bias')": {"key_metadata": [{"key": "down_blocks_1", "key_type": 2}, {"key": "attentions_1", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "attn1", "key_type": 2}, {"key": "to_out_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_1', 'attentions_1', 'transformer_blocks_0', 'attn1', 'to_out_0', 'kernel')": {"key_metadata": [{"key": "down_blocks_1", "key_type": 2}, {"key": "attentions_1", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "attn1", "key_type": 2}, {"key": "to_out_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_1', 'attentions_1', 'transformer_blocks_0', 'attn1', 'to_q', 'kernel')": {"key_metadata": [{"key": "down_blocks_1", "key_type": 2}, {"key": "attentions_1", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "attn1", "key_type": 2}, {"key": "to_q", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_1', 'attentions_1', 'transformer_blocks_0', 'attn1', 'to_v', 'kernel')": {"key_metadata": [{"key": "down_blocks_1", "key_type": 2}, {"key": "attentions_1", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "attn1", "key_type": 2}, {"key": "to_v", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_1', 'attentions_1', 'transformer_blocks_0', 'attn2', 'to_k', 'kernel')": {"key_metadata": [{"key": "down_blocks_1", "key_type": 2}, {"key": "attentions_1", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "attn2", "key_type": 2}, {"key": "to_k", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_1', 'attentions_1', 'transformer_blocks_0', 'attn2', 'to_out_0', 'bias')": {"key_metadata": [{"key": "down_blocks_1", "key_type": 2}, {"key": "attentions_1", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "attn2", "key_type": 2}, {"key": "to_out_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_1', 'attentions_1', 'transformer_blocks_0', 'attn2', 'to_out_0', 'kernel')": {"key_metadata": [{"key": "down_blocks_1", "key_type": 2}, {"key": "attentions_1", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "attn2", "key_type": 2}, {"key": "to_out_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_1', 'attentions_1', 'transformer_blocks_0', 'attn2', 'to_q', 'kernel')": {"key_metadata": [{"key": "down_blocks_1", "key_type": 2}, {"key": "attentions_1", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "attn2", "key_type": 2}, {"key": "to_q", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_1', 'attentions_1', 'transformer_blocks_0', 'attn2', 'to_v', 'kernel')": {"key_metadata": [{"key": "down_blocks_1", "key_type": 2}, {"key": "attentions_1", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "attn2", "key_type": 2}, {"key": "to_v", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_1', 'attentions_1', 'transformer_blocks_0', 'ff', 'net_0', 'proj', 'bias')": {"key_metadata": [{"key": "down_blocks_1", "key_type": 2}, {"key": "attentions_1", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "ff", "key_type": 2}, {"key": "net_0", "key_type": 2}, {"key": "proj", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_1', 'attentions_1', 'transformer_blocks_0', 'ff', 'net_0', 'proj', 'kernel')": {"key_metadata": [{"key": "down_blocks_1", "key_type": 2}, {"key": "attentions_1", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "ff", "key_type": 2}, {"key": "net_0", "key_type": 2}, {"key": "proj", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_1', 'attentions_1', 'transformer_blocks_0', 'ff', 'net_2', 'bias')": {"key_metadata": [{"key": "down_blocks_1", "key_type": 2}, {"key": "attentions_1", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "ff", "key_type": 2}, {"key": "net_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_1', 'attentions_1', 'transformer_blocks_0', 'ff', 'net_2', 'kernel')": {"key_metadata": [{"key": "down_blocks_1", "key_type": 2}, {"key": "attentions_1", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "ff", "key_type": 2}, {"key": "net_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_1', 'attentions_1', 'transformer_blocks_0', 'norm1', 'bias')": {"key_metadata": [{"key": "down_blocks_1", "key_type": 2}, {"key": "attentions_1", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "norm1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_1', 'attentions_1', 'transformer_blocks_0', 'norm1', 'scale')": {"key_metadata": [{"key": "down_blocks_1", "key_type": 2}, {"key": "attentions_1", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "norm1", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_1', 'attentions_1', 'transformer_blocks_0', 'norm2', 'bias')": {"key_metadata": [{"key": "down_blocks_1", "key_type": 2}, {"key": "attentions_1", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "norm2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_1', 'attentions_1', 'transformer_blocks_0', 'norm2', 'scale')": {"key_metadata": [{"key": "down_blocks_1", "key_type": 2}, {"key": "attentions_1", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "norm2", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_1', 'attentions_1', 'transformer_blocks_0', 'norm3', 'bias')": {"key_metadata": [{"key": "down_blocks_1", "key_type": 2}, {"key": "attentions_1", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "norm3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_1', 'attentions_1', 'transformer_blocks_0', 'norm3', 'scale')": {"key_metadata": [{"key": "down_blocks_1", "key_type": 2}, {"key": "attentions_1", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "norm3", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_1', 'downsamplers_0', 'conv', 'bias')": {"key_metadata": [{"key": "down_blocks_1", "key_type": 2}, {"key": "downsamplers_0", "key_type": 2}, {"key": "conv", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_1', 'downsamplers_0', 'conv', 'kernel')": {"key_metadata": [{"key": "down_blocks_1", "key_type": 2}, {"key": "downsamplers_0", "key_type": 2}, {"key": "conv", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_1', 'resnets_0', 'conv1', 'bias')": {"key_metadata": [{"key": "down_blocks_1", "key_type": 2}, {"key": "resnets_0", "key_type": 2}, {"key": "conv1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_1', 'resnets_0', 'conv1', 'kernel')": {"key_metadata": [{"key": "down_blocks_1", "key_type": 2}, {"key": "resnets_0", "key_type": 2}, {"key": "conv1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_1', 'resnets_0', 'conv2', 'bias')": {"key_metadata": [{"key": "down_blocks_1", "key_type": 2}, {"key": "resnets_0", "key_type": 2}, {"key": "conv2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_1', 'resnets_0', 'conv2', 'kernel')": {"key_metadata": [{"key": "down_blocks_1", "key_type": 2}, {"key": "resnets_0", "key_type": 2}, {"key": "conv2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_1', 'resnets_0', 'conv_shortcut', 'bias')": {"key_metadata": [{"key": "down_blocks_1", "key_type": 2}, {"key": "resnets_0", "key_type": 2}, {"key": "conv_shortcut", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_1', 'resnets_0', 'conv_shortcut', 'kernel')": {"key_metadata": [{"key": "down_blocks_1", "key_type": 2}, {"key": "resnets_0", "key_type": 2}, {"key": "conv_shortcut", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_1', 'resnets_0', 'norm1', 'bias')": {"key_metadata": [{"key": "down_blocks_1", "key_type": 2}, {"key": "resnets_0", "key_type": 2}, {"key": "norm1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_1', 'resnets_0', 'norm1', 'scale')": {"key_metadata": [{"key": "down_blocks_1", "key_type": 2}, {"key": "resnets_0", "key_type": 2}, {"key": "norm1", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_1', 'resnets_0', 'norm2', 'bias')": {"key_metadata": [{"key": "down_blocks_1", "key_type": 2}, {"key": "resnets_0", "key_type": 2}, {"key": "norm2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_1', 'resnets_0', 'norm2', 'scale')": {"key_metadata": [{"key": "down_blocks_1", "key_type": 2}, {"key": "resnets_0", "key_type": 2}, {"key": "norm2", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_1', 'resnets_0', 'time_emb_proj', 'bias')": {"key_metadata": [{"key": "down_blocks_1", "key_type": 2}, {"key": "resnets_0", "key_type": 2}, {"key": "time_emb_proj", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_1', 'resnets_0', 'time_emb_proj', 'kernel')": {"key_metadata": [{"key": "down_blocks_1", "key_type": 2}, {"key": "resnets_0", "key_type": 2}, {"key": "time_emb_proj", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_1', 'resnets_1', 'conv1', 'bias')": {"key_metadata": [{"key": "down_blocks_1", "key_type": 2}, {"key": "resnets_1", "key_type": 2}, {"key": "conv1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_1', 'resnets_1', 'conv1', 'kernel')": {"key_metadata": [{"key": "down_blocks_1", "key_type": 2}, {"key": "resnets_1", "key_type": 2}, {"key": "conv1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_1', 'resnets_1', 'conv2', 'bias')": {"key_metadata": [{"key": "down_blocks_1", "key_type": 2}, {"key": "resnets_1", "key_type": 2}, {"key": "conv2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_1', 'resnets_1', 'conv2', 'kernel')": {"key_metadata": [{"key": "down_blocks_1", "key_type": 2}, {"key": "resnets_1", "key_type": 2}, {"key": "conv2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_1', 'resnets_1', 'norm1', 'bias')": {"key_metadata": [{"key": "down_blocks_1", "key_type": 2}, {"key": "resnets_1", "key_type": 2}, {"key": "norm1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_1', 'resnets_1', 'norm1', 'scale')": {"key_metadata": [{"key": "down_blocks_1", "key_type": 2}, {"key": "resnets_1", "key_type": 2}, {"key": "norm1", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_1', 'resnets_1', 'norm2', 'bias')": {"key_metadata": [{"key": "down_blocks_1", "key_type": 2}, {"key": "resnets_1", "key_type": 2}, {"key": "norm2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_1', 'resnets_1', 'norm2', 'scale')": {"key_metadata": [{"key": "down_blocks_1", "key_type": 2}, {"key": "resnets_1", "key_type": 2}, {"key": "norm2", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_1', 'resnets_1', 'time_emb_proj', 'bias')": {"key_metadata": [{"key": "down_blocks_1", "key_type": 2}, {"key": "resnets_1", "key_type": 2}, {"key": "time_emb_proj", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_1', 'resnets_1', 'time_emb_proj', 'kernel')": {"key_metadata": [{"key": "down_blocks_1", "key_type": 2}, {"key": "resnets_1", "key_type": 2}, {"key": "time_emb_proj", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_2', 'attentions_0', 'norm', 'bias')": {"key_metadata": [{"key": "down_blocks_2", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "norm", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_2', 'attentions_0', 'norm', 'scale')": {"key_metadata": [{"key": "down_blocks_2", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "norm", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_2', 'attentions_0', 'proj_in', 'bias')": {"key_metadata": [{"key": "down_blocks_2", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "proj_in", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_2', 'attentions_0', 'proj_in', 'kernel')": {"key_metadata": [{"key": "down_blocks_2", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "proj_in", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_2', 'attentions_0', 'proj_out', 'bias')": {"key_metadata": [{"key": "down_blocks_2", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "proj_out", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_2', 'attentions_0', 'proj_out', 'kernel')": {"key_metadata": [{"key": "down_blocks_2", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "proj_out", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_2', 'attentions_0', 'transformer_blocks_0', 'attn1', 'to_k', 'kernel')": {"key_metadata": [{"key": "down_blocks_2", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "attn1", "key_type": 2}, {"key": "to_k", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_2', 'attentions_0', 'transformer_blocks_0', 'attn1', 'to_out_0', 'bias')": {"key_metadata": [{"key": "down_blocks_2", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "attn1", "key_type": 2}, {"key": "to_out_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_2', 'attentions_0', 'transformer_blocks_0', 'attn1', 'to_out_0', 'kernel')": {"key_metadata": [{"key": "down_blocks_2", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "attn1", "key_type": 2}, {"key": "to_out_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_2', 'attentions_0', 'transformer_blocks_0', 'attn1', 'to_q', 'kernel')": {"key_metadata": [{"key": "down_blocks_2", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "attn1", "key_type": 2}, {"key": "to_q", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_2', 'attentions_0', 'transformer_blocks_0', 'attn1', 'to_v', 'kernel')": {"key_metadata": [{"key": "down_blocks_2", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "attn1", "key_type": 2}, {"key": "to_v", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_2', 'attentions_0', 'transformer_blocks_0', 'attn2', 'to_k', 'kernel')": {"key_metadata": [{"key": "down_blocks_2", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "attn2", "key_type": 2}, {"key": "to_k", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_2', 'attentions_0', 'transformer_blocks_0', 'attn2', 'to_out_0', 'bias')": {"key_metadata": [{"key": "down_blocks_2", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "attn2", "key_type": 2}, {"key": "to_out_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_2', 'attentions_0', 'transformer_blocks_0', 'attn2', 'to_out_0', 'kernel')": {"key_metadata": [{"key": "down_blocks_2", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "attn2", "key_type": 2}, {"key": "to_out_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_2', 'attentions_0', 'transformer_blocks_0', 'attn2', 'to_q', 'kernel')": {"key_metadata": [{"key": "down_blocks_2", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "attn2", "key_type": 2}, {"key": "to_q", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_2', 'attentions_0', 'transformer_blocks_0', 'attn2', 'to_v', 'kernel')": {"key_metadata": [{"key": "down_blocks_2", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "attn2", "key_type": 2}, {"key": "to_v", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_2', 'attentions_0', 'transformer_blocks_0', 'ff', 'net_0', 'proj', 'bias')": {"key_metadata": [{"key": "down_blocks_2", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "ff", "key_type": 2}, {"key": "net_0", "key_type": 2}, {"key": "proj", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_2', 'attentions_0', 'transformer_blocks_0', 'ff', 'net_0', 'proj', 'kernel')": {"key_metadata": [{"key": "down_blocks_2", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "ff", "key_type": 2}, {"key": "net_0", "key_type": 2}, {"key": "proj", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_2', 'attentions_0', 'transformer_blocks_0', 'ff', 'net_2', 'bias')": {"key_metadata": [{"key": "down_blocks_2", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "ff", "key_type": 2}, {"key": "net_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_2', 'attentions_0', 'transformer_blocks_0', 'ff', 'net_2', 'kernel')": {"key_metadata": [{"key": "down_blocks_2", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "ff", "key_type": 2}, {"key": "net_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_2', 'attentions_0', 'transformer_blocks_0', 'norm1', 'bias')": {"key_metadata": [{"key": "down_blocks_2", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "norm1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_2', 'attentions_0', 'transformer_blocks_0', 'norm1', 'scale')": {"key_metadata": [{"key": "down_blocks_2", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "norm1", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_2', 'attentions_0', 'transformer_blocks_0', 'norm2', 'bias')": {"key_metadata": [{"key": "down_blocks_2", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "norm2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_2', 'attentions_0', 'transformer_blocks_0', 'norm2', 'scale')": {"key_metadata": [{"key": "down_blocks_2", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "norm2", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_2', 'attentions_0', 'transformer_blocks_0', 'norm3', 'bias')": {"key_metadata": [{"key": "down_blocks_2", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "norm3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_2', 'attentions_0', 'transformer_blocks_0', 'norm3', 'scale')": {"key_metadata": [{"key": "down_blocks_2", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "norm3", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_2', 'attentions_1', 'norm', 'bias')": {"key_metadata": [{"key": "down_blocks_2", "key_type": 2}, {"key": "attentions_1", "key_type": 2}, {"key": "norm", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_2', 'attentions_1', 'norm', 'scale')": {"key_metadata": [{"key": "down_blocks_2", "key_type": 2}, {"key": "attentions_1", "key_type": 2}, {"key": "norm", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_2', 'attentions_1', 'proj_in', 'bias')": {"key_metadata": [{"key": "down_blocks_2", "key_type": 2}, {"key": "attentions_1", "key_type": 2}, {"key": "proj_in", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_2', 'attentions_1', 'proj_in', 'kernel')": {"key_metadata": [{"key": "down_blocks_2", "key_type": 2}, {"key": "attentions_1", "key_type": 2}, {"key": "proj_in", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_2', 'attentions_1', 'proj_out', 'bias')": {"key_metadata": [{"key": "down_blocks_2", "key_type": 2}, {"key": "attentions_1", "key_type": 2}, {"key": "proj_out", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_2', 'attentions_1', 'proj_out', 'kernel')": {"key_metadata": [{"key": "down_blocks_2", "key_type": 2}, {"key": "attentions_1", "key_type": 2}, {"key": "proj_out", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_2', 'attentions_1', 'transformer_blocks_0', 'attn1', 'to_k', 'kernel')": {"key_metadata": [{"key": "down_blocks_2", "key_type": 2}, {"key": "attentions_1", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "attn1", "key_type": 2}, {"key": "to_k", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_2', 'attentions_1', 'transformer_blocks_0', 'attn1', 'to_out_0', 'bias')": {"key_metadata": [{"key": "down_blocks_2", "key_type": 2}, {"key": "attentions_1", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "attn1", "key_type": 2}, {"key": "to_out_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_2', 'attentions_1', 'transformer_blocks_0', 'attn1', 'to_out_0', 'kernel')": {"key_metadata": [{"key": "down_blocks_2", "key_type": 2}, {"key": "attentions_1", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "attn1", "key_type": 2}, {"key": "to_out_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_2', 'attentions_1', 'transformer_blocks_0', 'attn1', 'to_q', 'kernel')": {"key_metadata": [{"key": "down_blocks_2", "key_type": 2}, {"key": "attentions_1", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "attn1", "key_type": 2}, {"key": "to_q", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_2', 'attentions_1', 'transformer_blocks_0', 'attn1', 'to_v', 'kernel')": {"key_metadata": [{"key": "down_blocks_2", "key_type": 2}, {"key": "attentions_1", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "attn1", "key_type": 2}, {"key": "to_v", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_2', 'attentions_1', 'transformer_blocks_0', 'attn2', 'to_k', 'kernel')": {"key_metadata": [{"key": "down_blocks_2", "key_type": 2}, {"key": "attentions_1", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "attn2", "key_type": 2}, {"key": "to_k", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_2', 'attentions_1', 'transformer_blocks_0', 'attn2', 'to_out_0', 'bias')": {"key_metadata": [{"key": "down_blocks_2", "key_type": 2}, {"key": "attentions_1", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "attn2", "key_type": 2}, {"key": "to_out_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_2', 'attentions_1', 'transformer_blocks_0', 'attn2', 'to_out_0', 'kernel')": {"key_metadata": [{"key": "down_blocks_2", "key_type": 2}, {"key": "attentions_1", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "attn2", "key_type": 2}, {"key": "to_out_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_2', 'attentions_1', 'transformer_blocks_0', 'attn2', 'to_q', 'kernel')": {"key_metadata": [{"key": "down_blocks_2", "key_type": 2}, {"key": "attentions_1", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "attn2", "key_type": 2}, {"key": "to_q", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_2', 'attentions_1', 'transformer_blocks_0', 'attn2', 'to_v', 'kernel')": {"key_metadata": [{"key": "down_blocks_2", "key_type": 2}, {"key": "attentions_1", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "attn2", "key_type": 2}, {"key": "to_v", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_2', 'attentions_1', 'transformer_blocks_0', 'ff', 'net_0', 'proj', 'bias')": {"key_metadata": [{"key": "down_blocks_2", "key_type": 2}, {"key": "attentions_1", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "ff", "key_type": 2}, {"key": "net_0", "key_type": 2}, {"key": "proj", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_2', 'attentions_1', 'transformer_blocks_0', 'ff', 'net_0', 'proj', 'kernel')": {"key_metadata": [{"key": "down_blocks_2", "key_type": 2}, {"key": "attentions_1", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "ff", "key_type": 2}, {"key": "net_0", "key_type": 2}, {"key": "proj", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_2', 'attentions_1', 'transformer_blocks_0', 'ff', 'net_2', 'bias')": {"key_metadata": [{"key": "down_blocks_2", "key_type": 2}, {"key": "attentions_1", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "ff", "key_type": 2}, {"key": "net_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_2', 'attentions_1', 'transformer_blocks_0', 'ff', 'net_2', 'kernel')": {"key_metadata": [{"key": "down_blocks_2", "key_type": 2}, {"key": "attentions_1", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "ff", "key_type": 2}, {"key": "net_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_2', 'attentions_1', 'transformer_blocks_0', 'norm1', 'bias')": {"key_metadata": [{"key": "down_blocks_2", "key_type": 2}, {"key": "attentions_1", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "norm1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_2', 'attentions_1', 'transformer_blocks_0', 'norm1', 'scale')": {"key_metadata": [{"key": "down_blocks_2", "key_type": 2}, {"key": "attentions_1", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "norm1", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_2', 'attentions_1', 'transformer_blocks_0', 'norm2', 'bias')": {"key_metadata": [{"key": "down_blocks_2", "key_type": 2}, {"key": "attentions_1", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "norm2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_2', 'attentions_1', 'transformer_blocks_0', 'norm2', 'scale')": {"key_metadata": [{"key": "down_blocks_2", "key_type": 2}, {"key": "attentions_1", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "norm2", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_2', 'attentions_1', 'transformer_blocks_0', 'norm3', 'bias')": {"key_metadata": [{"key": "down_blocks_2", "key_type": 2}, {"key": "attentions_1", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "norm3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_2', 'attentions_1', 'transformer_blocks_0', 'norm3', 'scale')": {"key_metadata": [{"key": "down_blocks_2", "key_type": 2}, {"key": "attentions_1", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "norm3", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_2', 'downsamplers_0', 'conv', 'bias')": {"key_metadata": [{"key": "down_blocks_2", "key_type": 2}, {"key": "downsamplers_0", "key_type": 2}, {"key": "conv", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_2', 'downsamplers_0', 'conv', 'kernel')": {"key_metadata": [{"key": "down_blocks_2", "key_type": 2}, {"key": "downsamplers_0", "key_type": 2}, {"key": "conv", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_2', 'resnets_0', 'conv1', 'bias')": {"key_metadata": [{"key": "down_blocks_2", "key_type": 2}, {"key": "resnets_0", "key_type": 2}, {"key": "conv1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_2', 'resnets_0', 'conv1', 'kernel')": {"key_metadata": [{"key": "down_blocks_2", "key_type": 2}, {"key": "resnets_0", "key_type": 2}, {"key": "conv1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_2', 'resnets_0', 'conv2', 'bias')": {"key_metadata": [{"key": "down_blocks_2", "key_type": 2}, {"key": "resnets_0", "key_type": 2}, {"key": "conv2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_2', 'resnets_0', 'conv2', 'kernel')": {"key_metadata": [{"key": "down_blocks_2", "key_type": 2}, {"key": "resnets_0", "key_type": 2}, {"key": "conv2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_2', 'resnets_0', 'conv_shortcut', 'bias')": {"key_metadata": [{"key": "down_blocks_2", "key_type": 2}, {"key": "resnets_0", "key_type": 2}, {"key": "conv_shortcut", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_2', 'resnets_0', 'conv_shortcut', 'kernel')": {"key_metadata": [{"key": "down_blocks_2", "key_type": 2}, {"key": "resnets_0", "key_type": 2}, {"key": "conv_shortcut", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_2', 'resnets_0', 'norm1', 'bias')": {"key_metadata": [{"key": "down_blocks_2", "key_type": 2}, {"key": "resnets_0", "key_type": 2}, {"key": "norm1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_2', 'resnets_0', 'norm1', 'scale')": {"key_metadata": [{"key": "down_blocks_2", "key_type": 2}, {"key": "resnets_0", "key_type": 2}, {"key": "norm1", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_2', 'resnets_0', 'norm2', 'bias')": {"key_metadata": [{"key": "down_blocks_2", "key_type": 2}, {"key": "resnets_0", "key_type": 2}, {"key": "norm2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_2', 'resnets_0', 'norm2', 'scale')": {"key_metadata": [{"key": "down_blocks_2", "key_type": 2}, {"key": "resnets_0", "key_type": 2}, {"key": "norm2", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_2', 'resnets_0', 'time_emb_proj', 'bias')": {"key_metadata": [{"key": "down_blocks_2", "key_type": 2}, {"key": "resnets_0", "key_type": 2}, {"key": "time_emb_proj", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_2', 'resnets_0', 'time_emb_proj', 'kernel')": {"key_metadata": [{"key": "down_blocks_2", "key_type": 2}, {"key": "resnets_0", "key_type": 2}, {"key": "time_emb_proj", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_2', 'resnets_1', 'conv1', 'bias')": {"key_metadata": [{"key": "down_blocks_2", "key_type": 2}, {"key": "resnets_1", "key_type": 2}, {"key": "conv1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_2', 'resnets_1', 'conv1', 'kernel')": {"key_metadata": [{"key": "down_blocks_2", "key_type": 2}, {"key": "resnets_1", "key_type": 2}, {"key": "conv1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_2', 'resnets_1', 'conv2', 'bias')": {"key_metadata": [{"key": "down_blocks_2", "key_type": 2}, {"key": "resnets_1", "key_type": 2}, {"key": "conv2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_2', 'resnets_1', 'conv2', 'kernel')": {"key_metadata": [{"key": "down_blocks_2", "key_type": 2}, {"key": "resnets_1", "key_type": 2}, {"key": "conv2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_2', 'resnets_1', 'norm1', 'bias')": {"key_metadata": [{"key": "down_blocks_2", "key_type": 2}, {"key": "resnets_1", "key_type": 2}, {"key": "norm1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_2', 'resnets_1', 'norm1', 'scale')": {"key_metadata": [{"key": "down_blocks_2", "key_type": 2}, {"key": "resnets_1", "key_type": 2}, {"key": "norm1", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_2', 'resnets_1', 'norm2', 'bias')": {"key_metadata": [{"key": "down_blocks_2", "key_type": 2}, {"key": "resnets_1", "key_type": 2}, {"key": "norm2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_2', 'resnets_1', 'norm2', 'scale')": {"key_metadata": [{"key": "down_blocks_2", "key_type": 2}, {"key": "resnets_1", "key_type": 2}, {"key": "norm2", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_2', 'resnets_1', 'time_emb_proj', 'bias')": {"key_metadata": [{"key": "down_blocks_2", "key_type": 2}, {"key": "resnets_1", "key_type": 2}, {"key": "time_emb_proj", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_2', 'resnets_1', 'time_emb_proj', 'kernel')": {"key_metadata": [{"key": "down_blocks_2", "key_type": 2}, {"key": "resnets_1", "key_type": 2}, {"key": "time_emb_proj", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_3', 'resnets_0', 'conv1', 'bias')": {"key_metadata": [{"key": "down_blocks_3", "key_type": 2}, {"key": "resnets_0", "key_type": 2}, {"key": "conv1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_3', 'resnets_0', 'conv1', 'kernel')": {"key_metadata": [{"key": "down_blocks_3", "key_type": 2}, {"key": "resnets_0", "key_type": 2}, {"key": "conv1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_3', 'resnets_0', 'conv2', 'bias')": {"key_metadata": [{"key": "down_blocks_3", "key_type": 2}, {"key": "resnets_0", "key_type": 2}, {"key": "conv2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_3', 'resnets_0', 'conv2', 'kernel')": {"key_metadata": [{"key": "down_blocks_3", "key_type": 2}, {"key": "resnets_0", "key_type": 2}, {"key": "conv2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_3', 'resnets_0', 'norm1', 'bias')": {"key_metadata": [{"key": "down_blocks_3", "key_type": 2}, {"key": "resnets_0", "key_type": 2}, {"key": "norm1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_3', 'resnets_0', 'norm1', 'scale')": {"key_metadata": [{"key": "down_blocks_3", "key_type": 2}, {"key": "resnets_0", "key_type": 2}, {"key": "norm1", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_3', 'resnets_0', 'norm2', 'bias')": {"key_metadata": [{"key": "down_blocks_3", "key_type": 2}, {"key": "resnets_0", "key_type": 2}, {"key": "norm2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_3', 'resnets_0', 'norm2', 'scale')": {"key_metadata": [{"key": "down_blocks_3", "key_type": 2}, {"key": "resnets_0", "key_type": 2}, {"key": "norm2", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_3', 'resnets_0', 'time_emb_proj', 'bias')": {"key_metadata": [{"key": "down_blocks_3", "key_type": 2}, {"key": "resnets_0", "key_type": 2}, {"key": "time_emb_proj", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_3', 'resnets_0', 'time_emb_proj', 'kernel')": {"key_metadata": [{"key": "down_blocks_3", "key_type": 2}, {"key": "resnets_0", "key_type": 2}, {"key": "time_emb_proj", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_3', 'resnets_1', 'conv1', 'bias')": {"key_metadata": [{"key": "down_blocks_3", "key_type": 2}, {"key": "resnets_1", "key_type": 2}, {"key": "conv1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_3', 'resnets_1', 'conv1', 'kernel')": {"key_metadata": [{"key": "down_blocks_3", "key_type": 2}, {"key": "resnets_1", "key_type": 2}, {"key": "conv1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_3', 'resnets_1', 'conv2', 'bias')": {"key_metadata": [{"key": "down_blocks_3", "key_type": 2}, {"key": "resnets_1", "key_type": 2}, {"key": "conv2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_3', 'resnets_1', 'conv2', 'kernel')": {"key_metadata": [{"key": "down_blocks_3", "key_type": 2}, {"key": "resnets_1", "key_type": 2}, {"key": "conv2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_3', 'resnets_1', 'norm1', 'bias')": {"key_metadata": [{"key": "down_blocks_3", "key_type": 2}, {"key": "resnets_1", "key_type": 2}, {"key": "norm1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_3', 'resnets_1', 'norm1', 'scale')": {"key_metadata": [{"key": "down_blocks_3", "key_type": 2}, {"key": "resnets_1", "key_type": 2}, {"key": "norm1", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_3', 'resnets_1', 'norm2', 'bias')": {"key_metadata": [{"key": "down_blocks_3", "key_type": 2}, {"key": "resnets_1", "key_type": 2}, {"key": "norm2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_3', 'resnets_1', 'norm2', 'scale')": {"key_metadata": [{"key": "down_blocks_3", "key_type": 2}, {"key": "resnets_1", "key_type": 2}, {"key": "norm2", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_3', 'resnets_1', 'time_emb_proj', 'bias')": {"key_metadata": [{"key": "down_blocks_3", "key_type": 2}, {"key": "resnets_1", "key_type": 2}, {"key": "time_emb_proj", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('down_blocks_3', 'resnets_1', 'time_emb_proj', 'kernel')": {"key_metadata": [{"key": "down_blocks_3", "key_type": 2}, {"key": "resnets_1", "key_type": 2}, {"key": "time_emb_proj", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('mid_block', 'attentions_0', 'norm', 'bias')": {"key_metadata": [{"key": "mid_block", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "norm", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('mid_block', 'attentions_0', 'norm', 'scale')": {"key_metadata": [{"key": "mid_block", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "norm", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('mid_block', 'attentions_0', 'proj_in', 'bias')": {"key_metadata": [{"key": "mid_block", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "proj_in", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('mid_block', 'attentions_0', 'proj_in', 'kernel')": {"key_metadata": [{"key": "mid_block", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "proj_in", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('mid_block', 'attentions_0', 'proj_out', 'bias')": {"key_metadata": [{"key": "mid_block", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "proj_out", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('mid_block', 'attentions_0', 'proj_out', 'kernel')": {"key_metadata": [{"key": "mid_block", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "proj_out", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('mid_block', 'attentions_0', 'transformer_blocks_0', 'attn1', 'to_k', 'kernel')": {"key_metadata": [{"key": "mid_block", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "attn1", "key_type": 2}, {"key": "to_k", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('mid_block', 'attentions_0', 'transformer_blocks_0', 'attn1', 'to_out_0', 'bias')": {"key_metadata": [{"key": "mid_block", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "attn1", "key_type": 2}, {"key": "to_out_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('mid_block', 'attentions_0', 'transformer_blocks_0', 'attn1', 'to_out_0', 'kernel')": {"key_metadata": [{"key": "mid_block", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "attn1", "key_type": 2}, {"key": "to_out_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('mid_block', 'attentions_0', 'transformer_blocks_0', 'attn1', 'to_q', 'kernel')": {"key_metadata": [{"key": "mid_block", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "attn1", "key_type": 2}, {"key": "to_q", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('mid_block', 'attentions_0', 'transformer_blocks_0', 'attn1', 'to_v', 'kernel')": {"key_metadata": [{"key": "mid_block", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "attn1", "key_type": 2}, {"key": "to_v", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('mid_block', 'attentions_0', 'transformer_blocks_0', 'attn2', 'to_k', 'kernel')": {"key_metadata": [{"key": "mid_block", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "attn2", "key_type": 2}, {"key": "to_k", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('mid_block', 'attentions_0', 'transformer_blocks_0', 'attn2', 'to_out_0', 'bias')": {"key_metadata": [{"key": "mid_block", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "attn2", "key_type": 2}, {"key": "to_out_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('mid_block', 'attentions_0', 'transformer_blocks_0', 'attn2', 'to_out_0', 'kernel')": {"key_metadata": [{"key": "mid_block", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "attn2", "key_type": 2}, {"key": "to_out_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('mid_block', 'attentions_0', 'transformer_blocks_0', 'attn2', 'to_q', 'kernel')": {"key_metadata": [{"key": "mid_block", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "attn2", "key_type": 2}, {"key": "to_q", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('mid_block', 'attentions_0', 'transformer_blocks_0', 'attn2', 'to_v', 'kernel')": {"key_metadata": [{"key": "mid_block", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "attn2", "key_type": 2}, {"key": "to_v", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('mid_block', 'attentions_0', 'transformer_blocks_0', 'ff', 'net_0', 'proj', 'bias')": {"key_metadata": [{"key": "mid_block", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "ff", "key_type": 2}, {"key": "net_0", "key_type": 2}, {"key": "proj", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('mid_block', 'attentions_0', 'transformer_blocks_0', 'ff', 'net_0', 'proj', 'kernel')": {"key_metadata": [{"key": "mid_block", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "ff", "key_type": 2}, {"key": "net_0", "key_type": 2}, {"key": "proj", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('mid_block', 'attentions_0', 'transformer_blocks_0', 'ff', 'net_2', 'bias')": {"key_metadata": [{"key": "mid_block", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "ff", "key_type": 2}, {"key": "net_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('mid_block', 'attentions_0', 'transformer_blocks_0', 'ff', 'net_2', 'kernel')": {"key_metadata": [{"key": "mid_block", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "ff", "key_type": 2}, {"key": "net_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('mid_block', 'attentions_0', 'transformer_blocks_0', 'norm1', 'bias')": {"key_metadata": [{"key": "mid_block", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "norm1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('mid_block', 'attentions_0', 'transformer_blocks_0', 'norm1', 'scale')": {"key_metadata": [{"key": "mid_block", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "norm1", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('mid_block', 'attentions_0', 'transformer_blocks_0', 'norm2', 'bias')": {"key_metadata": [{"key": "mid_block", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "norm2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('mid_block', 'attentions_0', 'transformer_blocks_0', 'norm2', 'scale')": {"key_metadata": [{"key": "mid_block", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "norm2", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('mid_block', 'attentions_0', 'transformer_blocks_0', 'norm3', 'bias')": {"key_metadata": [{"key": "mid_block", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "norm3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('mid_block', 'attentions_0', 'transformer_blocks_0', 'norm3', 'scale')": {"key_metadata": [{"key": "mid_block", "key_type": 2}, {"key": "attentions_0", "key_type": 2}, {"key": "transformer_blocks_0", "key_type": 2}, {"key": "norm3", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('mid_block', 'resnets_0', 'conv1', 'bias')": {"key_metadata": [{"key": "mid_block", "key_type": 2}, {"key": "resnets_0", "key_type": 2}, {"key": "conv1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('mid_block', 'resnets_0', 'conv1', 'kernel')": {"key_metadata": [{"key": "mid_block", "key_type": 2}, {"key": "resnets_0", "key_type": 2}, {"key": "conv1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('mid_block', 'resnets_0', 'conv2', 'bias')": {"key_metadata": [{"key": "mid_block", "key_type": 2}, {"key": "resnets_0", "key_type": 2}, {"key": "conv2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('mid_block', 'resnets_0', 'conv2', 'kernel')": {"key_metadata": [{"key": "mid_block", "key_type": 2}, {"key": "resnets_0", "key_type": 2}, {"key": "conv2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('mid_block', 'resnets_0', 'norm1', 'bias')": {"key_metadata": [{"key": "mid_block", "key_type": 2}, {"key": "resnets_0", "key_type": 2}, {"key": "norm1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('mid_block', 'resnets_0', 'norm1', 'scale')": {"key_metadata": [{"key": "mid_block", "key_type": 2}, {"key": "resnets_0", "key_type": 2}, {"key": "norm1", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('mid_block', 'resnets_0', 'norm2', 'bias')": {"key_metadata": [{"key": "mid_block", "key_type": 2}, {"key": "resnets_0", "key_type": 2}, {"key": "norm2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('mid_block', 'resnets_0', 'norm2', 'scale')": {"key_metadata": [{"key": "mid_block", "key_type": 2}, {"key": "resnets_0", "key_type": 2}, {"key": "norm2", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('mid_block', 'resnets_0', 'time_emb_proj', 'bias')": {"key_metadata": [{"key": "mid_block", "key_type": 2}, {"key": "resnets_0", "key_type": 2}, {"key": "time_emb_proj", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('mid_block', 'resnets_0', 'time_emb_proj', 'kernel')": {"key_metadata": [{"key": "mid_block", "key_type": 2}, {"key": "resnets_0", "key_type": 2}, {"key": "time_emb_proj", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('mid_block', 'resnets_1', 'conv1', 'bias')": {"key_metadata": [{"key": "mid_block", "key_type": 2}, {"key": "resnets_1", "key_type": 2}, {"key": "conv1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('mid_block', 'resnets_1', 'conv1', 'kernel')": {"key_metadata": [{"key": "mid_block", "key_type": 2}, {"key": "resnets_1", "key_type": 2}, {"key": "conv1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('mid_block', 'resnets_1', 'conv2', 'bias')": {"key_metadata": [{"key": "mid_block", "key_type": 2}, {"key": "resnets_1", "key_type": 2}, {"key": "conv2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('mid_block', 'resnets_1', 'conv2', 'kernel')": {"key_metadata": [{"key": "mid_block", "key_type": 2}, {"key": "resnets_1", "key_type": 2}, {"key": "conv2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('mid_block', 'resnets_1', 'norm1', 'bias')": {"key_metadata": [{"key": "mid_block", "key_type": 2}, {"key": "resnets_1", "key_type": 2}, {"key": "norm1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('mid_block', 'resnets_1', 'norm1', 'scale')": {"key_metadata": [{"key": "mid_block", "key_type": 2}, {"key": "resnets_1", "key_type": 2}, {"key": "norm1", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('mid_block', 'resnets_1', 'norm2', 'bias')": {"key_metadata": [{"key": "mid_block", "key_type": 2}, {"key": "resnets_1", "key_type": 2}, {"key": "norm2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('mid_block', 'resnets_1', 'norm2', 'scale')": {"key_metadata": [{"key": "mid_block", "key_type": 2}, {"key": "resnets_1", "key_type": 2}, {"key": "norm2", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('mid_block', 'resnets_1', 'time_emb_proj', 'bias')": {"key_metadata": [{"key": "mid_block", "key_type": 2}, {"key": "resnets_1", "key_type": 2}, {"key": "time_emb_proj", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('mid_block', 'resnets_1', 'time_emb_proj', 'kernel')": {"key_metadata": [{"key": "mid_block", "key_type": 2}, {"key": "resnets_1", "key_type": 2}, {"key": "time_emb_proj", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('time_embedding', 'linear_1', 'bias')": {"key_metadata": [{"key": "time_embedding", "key_type": 2}, {"key": "linear_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('time_embedding', 'linear_1', 'kernel')": {"key_metadata": [{"key": "time_embedding", "key_type": 2}, {"key": "linear_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('time_embedding', 'linear_2', 'bias')": {"key_metadata": [{"key": "time_embedding", "key_type": 2}, {"key": "linear_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}, "('time_embedding', 'linear_2', 'kernel')": {"key_metadata": [{"key": "time_embedding", "key_type": 2}, {"key": "linear_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": true}}}, "use_zarr3": false}