2b-it/_METADATA

{"tree_metadata": {"('blocks.0', 'channel_pre_norm', 'scale')": {"key_metadata": [{"key": "blocks.0", "key_type": 2}, {"key": "channel_pre_norm", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.0', 'mlp_block', 'ffw_down', 'bias')": {"key_metadata": [{"key": "blocks.0", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_down", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.0', 'mlp_block', 'ffw_down', 'kernel')": {"key_metadata": [{"key": "blocks.0", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_down", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.0', 'mlp_block', 'ffw_up', 'b')": {"key_metadata": [{"key": "blocks.0", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_up", "key_type": 2}, {"key": "b", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.0', 'mlp_block', 'ffw_up', 'w')": {"key_metadata": [{"key": "blocks.0", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_up", "key_type": 2}, {"key": "w", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.0', 'recurrent_block', 'conv_1d', 'b')": {"key_metadata": [{"key": "blocks.0", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "conv_1d", "key_type": 2}, {"key": "b", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.0', 'recurrent_block', 'conv_1d', 'w')": {"key_metadata": [{"key": "blocks.0", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "conv_1d", "key_type": 2}, {"key": "w", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.0', 'recurrent_block', 'linear_out', 'bias')": {"key_metadata": [{"key": "blocks.0", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_out", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.0', 'recurrent_block', 'linear_out', 'kernel')": {"key_metadata": [{"key": "blocks.0", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_out", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.0', 'recurrent_block', 'linear_x', 'bias')": {"key_metadata": [{"key": "blocks.0", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_x", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.0', 'recurrent_block', 'linear_x', 'kernel')": {"key_metadata": [{"key": "blocks.0", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_x", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.0', 'recurrent_block', 'linear_y', 'bias')": {"key_metadata": [{"key": "blocks.0", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_y", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.0', 'recurrent_block', 'linear_y', 'kernel')": {"key_metadata": [{"key": "blocks.0", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_y", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.0', 'recurrent_block', 'rg_lru', 'a_gate', 'b')": {"key_metadata": [{"key": "blocks.0", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "a_gate", "key_type": 2}, {"key": "b", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.0', 'recurrent_block', 'rg_lru', 'a_gate', 'w')": {"key_metadata": [{"key": "blocks.0", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "a_gate", "key_type": 2}, {"key": "w", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.0', 'recurrent_block', 'rg_lru', 'a_param')": {"key_metadata": [{"key": "blocks.0", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "a_param", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.0', 'recurrent_block', 'rg_lru', 'input_gate', 'b')": {"key_metadata": [{"key": "blocks.0", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "input_gate", "key_type": 2}, {"key": "b", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.0', 'recurrent_block', 'rg_lru', 'input_gate', 'w')": {"key_metadata": [{"key": "blocks.0", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "input_gate", "key_type": 2}, {"key": "w", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.0', 'temporal_pre_norm', 'scale')": {"key_metadata": [{"key": "blocks.0", "key_type": 2}, {"key": "temporal_pre_norm", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.1', 'channel_pre_norm', 'scale')": {"key_metadata": [{"key": "blocks.1", "key_type": 2}, {"key": "channel_pre_norm", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.1', 'mlp_block', 'ffw_down', 'bias')": {"key_metadata": [{"key": "blocks.1", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_down", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.1', 'mlp_block', 'ffw_down', 'kernel')": {"key_metadata": [{"key": "blocks.1", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_down", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.1', 'mlp_block', 'ffw_up', 'b')": {"key_metadata": [{"key": "blocks.1", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_up", "key_type": 2}, {"key": "b", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.1', 'mlp_block', 'ffw_up', 'w')": {"key_metadata": [{"key": "blocks.1", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_up", "key_type": 2}, {"key": "w", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.1', 'recurrent_block', 'conv_1d', 'b')": {"key_metadata": [{"key": "blocks.1", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "conv_1d", "key_type": 2}, {"key": "b", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.1', 'recurrent_block', 'conv_1d', 'w')": {"key_metadata": [{"key": "blocks.1", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "conv_1d", "key_type": 2}, {"key": "w", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.1', 'recurrent_block', 'linear_out', 'bias')": {"key_metadata": [{"key": "blocks.1", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_out", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.1', 'recurrent_block', 'linear_out', 'kernel')": {"key_metadata": [{"key": "blocks.1", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_out", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.1', 'recurrent_block', 'linear_x', 'bias')": {"key_metadata": [{"key": "blocks.1", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_x", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.1', 'recurrent_block', 'linear_x', 'kernel')": {"key_metadata": [{"key": "blocks.1", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_x", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.1', 'recurrent_block', 'linear_y', 'bias')": {"key_metadata": [{"key": "blocks.1", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_y", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.1', 'recurrent_block', 'linear_y', 'kernel')": {"key_metadata": [{"key": "blocks.1", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_y", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.1', 'recurrent_block', 'rg_lru', 'a_gate', 'b')": {"key_metadata": [{"key": "blocks.1", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "a_gate", "key_type": 2}, {"key": "b", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.1', 'recurrent_block', 'rg_lru', 'a_gate', 'w')": {"key_metadata": [{"key": "blocks.1", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "a_gate", "key_type": 2}, {"key": "w", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.1', 'recurrent_block', 'rg_lru', 'a_param')": {"key_metadata": [{"key": "blocks.1", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "a_param", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.1', 'recurrent_block', 'rg_lru', 'input_gate', 'b')": {"key_metadata": [{"key": "blocks.1", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "input_gate", "key_type": 2}, {"key": "b", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.1', 'recurrent_block', 'rg_lru', 'input_gate', 'w')": {"key_metadata": [{"key": "blocks.1", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "input_gate", "key_type": 2}, {"key": "w", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.1', 'temporal_pre_norm', 'scale')": {"key_metadata": [{"key": "blocks.1", "key_type": 2}, {"key": "temporal_pre_norm", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.10', 'channel_pre_norm', 'scale')": {"key_metadata": [{"key": "blocks.10", "key_type": 2}, {"key": "channel_pre_norm", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.10', 'mlp_block', 'ffw_down', 'bias')": {"key_metadata": [{"key": "blocks.10", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_down", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.10', 'mlp_block', 'ffw_down', 'kernel')": {"key_metadata": [{"key": "blocks.10", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_down", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.10', 'mlp_block', 'ffw_up', 'b')": {"key_metadata": [{"key": "blocks.10", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_up", "key_type": 2}, {"key": "b", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.10', 'mlp_block', 'ffw_up', 'w')": {"key_metadata": [{"key": "blocks.10", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_up", "key_type": 2}, {"key": "w", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.10', 'recurrent_block', 'conv_1d', 'b')": {"key_metadata": [{"key": "blocks.10", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "conv_1d", "key_type": 2}, {"key": "b", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.10', 'recurrent_block', 'conv_1d', 'w')": {"key_metadata": [{"key": "blocks.10", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "conv_1d", "key_type": 2}, {"key": "w", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.10', 'recurrent_block', 'linear_out', 'bias')": {"key_metadata": [{"key": "blocks.10", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_out", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.10', 'recurrent_block', 'linear_out', 'kernel')": {"key_metadata": [{"key": "blocks.10", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_out", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.10', 'recurrent_block', 'linear_x', 'bias')": {"key_metadata": [{"key": "blocks.10", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_x", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.10', 'recurrent_block', 'linear_x', 'kernel')": {"key_metadata": [{"key": "blocks.10", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_x", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.10', 'recurrent_block', 'linear_y', 'bias')": {"key_metadata": [{"key": "blocks.10", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_y", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.10', 'recurrent_block', 'linear_y', 'kernel')": {"key_metadata": [{"key": "blocks.10", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_y", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.10', 'recurrent_block', 'rg_lru', 'a_gate', 'b')": {"key_metadata": [{"key": "blocks.10", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "a_gate", "key_type": 2}, {"key": "b", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.10', 'recurrent_block', 'rg_lru', 'a_gate', 'w')": {"key_metadata": [{"key": "blocks.10", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "a_gate", "key_type": 2}, {"key": "w", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.10', 'recurrent_block', 'rg_lru', 'a_param')": {"key_metadata": [{"key": "blocks.10", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "a_param", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.10', 'recurrent_block', 'rg_lru', 'input_gate', 'b')": {"key_metadata": [{"key": "blocks.10", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "input_gate", "key_type": 2}, {"key": "b", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.10', 'recurrent_block', 'rg_lru', 'input_gate', 'w')": {"key_metadata": [{"key": "blocks.10", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "input_gate", "key_type": 2}, {"key": "w", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.10', 'temporal_pre_norm', 'scale')": {"key_metadata": [{"key": "blocks.10", "key_type": 2}, {"key": "temporal_pre_norm", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.11', 'attention_block', 'proj_final', 'bias')": {"key_metadata": [{"key": "blocks.11", "key_type": 2}, {"key": "attention_block", "key_type": 2}, {"key": "proj_final", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.11', 'attention_block', 'proj_final', 'kernel')": {"key_metadata": [{"key": "blocks.11", "key_type": 2}, {"key": "attention_block", "key_type": 2}, {"key": "proj_final", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.11', 'attention_block', 'proj_k', 'kernel')": {"key_metadata": [{"key": "blocks.11", "key_type": 2}, {"key": "attention_block", "key_type": 2}, {"key": "proj_k", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.11', 'attention_block', 'proj_q', 'kernel')": {"key_metadata": [{"key": "blocks.11", "key_type": 2}, {"key": "attention_block", "key_type": 2}, {"key": "proj_q", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.11', 'attention_block', 'proj_v', 'kernel')": {"key_metadata": [{"key": "blocks.11", "key_type": 2}, {"key": "attention_block", "key_type": 2}, {"key": "proj_v", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.11', 'channel_pre_norm', 'scale')": {"key_metadata": [{"key": "blocks.11", "key_type": 2}, {"key": "channel_pre_norm", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.11', 'mlp_block', 'ffw_down', 'bias')": {"key_metadata": [{"key": "blocks.11", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_down", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.11', 'mlp_block', 'ffw_down', 'kernel')": {"key_metadata": [{"key": "blocks.11", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_down", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.11', 'mlp_block', 'ffw_up', 'b')": {"key_metadata": [{"key": "blocks.11", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_up", "key_type": 2}, {"key": "b", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.11', 'mlp_block', 'ffw_up', 'w')": {"key_metadata": [{"key": "blocks.11", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_up", "key_type": 2}, {"key": "w", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.11', 'temporal_pre_norm', 'scale')": {"key_metadata": [{"key": "blocks.11", "key_type": 2}, {"key": "temporal_pre_norm", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.12', 'channel_pre_norm', 'scale')": {"key_metadata": [{"key": "blocks.12", "key_type": 2}, {"key": "channel_pre_norm", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.12', 'mlp_block', 'ffw_down', 'bias')": {"key_metadata": [{"key": "blocks.12", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_down", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.12', 'mlp_block', 'ffw_down', 'kernel')": {"key_metadata": [{"key": "blocks.12", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_down", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.12', 'mlp_block', 'ffw_up', 'b')": {"key_metadata": [{"key": "blocks.12", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_up", "key_type": 2}, {"key": "b", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.12', 'mlp_block', 'ffw_up', 'w')": {"key_metadata": [{"key": "blocks.12", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_up", "key_type": 2}, {"key": "w", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.12', 'recurrent_block', 'conv_1d', 'b')": {"key_metadata": [{"key": "blocks.12", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "conv_1d", "key_type": 2}, {"key": "b", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.12', 'recurrent_block', 'conv_1d', 'w')": {"key_metadata": [{"key": "blocks.12", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "conv_1d", "key_type": 2}, {"key": "w", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.12', 'recurrent_block', 'linear_out', 'bias')": {"key_metadata": [{"key": "blocks.12", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_out", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.12', 'recurrent_block', 'linear_out', 'kernel')": {"key_metadata": [{"key": "blocks.12", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_out", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.12', 'recurrent_block', 'linear_x', 'bias')": {"key_metadata": [{"key": "blocks.12", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_x", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.12', 'recurrent_block', 'linear_x', 'kernel')": {"key_metadata": [{"key": "blocks.12", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_x", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.12', 'recurrent_block', 'linear_y', 'bias')": {"key_metadata": [{"key": "blocks.12", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_y", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.12', 'recurrent_block', 'linear_y', 'kernel')": {"key_metadata": [{"key": "blocks.12", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_y", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.12', 'recurrent_block', 'rg_lru', 'a_gate', 'b')": {"key_metadata": [{"key": "blocks.12", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "a_gate", "key_type": 2}, {"key": "b", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.12', 'recurrent_block', 'rg_lru', 'a_gate', 'w')": {"key_metadata": [{"key": "blocks.12", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "a_gate", "key_type": 2}, {"key": "w", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.12', 'recurrent_block', 'rg_lru', 'a_param')": {"key_metadata": [{"key": "blocks.12", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "a_param", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.12', 'recurrent_block', 'rg_lru', 'input_gate', 'b')": {"key_metadata": [{"key": "blocks.12", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "input_gate", "key_type": 2}, {"key": "b", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.12', 'recurrent_block', 'rg_lru', 'input_gate', 'w')": {"key_metadata": [{"key": "blocks.12", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "input_gate", "key_type": 2}, {"key": "w", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.12', 'temporal_pre_norm', 'scale')": {"key_metadata": [{"key": "blocks.12", "key_type": 2}, {"key": "temporal_pre_norm", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.13', 'channel_pre_norm', 'scale')": {"key_metadata": [{"key": "blocks.13", "key_type": 2}, {"key": "channel_pre_norm", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.13', 'mlp_block', 'ffw_down', 'bias')": {"key_metadata": [{"key": "blocks.13", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_down", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.13', 'mlp_block', 'ffw_down', 'kernel')": {"key_metadata": [{"key": "blocks.13", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_down", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.13', 'mlp_block', 'ffw_up', 'b')": {"key_metadata": [{"key": "blocks.13", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_up", "key_type": 2}, {"key": "b", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.13', 'mlp_block', 'ffw_up', 'w')": {"key_metadata": [{"key": "blocks.13", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_up", "key_type": 2}, {"key": "w", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.13', 'recurrent_block', 'conv_1d', 'b')": {"key_metadata": [{"key": "blocks.13", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "conv_1d", "key_type": 2}, {"key": "b", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.13', 'recurrent_block', 'conv_1d', 'w')": {"key_metadata": [{"key": "blocks.13", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "conv_1d", "key_type": 2}, {"key": "w", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.13', 'recurrent_block', 'linear_out', 'bias')": {"key_metadata": [{"key": "blocks.13", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_out", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.13', 'recurrent_block', 'linear_out', 'kernel')": {"key_metadata": [{"key": "blocks.13", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_out", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.13', 'recurrent_block', 'linear_x', 'bias')": {"key_metadata": [{"key": "blocks.13", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_x", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.13', 'recurrent_block', 'linear_x', 'kernel')": {"key_metadata": [{"key": "blocks.13", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_x", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.13', 'recurrent_block', 'linear_y', 'bias')": {"key_metadata": [{"key": "blocks.13", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_y", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.13', 'recurrent_block', 'linear_y', 'kernel')": {"key_metadata": [{"key": "blocks.13", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_y", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.13', 'recurrent_block', 'rg_lru', 'a_gate', 'b')": {"key_metadata": [{"key": "blocks.13", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "a_gate", "key_type": 2}, {"key": "b", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.13', 'recurrent_block', 'rg_lru', 'a_gate', 'w')": {"key_metadata": [{"key": "blocks.13", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "a_gate", "key_type": 2}, {"key": "w", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.13', 'recurrent_block', 'rg_lru', 'a_param')": {"key_metadata": [{"key": "blocks.13", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "a_param", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.13', 'recurrent_block', 'rg_lru', 'input_gate', 'b')": {"key_metadata": [{"key": "blocks.13", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "input_gate", "key_type": 2}, {"key": "b", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.13', 'recurrent_block', 'rg_lru', 'input_gate', 'w')": {"key_metadata": [{"key": "blocks.13", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "input_gate", "key_type": 2}, {"key": "w", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.13', 'temporal_pre_norm', 'scale')": {"key_metadata": [{"key": "blocks.13", "key_type": 2}, {"key": "temporal_pre_norm", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.14', 'attention_block', 'proj_final', 'bias')": {"key_metadata": [{"key": "blocks.14", "key_type": 2}, {"key": "attention_block", "key_type": 2}, {"key": "proj_final", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.14', 'attention_block', 'proj_final', 'kernel')": {"key_metadata": [{"key": "blocks.14", "key_type": 2}, {"key": "attention_block", "key_type": 2}, {"key": "proj_final", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.14', 'attention_block', 'proj_k', 'kernel')": {"key_metadata": [{"key": "blocks.14", "key_type": 2}, {"key": "attention_block", "key_type": 2}, {"key": "proj_k", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.14', 'attention_block', 'proj_q', 'kernel')": {"key_metadata": [{"key": "blocks.14", "key_type": 2}, {"key": "attention_block", "key_type": 2}, {"key": "proj_q", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.14', 'attention_block', 'proj_v', 'kernel')": {"key_metadata": [{"key": "blocks.14", "key_type": 2}, {"key": "attention_block", "key_type": 2}, {"key": "proj_v", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.14', 'channel_pre_norm', 'scale')": {"key_metadata": [{"key": "blocks.14", "key_type": 2}, {"key": "channel_pre_norm", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.14', 'mlp_block', 'ffw_down', 'bias')": {"key_metadata": [{"key": "blocks.14", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_down", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.14', 'mlp_block', 'ffw_down', 'kernel')": {"key_metadata": [{"key": "blocks.14", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_down", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.14', 'mlp_block', 'ffw_up', 'b')": {"key_metadata": [{"key": "blocks.14", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_up", "key_type": 2}, {"key": "b", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.14', 'mlp_block', 'ffw_up', 'w')": {"key_metadata": [{"key": "blocks.14", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_up", "key_type": 2}, {"key": "w", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.14', 'temporal_pre_norm', 'scale')": {"key_metadata": [{"key": "blocks.14", "key_type": 2}, {"key": "temporal_pre_norm", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.15', 'channel_pre_norm', 'scale')": {"key_metadata": [{"key": "blocks.15", "key_type": 2}, {"key": "channel_pre_norm", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.15', 'mlp_block', 'ffw_down', 'bias')": {"key_metadata": [{"key": "blocks.15", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_down", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.15', 'mlp_block', 'ffw_down', 'kernel')": {"key_metadata": [{"key": "blocks.15", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_down", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.15', 'mlp_block', 'ffw_up', 'b')": {"key_metadata": [{"key": "blocks.15", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_up", "key_type": 2}, {"key": "b", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.15', 'mlp_block', 'ffw_up', 'w')": {"key_metadata": [{"key": "blocks.15", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_up", "key_type": 2}, {"key": "w", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.15', 'recurrent_block', 'conv_1d', 'b')": {"key_metadata": [{"key": "blocks.15", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "conv_1d", "key_type": 2}, {"key": "b", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.15', 'recurrent_block', 'conv_1d', 'w')": {"key_metadata": [{"key": "blocks.15", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "conv_1d", "key_type": 2}, {"key": "w", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.15', 'recurrent_block', 'linear_out', 'bias')": {"key_metadata": [{"key": "blocks.15", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_out", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.15', 'recurrent_block', 'linear_out', 'kernel')": {"key_metadata": [{"key": "blocks.15", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_out", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.15', 'recurrent_block', 'linear_x', 'bias')": {"key_metadata": [{"key": "blocks.15", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_x", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.15', 'recurrent_block', 'linear_x', 'kernel')": {"key_metadata": [{"key": "blocks.15", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_x", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.15', 'recurrent_block', 'linear_y', 'bias')": {"key_metadata": [{"key": "blocks.15", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_y", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.15', 'recurrent_block', 'linear_y', 'kernel')": {"key_metadata": [{"key": "blocks.15", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_y", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.15', 'recurrent_block', 'rg_lru', 'a_gate', 'b')": {"key_metadata": [{"key": "blocks.15", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "a_gate", "key_type": 2}, {"key": "b", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.15', 'recurrent_block', 'rg_lru', 'a_gate', 'w')": {"key_metadata": [{"key": "blocks.15", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "a_gate", "key_type": 2}, {"key": "w", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.15', 'recurrent_block', 'rg_lru', 'a_param')": {"key_metadata": [{"key": "blocks.15", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "a_param", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.15', 'recurrent_block', 'rg_lru', 'input_gate', 'b')": {"key_metadata": [{"key": "blocks.15", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "input_gate", "key_type": 2}, {"key": "b", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.15', 'recurrent_block', 'rg_lru', 'input_gate', 'w')": {"key_metadata": [{"key": "blocks.15", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "input_gate", "key_type": 2}, {"key": "w", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.15', 'temporal_pre_norm', 'scale')": {"key_metadata": [{"key": "blocks.15", "key_type": 2}, {"key": "temporal_pre_norm", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.16', 'channel_pre_norm', 'scale')": {"key_metadata": [{"key": "blocks.16", "key_type": 2}, {"key": "channel_pre_norm", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.16', 'mlp_block', 'ffw_down', 'bias')": {"key_metadata": [{"key": "blocks.16", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_down", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.16', 'mlp_block', 'ffw_down', 'kernel')": {"key_metadata": [{"key": "blocks.16", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_down", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.16', 'mlp_block', 'ffw_up', 'b')": {"key_metadata": [{"key": "blocks.16", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_up", "key_type": 2}, {"key": "b", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.16', 'mlp_block', 'ffw_up', 'w')": {"key_metadata": [{"key": "blocks.16", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_up", "key_type": 2}, {"key": "w", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.16', 'recurrent_block', 'conv_1d', 'b')": {"key_metadata": [{"key": "blocks.16", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "conv_1d", "key_type": 2}, {"key": "b", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.16', 'recurrent_block', 'conv_1d', 'w')": {"key_metadata": [{"key": "blocks.16", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "conv_1d", "key_type": 2}, {"key": "w", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.16', 'recurrent_block', 'linear_out', 'bias')": {"key_metadata": [{"key": "blocks.16", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_out", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.16', 'recurrent_block', 'linear_out', 'kernel')": {"key_metadata": [{"key": "blocks.16", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_out", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.16', 'recurrent_block', 'linear_x', 'bias')": {"key_metadata": [{"key": "blocks.16", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_x", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.16', 'recurrent_block', 'linear_x', 'kernel')": {"key_metadata": [{"key": "blocks.16", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_x", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.16', 'recurrent_block', 'linear_y', 'bias')": {"key_metadata": [{"key": "blocks.16", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_y", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.16', 'recurrent_block', 'linear_y', 'kernel')": {"key_metadata": [{"key": "blocks.16", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_y", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.16', 'recurrent_block', 'rg_lru', 'a_gate', 'b')": {"key_metadata": [{"key": "blocks.16", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "a_gate", "key_type": 2}, {"key": "b", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.16', 'recurrent_block', 'rg_lru', 'a_gate', 'w')": {"key_metadata": [{"key": "blocks.16", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "a_gate", "key_type": 2}, {"key": "w", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.16', 'recurrent_block', 'rg_lru', 'a_param')": {"key_metadata": [{"key": "blocks.16", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "a_param", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.16', 'recurrent_block', 'rg_lru', 'input_gate', 'b')": {"key_metadata": [{"key": "blocks.16", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "input_gate", "key_type": 2}, {"key": "b", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.16', 'recurrent_block', 'rg_lru', 'input_gate', 'w')": {"key_metadata": [{"key": "blocks.16", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "input_gate", "key_type": 2}, {"key": "w", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.16', 'temporal_pre_norm', 'scale')": {"key_metadata": [{"key": "blocks.16", "key_type": 2}, {"key": "temporal_pre_norm", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.17', 'attention_block', 'proj_final', 'bias')": {"key_metadata": [{"key": "blocks.17", "key_type": 2}, {"key": "attention_block", "key_type": 2}, {"key": "proj_final", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.17', 'attention_block', 'proj_final', 'kernel')": {"key_metadata": [{"key": "blocks.17", "key_type": 2}, {"key": "attention_block", "key_type": 2}, {"key": "proj_final", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.17', 'attention_block', 'proj_k', 'kernel')": {"key_metadata": [{"key": "blocks.17", "key_type": 2}, {"key": "attention_block", "key_type": 2}, {"key": "proj_k", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.17', 'attention_block', 'proj_q', 'kernel')": {"key_metadata": [{"key": "blocks.17", "key_type": 2}, {"key": "attention_block", "key_type": 2}, {"key": "proj_q", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.17', 'attention_block', 'proj_v', 'kernel')": {"key_metadata": [{"key": "blocks.17", "key_type": 2}, {"key": "attention_block", "key_type": 2}, {"key": "proj_v", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.17', 'channel_pre_norm', 'scale')": {"key_metadata": [{"key": "blocks.17", "key_type": 2}, {"key": "channel_pre_norm", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.17', 'mlp_block', 'ffw_down', 'bias')": {"key_metadata": [{"key": "blocks.17", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_down", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.17', 'mlp_block', 'ffw_down', 'kernel')": {"key_metadata": [{"key": "blocks.17", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_down", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.17', 'mlp_block', 'ffw_up', 'b')": {"key_metadata": [{"key": "blocks.17", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_up", "key_type": 2}, {"key": "b", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.17', 'mlp_block', 'ffw_up', 'w')": {"key_metadata": [{"key": "blocks.17", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_up", "key_type": 2}, {"key": "w", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.17', 'temporal_pre_norm', 'scale')": {"key_metadata": [{"key": "blocks.17", "key_type": 2}, {"key": "temporal_pre_norm", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.18', 'channel_pre_norm', 'scale')": {"key_metadata": [{"key": "blocks.18", "key_type": 2}, {"key": "channel_pre_norm", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.18', 'mlp_block', 'ffw_down', 'bias')": {"key_metadata": [{"key": "blocks.18", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_down", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.18', 'mlp_block', 'ffw_down', 'kernel')": {"key_metadata": [{"key": "blocks.18", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_down", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.18', 'mlp_block', 'ffw_up', 'b')": {"key_metadata": [{"key": "blocks.18", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_up", "key_type": 2}, {"key": "b", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.18', 'mlp_block', 'ffw_up', 'w')": {"key_metadata": [{"key": "blocks.18", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_up", "key_type": 2}, {"key": "w", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.18', 'recurrent_block', 'conv_1d', 'b')": {"key_metadata": [{"key": "blocks.18", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "conv_1d", "key_type": 2}, {"key": "b", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.18', 'recurrent_block', 'conv_1d', 'w')": {"key_metadata": [{"key": "blocks.18", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "conv_1d", "key_type": 2}, {"key": "w", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.18', 'recurrent_block', 'linear_out', 'bias')": {"key_metadata": [{"key": "blocks.18", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_out", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.18', 'recurrent_block', 'linear_out', 'kernel')": {"key_metadata": [{"key": "blocks.18", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_out", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.18', 'recurrent_block', 'linear_x', 'bias')": {"key_metadata": [{"key": "blocks.18", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_x", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.18', 'recurrent_block', 'linear_x', 'kernel')": {"key_metadata": [{"key": "blocks.18", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_x", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.18', 'recurrent_block', 'linear_y', 'bias')": {"key_metadata": [{"key": "blocks.18", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_y", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.18', 'recurrent_block', 'linear_y', 'kernel')": {"key_metadata": [{"key": "blocks.18", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_y", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.18', 'recurrent_block', 'rg_lru', 'a_gate', 'b')": {"key_metadata": [{"key": "blocks.18", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "a_gate", "key_type": 2}, {"key": "b", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.18', 'recurrent_block', 'rg_lru', 'a_gate', 'w')": {"key_metadata": [{"key": "blocks.18", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "a_gate", "key_type": 2}, {"key": "w", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.18', 'recurrent_block', 'rg_lru', 'a_param')": {"key_metadata": [{"key": "blocks.18", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "a_param", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.18', 'recurrent_block', 'rg_lru', 'input_gate', 'b')": {"key_metadata": [{"key": "blocks.18", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "input_gate", "key_type": 2}, {"key": "b", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.18', 'recurrent_block', 'rg_lru', 'input_gate', 'w')": {"key_metadata": [{"key": "blocks.18", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "input_gate", "key_type": 2}, {"key": "w", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.18', 'temporal_pre_norm', 'scale')": {"key_metadata": [{"key": "blocks.18", "key_type": 2}, {"key": "temporal_pre_norm", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.19', 'channel_pre_norm', 'scale')": {"key_metadata": [{"key": "blocks.19", "key_type": 2}, {"key": "channel_pre_norm", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.19', 'mlp_block', 'ffw_down', 'bias')": {"key_metadata": [{"key": "blocks.19", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_down", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.19', 'mlp_block', 'ffw_down', 'kernel')": {"key_metadata": [{"key": "blocks.19", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_down", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.19', 'mlp_block', 'ffw_up', 'b')": {"key_metadata": [{"key": "blocks.19", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_up", "key_type": 2}, {"key": "b", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.19', 'mlp_block', 'ffw_up', 'w')": {"key_metadata": [{"key": "blocks.19", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_up", "key_type": 2}, {"key": "w", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.19', 'recurrent_block', 'conv_1d', 'b')": {"key_metadata": [{"key": "blocks.19", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "conv_1d", "key_type": 2}, {"key": "b", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.19', 'recurrent_block', 'conv_1d', 'w')": {"key_metadata": [{"key": "blocks.19", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "conv_1d", "key_type": 2}, {"key": "w", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.19', 'recurrent_block', 'linear_out', 'bias')": {"key_metadata": [{"key": "blocks.19", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_out", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.19', 'recurrent_block', 'linear_out', 'kernel')": {"key_metadata": [{"key": "blocks.19", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_out", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.19', 'recurrent_block', 'linear_x', 'bias')": {"key_metadata": [{"key": "blocks.19", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_x", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.19', 'recurrent_block', 'linear_x', 'kernel')": {"key_metadata": [{"key": "blocks.19", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_x", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.19', 'recurrent_block', 'linear_y', 'bias')": {"key_metadata": [{"key": "blocks.19", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_y", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.19', 'recurrent_block', 'linear_y', 'kernel')": {"key_metadata": [{"key": "blocks.19", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_y", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.19', 'recurrent_block', 'rg_lru', 'a_gate', 'b')": {"key_metadata": [{"key": "blocks.19", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "a_gate", "key_type": 2}, {"key": "b", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.19', 'recurrent_block', 'rg_lru', 'a_gate', 'w')": {"key_metadata": [{"key": "blocks.19", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "a_gate", "key_type": 2}, {"key": "w", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.19', 'recurrent_block', 'rg_lru', 'a_param')": {"key_metadata": [{"key": "blocks.19", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "a_param", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.19', 'recurrent_block', 'rg_lru', 'input_gate', 'b')": {"key_metadata": [{"key": "blocks.19", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "input_gate", "key_type": 2}, {"key": "b", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.19', 'recurrent_block', 'rg_lru', 'input_gate', 'w')": {"key_metadata": [{"key": "blocks.19", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "input_gate", "key_type": 2}, {"key": "w", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.19', 'temporal_pre_norm', 'scale')": {"key_metadata": [{"key": "blocks.19", "key_type": 2}, {"key": "temporal_pre_norm", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.2', 'attention_block', 'proj_final', 'bias')": {"key_metadata": [{"key": "blocks.2", "key_type": 2}, {"key": "attention_block", "key_type": 2}, {"key": "proj_final", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.2', 'attention_block', 'proj_final', 'kernel')": {"key_metadata": [{"key": "blocks.2", "key_type": 2}, {"key": "attention_block", "key_type": 2}, {"key": "proj_final", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.2', 'attention_block', 'proj_k', 'kernel')": {"key_metadata": [{"key": "blocks.2", "key_type": 2}, {"key": "attention_block", "key_type": 2}, {"key": "proj_k", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.2', 'attention_block', 'proj_q', 'kernel')": {"key_metadata": [{"key": "blocks.2", "key_type": 2}, {"key": "attention_block", "key_type": 2}, {"key": "proj_q", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.2', 'attention_block', 'proj_v', 'kernel')": {"key_metadata": [{"key": "blocks.2", "key_type": 2}, {"key": "attention_block", "key_type": 2}, {"key": "proj_v", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.2', 'channel_pre_norm', 'scale')": {"key_metadata": [{"key": "blocks.2", "key_type": 2}, {"key": "channel_pre_norm", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.2', 'mlp_block', 'ffw_down', 'bias')": {"key_metadata": [{"key": "blocks.2", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_down", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.2', 'mlp_block', 'ffw_down', 'kernel')": {"key_metadata": [{"key": "blocks.2", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_down", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.2', 'mlp_block', 'ffw_up', 'b')": {"key_metadata": [{"key": "blocks.2", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_up", "key_type": 2}, {"key": "b", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.2', 'mlp_block', 'ffw_up', 'w')": {"key_metadata": [{"key": "blocks.2", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_up", "key_type": 2}, {"key": "w", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.2', 'temporal_pre_norm', 'scale')": {"key_metadata": [{"key": "blocks.2", "key_type": 2}, {"key": "temporal_pre_norm", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.20', 'attention_block', 'proj_final', 'bias')": {"key_metadata": [{"key": "blocks.20", "key_type": 2}, {"key": "attention_block", "key_type": 2}, {"key": "proj_final", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.20', 'attention_block', 'proj_final', 'kernel')": {"key_metadata": [{"key": "blocks.20", "key_type": 2}, {"key": "attention_block", "key_type": 2}, {"key": "proj_final", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.20', 'attention_block', 'proj_k', 'kernel')": {"key_metadata": [{"key": "blocks.20", "key_type": 2}, {"key": "attention_block", "key_type": 2}, {"key": "proj_k", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.20', 'attention_block', 'proj_q', 'kernel')": {"key_metadata": [{"key": "blocks.20", "key_type": 2}, {"key": "attention_block", "key_type": 2}, {"key": "proj_q", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.20', 'attention_block', 'proj_v', 'kernel')": {"key_metadata": [{"key": "blocks.20", "key_type": 2}, {"key": "attention_block", "key_type": 2}, {"key": "proj_v", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.20', 'channel_pre_norm', 'scale')": {"key_metadata": [{"key": "blocks.20", "key_type": 2}, {"key": "channel_pre_norm", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.20', 'mlp_block', 'ffw_down', 'bias')": {"key_metadata": [{"key": "blocks.20", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_down", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.20', 'mlp_block', 'ffw_down', 'kernel')": {"key_metadata": [{"key": "blocks.20", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_down", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.20', 'mlp_block', 'ffw_up', 'b')": {"key_metadata": [{"key": "blocks.20", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_up", "key_type": 2}, {"key": "b", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.20', 'mlp_block', 'ffw_up', 'w')": {"key_metadata": [{"key": "blocks.20", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_up", "key_type": 2}, {"key": "w", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.20', 'temporal_pre_norm', 'scale')": {"key_metadata": [{"key": "blocks.20", "key_type": 2}, {"key": "temporal_pre_norm", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.21', 'channel_pre_norm', 'scale')": {"key_metadata": [{"key": "blocks.21", "key_type": 2}, {"key": "channel_pre_norm", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.21', 'mlp_block', 'ffw_down', 'bias')": {"key_metadata": [{"key": "blocks.21", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_down", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.21', 'mlp_block', 'ffw_down', 'kernel')": {"key_metadata": [{"key": "blocks.21", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_down", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.21', 'mlp_block', 'ffw_up', 'b')": {"key_metadata": [{"key": "blocks.21", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_up", "key_type": 2}, {"key": "b", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.21', 'mlp_block', 'ffw_up', 'w')": {"key_metadata": [{"key": "blocks.21", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_up", "key_type": 2}, {"key": "w", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.21', 'recurrent_block', 'conv_1d', 'b')": {"key_metadata": [{"key": "blocks.21", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "conv_1d", "key_type": 2}, {"key": "b", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.21', 'recurrent_block', 'conv_1d', 'w')": {"key_metadata": [{"key": "blocks.21", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "conv_1d", "key_type": 2}, {"key": "w", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.21', 'recurrent_block', 'linear_out', 'bias')": {"key_metadata": [{"key": "blocks.21", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_out", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.21', 'recurrent_block', 'linear_out', 'kernel')": {"key_metadata": [{"key": "blocks.21", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_out", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.21', 'recurrent_block', 'linear_x', 'bias')": {"key_metadata": [{"key": "blocks.21", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_x", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.21', 'recurrent_block', 'linear_x', 'kernel')": {"key_metadata": [{"key": "blocks.21", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_x", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.21', 'recurrent_block', 'linear_y', 'bias')": {"key_metadata": [{"key": "blocks.21", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_y", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.21', 'recurrent_block', 'linear_y', 'kernel')": {"key_metadata": [{"key": "blocks.21", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_y", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.21', 'recurrent_block', 'rg_lru', 'a_gate', 'b')": {"key_metadata": [{"key": "blocks.21", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "a_gate", "key_type": 2}, {"key": "b", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.21', 'recurrent_block', 'rg_lru', 'a_gate', 'w')": {"key_metadata": [{"key": "blocks.21", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "a_gate", "key_type": 2}, {"key": "w", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.21', 'recurrent_block', 'rg_lru', 'a_param')": {"key_metadata": [{"key": "blocks.21", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "a_param", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.21', 'recurrent_block', 'rg_lru', 'input_gate', 'b')": {"key_metadata": [{"key": "blocks.21", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "input_gate", "key_type": 2}, {"key": "b", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.21', 'recurrent_block', 'rg_lru', 'input_gate', 'w')": {"key_metadata": [{"key": "blocks.21", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "input_gate", "key_type": 2}, {"key": "w", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.21', 'temporal_pre_norm', 'scale')": {"key_metadata": [{"key": "blocks.21", "key_type": 2}, {"key": "temporal_pre_norm", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.22', 'channel_pre_norm', 'scale')": {"key_metadata": [{"key": "blocks.22", "key_type": 2}, {"key": "channel_pre_norm", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.22', 'mlp_block', 'ffw_down', 'bias')": {"key_metadata": [{"key": "blocks.22", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_down", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.22', 'mlp_block', 'ffw_down', 'kernel')": {"key_metadata": [{"key": "blocks.22", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_down", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.22', 'mlp_block', 'ffw_up', 'b')": {"key_metadata": [{"key": "blocks.22", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_up", "key_type": 2}, {"key": "b", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.22', 'mlp_block', 'ffw_up', 'w')": {"key_metadata": [{"key": "blocks.22", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_up", "key_type": 2}, {"key": "w", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.22', 'recurrent_block', 'conv_1d', 'b')": {"key_metadata": [{"key": "blocks.22", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "conv_1d", "key_type": 2}, {"key": "b", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.22', 'recurrent_block', 'conv_1d', 'w')": {"key_metadata": [{"key": "blocks.22", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "conv_1d", "key_type": 2}, {"key": "w", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.22', 'recurrent_block', 'linear_out', 'bias')": {"key_metadata": [{"key": "blocks.22", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_out", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.22', 'recurrent_block', 'linear_out', 'kernel')": {"key_metadata": [{"key": "blocks.22", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_out", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.22', 'recurrent_block', 'linear_x', 'bias')": {"key_metadata": [{"key": "blocks.22", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_x", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.22', 'recurrent_block', 'linear_x', 'kernel')": {"key_metadata": [{"key": "blocks.22", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_x", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.22', 'recurrent_block', 'linear_y', 'bias')": {"key_metadata": [{"key": "blocks.22", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_y", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.22', 'recurrent_block', 'linear_y', 'kernel')": {"key_metadata": [{"key": "blocks.22", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_y", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.22', 'recurrent_block', 'rg_lru', 'a_gate', 'b')": {"key_metadata": [{"key": "blocks.22", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "a_gate", "key_type": 2}, {"key": "b", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.22', 'recurrent_block', 'rg_lru', 'a_gate', 'w')": {"key_metadata": [{"key": "blocks.22", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "a_gate", "key_type": 2}, {"key": "w", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.22', 'recurrent_block', 'rg_lru', 'a_param')": {"key_metadata": [{"key": "blocks.22", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "a_param", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.22', 'recurrent_block', 'rg_lru', 'input_gate', 'b')": {"key_metadata": [{"key": "blocks.22", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "input_gate", "key_type": 2}, {"key": "b", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.22', 'recurrent_block', 'rg_lru', 'input_gate', 'w')": {"key_metadata": [{"key": "blocks.22", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "input_gate", "key_type": 2}, {"key": "w", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.22', 'temporal_pre_norm', 'scale')": {"key_metadata": [{"key": "blocks.22", "key_type": 2}, {"key": "temporal_pre_norm", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.23', 'attention_block', 'proj_final', 'bias')": {"key_metadata": [{"key": "blocks.23", "key_type": 2}, {"key": "attention_block", "key_type": 2}, {"key": "proj_final", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.23', 'attention_block', 'proj_final', 'kernel')": {"key_metadata": [{"key": "blocks.23", "key_type": 2}, {"key": "attention_block", "key_type": 2}, {"key": "proj_final", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.23', 'attention_block', 'proj_k', 'kernel')": {"key_metadata": [{"key": "blocks.23", "key_type": 2}, {"key": "attention_block", "key_type": 2}, {"key": "proj_k", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.23', 'attention_block', 'proj_q', 'kernel')": {"key_metadata": [{"key": "blocks.23", "key_type": 2}, {"key": "attention_block", "key_type": 2}, {"key": "proj_q", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.23', 'attention_block', 'proj_v', 'kernel')": {"key_metadata": [{"key": "blocks.23", "key_type": 2}, {"key": "attention_block", "key_type": 2}, {"key": "proj_v", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.23', 'channel_pre_norm', 'scale')": {"key_metadata": [{"key": "blocks.23", "key_type": 2}, {"key": "channel_pre_norm", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.23', 'mlp_block', 'ffw_down', 'bias')": {"key_metadata": [{"key": "blocks.23", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_down", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.23', 'mlp_block', 'ffw_down', 'kernel')": {"key_metadata": [{"key": "blocks.23", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_down", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.23', 'mlp_block', 'ffw_up', 'b')": {"key_metadata": [{"key": "blocks.23", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_up", "key_type": 2}, {"key": "b", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.23', 'mlp_block', 'ffw_up', 'w')": {"key_metadata": [{"key": "blocks.23", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_up", "key_type": 2}, {"key": "w", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.23', 'temporal_pre_norm', 'scale')": {"key_metadata": [{"key": "blocks.23", "key_type": 2}, {"key": "temporal_pre_norm", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.24', 'channel_pre_norm', 'scale')": {"key_metadata": [{"key": "blocks.24", "key_type": 2}, {"key": "channel_pre_norm", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.24', 'mlp_block', 'ffw_down', 'bias')": {"key_metadata": [{"key": "blocks.24", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_down", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.24', 'mlp_block', 'ffw_down', 'kernel')": {"key_metadata": [{"key": "blocks.24", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_down", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.24', 'mlp_block', 'ffw_up', 'b')": {"key_metadata": [{"key": "blocks.24", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_up", "key_type": 2}, {"key": "b", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.24', 'mlp_block', 'ffw_up', 'w')": {"key_metadata": [{"key": "blocks.24", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_up", "key_type": 2}, {"key": "w", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.24', 'recurrent_block', 'conv_1d', 'b')": {"key_metadata": [{"key": "blocks.24", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "conv_1d", "key_type": 2}, {"key": "b", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.24', 'recurrent_block', 'conv_1d', 'w')": {"key_metadata": [{"key": "blocks.24", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "conv_1d", "key_type": 2}, {"key": "w", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.24', 'recurrent_block', 'linear_out', 'bias')": {"key_metadata": [{"key": "blocks.24", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_out", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.24', 'recurrent_block', 'linear_out', 'kernel')": {"key_metadata": [{"key": "blocks.24", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_out", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.24', 'recurrent_block', 'linear_x', 'bias')": {"key_metadata": [{"key": "blocks.24", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_x", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.24', 'recurrent_block', 'linear_x', 'kernel')": {"key_metadata": [{"key": "blocks.24", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_x", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.24', 'recurrent_block', 'linear_y', 'bias')": {"key_metadata": [{"key": "blocks.24", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_y", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.24', 'recurrent_block', 'linear_y', 'kernel')": {"key_metadata": [{"key": "blocks.24", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_y", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.24', 'recurrent_block', 'rg_lru', 'a_gate', 'b')": {"key_metadata": [{"key": "blocks.24", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "a_gate", "key_type": 2}, {"key": "b", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.24', 'recurrent_block', 'rg_lru', 'a_gate', 'w')": {"key_metadata": [{"key": "blocks.24", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "a_gate", "key_type": 2}, {"key": "w", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.24', 'recurrent_block', 'rg_lru', 'a_param')": {"key_metadata": [{"key": "blocks.24", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "a_param", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.24', 'recurrent_block', 'rg_lru', 'input_gate', 'b')": {"key_metadata": [{"key": "blocks.24", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "input_gate", "key_type": 2}, {"key": "b", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.24', 'recurrent_block', 'rg_lru', 'input_gate', 'w')": {"key_metadata": [{"key": "blocks.24", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "input_gate", "key_type": 2}, {"key": "w", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.24', 'temporal_pre_norm', 'scale')": {"key_metadata": [{"key": "blocks.24", "key_type": 2}, {"key": "temporal_pre_norm", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.25', 'channel_pre_norm', 'scale')": {"key_metadata": [{"key": "blocks.25", "key_type": 2}, {"key": "channel_pre_norm", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.25', 'mlp_block', 'ffw_down', 'bias')": {"key_metadata": [{"key": "blocks.25", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_down", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.25', 'mlp_block', 'ffw_down', 'kernel')": {"key_metadata": [{"key": "blocks.25", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_down", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.25', 'mlp_block', 'ffw_up', 'b')": {"key_metadata": [{"key": "blocks.25", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_up", "key_type": 2}, {"key": "b", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.25', 'mlp_block', 'ffw_up', 'w')": {"key_metadata": [{"key": "blocks.25", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_up", "key_type": 2}, {"key": "w", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.25', 'recurrent_block', 'conv_1d', 'b')": {"key_metadata": [{"key": "blocks.25", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "conv_1d", "key_type": 2}, {"key": "b", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.25', 'recurrent_block', 'conv_1d', 'w')": {"key_metadata": [{"key": "blocks.25", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "conv_1d", "key_type": 2}, {"key": "w", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.25', 'recurrent_block', 'linear_out', 'bias')": {"key_metadata": [{"key": "blocks.25", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_out", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.25', 'recurrent_block', 'linear_out', 'kernel')": {"key_metadata": [{"key": "blocks.25", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_out", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.25', 'recurrent_block', 'linear_x', 'bias')": {"key_metadata": [{"key": "blocks.25", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_x", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.25', 'recurrent_block', 'linear_x', 'kernel')": {"key_metadata": [{"key": "blocks.25", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_x", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.25', 'recurrent_block', 'linear_y', 'bias')": {"key_metadata": [{"key": "blocks.25", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_y", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.25', 'recurrent_block', 'linear_y', 'kernel')": {"key_metadata": [{"key": "blocks.25", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_y", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.25', 'recurrent_block', 'rg_lru', 'a_gate', 'b')": {"key_metadata": [{"key": "blocks.25", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "a_gate", "key_type": 2}, {"key": "b", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.25', 'recurrent_block', 'rg_lru', 'a_gate', 'w')": {"key_metadata": [{"key": "blocks.25", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "a_gate", "key_type": 2}, {"key": "w", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.25', 'recurrent_block', 'rg_lru', 'a_param')": {"key_metadata": [{"key": "blocks.25", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "a_param", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.25', 'recurrent_block', 'rg_lru', 'input_gate', 'b')": {"key_metadata": [{"key": "blocks.25", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "input_gate", "key_type": 2}, {"key": "b", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.25', 'recurrent_block', 'rg_lru', 'input_gate', 'w')": {"key_metadata": [{"key": "blocks.25", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "input_gate", "key_type": 2}, {"key": "w", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.25', 'temporal_pre_norm', 'scale')": {"key_metadata": [{"key": "blocks.25", "key_type": 2}, {"key": "temporal_pre_norm", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.3', 'channel_pre_norm', 'scale')": {"key_metadata": [{"key": "blocks.3", "key_type": 2}, {"key": "channel_pre_norm", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.3', 'mlp_block', 'ffw_down', 'bias')": {"key_metadata": [{"key": "blocks.3", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_down", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.3', 'mlp_block', 'ffw_down', 'kernel')": {"key_metadata": [{"key": "blocks.3", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_down", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.3', 'mlp_block', 'ffw_up', 'b')": {"key_metadata": [{"key": "blocks.3", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_up", "key_type": 2}, {"key": "b", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.3', 'mlp_block', 'ffw_up', 'w')": {"key_metadata": [{"key": "blocks.3", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_up", "key_type": 2}, {"key": "w", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.3', 'recurrent_block', 'conv_1d', 'b')": {"key_metadata": [{"key": "blocks.3", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "conv_1d", "key_type": 2}, {"key": "b", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.3', 'recurrent_block', 'conv_1d', 'w')": {"key_metadata": [{"key": "blocks.3", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "conv_1d", "key_type": 2}, {"key": "w", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.3', 'recurrent_block', 'linear_out', 'bias')": {"key_metadata": [{"key": "blocks.3", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_out", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.3', 'recurrent_block', 'linear_out', 'kernel')": {"key_metadata": [{"key": "blocks.3", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_out", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.3', 'recurrent_block', 'linear_x', 'bias')": {"key_metadata": [{"key": "blocks.3", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_x", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.3', 'recurrent_block', 'linear_x', 'kernel')": {"key_metadata": [{"key": "blocks.3", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_x", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.3', 'recurrent_block', 'linear_y', 'bias')": {"key_metadata": [{"key": "blocks.3", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_y", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.3', 'recurrent_block', 'linear_y', 'kernel')": {"key_metadata": [{"key": "blocks.3", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_y", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.3', 'recurrent_block', 'rg_lru', 'a_gate', 'b')": {"key_metadata": [{"key": "blocks.3", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "a_gate", "key_type": 2}, {"key": "b", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.3', 'recurrent_block', 'rg_lru', 'a_gate', 'w')": {"key_metadata": [{"key": "blocks.3", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "a_gate", "key_type": 2}, {"key": "w", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.3', 'recurrent_block', 'rg_lru', 'a_param')": {"key_metadata": [{"key": "blocks.3", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "a_param", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.3', 'recurrent_block', 'rg_lru', 'input_gate', 'b')": {"key_metadata": [{"key": "blocks.3", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "input_gate", "key_type": 2}, {"key": "b", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.3', 'recurrent_block', 'rg_lru', 'input_gate', 'w')": {"key_metadata": [{"key": "blocks.3", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "input_gate", "key_type": 2}, {"key": "w", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.3', 'temporal_pre_norm', 'scale')": {"key_metadata": [{"key": "blocks.3", "key_type": 2}, {"key": "temporal_pre_norm", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.4', 'channel_pre_norm', 'scale')": {"key_metadata": [{"key": "blocks.4", "key_type": 2}, {"key": "channel_pre_norm", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.4', 'mlp_block', 'ffw_down', 'bias')": {"key_metadata": [{"key": "blocks.4", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_down", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.4', 'mlp_block', 'ffw_down', 'kernel')": {"key_metadata": [{"key": "blocks.4", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_down", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.4', 'mlp_block', 'ffw_up', 'b')": {"key_metadata": [{"key": "blocks.4", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_up", "key_type": 2}, {"key": "b", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.4', 'mlp_block', 'ffw_up', 'w')": {"key_metadata": [{"key": "blocks.4", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_up", "key_type": 2}, {"key": "w", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.4', 'recurrent_block', 'conv_1d', 'b')": {"key_metadata": [{"key": "blocks.4", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "conv_1d", "key_type": 2}, {"key": "b", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.4', 'recurrent_block', 'conv_1d', 'w')": {"key_metadata": [{"key": "blocks.4", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "conv_1d", "key_type": 2}, {"key": "w", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.4', 'recurrent_block', 'linear_out', 'bias')": {"key_metadata": [{"key": "blocks.4", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_out", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.4', 'recurrent_block', 'linear_out', 'kernel')": {"key_metadata": [{"key": "blocks.4", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_out", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.4', 'recurrent_block', 'linear_x', 'bias')": {"key_metadata": [{"key": "blocks.4", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_x", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.4', 'recurrent_block', 'linear_x', 'kernel')": {"key_metadata": [{"key": "blocks.4", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_x", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.4', 'recurrent_block', 'linear_y', 'bias')": {"key_metadata": [{"key": "blocks.4", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_y", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.4', 'recurrent_block', 'linear_y', 'kernel')": {"key_metadata": [{"key": "blocks.4", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_y", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.4', 'recurrent_block', 'rg_lru', 'a_gate', 'b')": {"key_metadata": [{"key": "blocks.4", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "a_gate", "key_type": 2}, {"key": "b", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.4', 'recurrent_block', 'rg_lru', 'a_gate', 'w')": {"key_metadata": [{"key": "blocks.4", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "a_gate", "key_type": 2}, {"key": "w", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.4', 'recurrent_block', 'rg_lru', 'a_param')": {"key_metadata": [{"key": "blocks.4", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "a_param", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.4', 'recurrent_block', 'rg_lru', 'input_gate', 'b')": {"key_metadata": [{"key": "blocks.4", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "input_gate", "key_type": 2}, {"key": "b", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.4', 'recurrent_block', 'rg_lru', 'input_gate', 'w')": {"key_metadata": [{"key": "blocks.4", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "input_gate", "key_type": 2}, {"key": "w", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.4', 'temporal_pre_norm', 'scale')": {"key_metadata": [{"key": "blocks.4", "key_type": 2}, {"key": "temporal_pre_norm", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.5', 'attention_block', 'proj_final', 'bias')": {"key_metadata": [{"key": "blocks.5", "key_type": 2}, {"key": "attention_block", "key_type": 2}, {"key": "proj_final", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.5', 'attention_block', 'proj_final', 'kernel')": {"key_metadata": [{"key": "blocks.5", "key_type": 2}, {"key": "attention_block", "key_type": 2}, {"key": "proj_final", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.5', 'attention_block', 'proj_k', 'kernel')": {"key_metadata": [{"key": "blocks.5", "key_type": 2}, {"key": "attention_block", "key_type": 2}, {"key": "proj_k", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.5', 'attention_block', 'proj_q', 'kernel')": {"key_metadata": [{"key": "blocks.5", "key_type": 2}, {"key": "attention_block", "key_type": 2}, {"key": "proj_q", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.5', 'attention_block', 'proj_v', 'kernel')": {"key_metadata": [{"key": "blocks.5", "key_type": 2}, {"key": "attention_block", "key_type": 2}, {"key": "proj_v", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.5', 'channel_pre_norm', 'scale')": {"key_metadata": [{"key": "blocks.5", "key_type": 2}, {"key": "channel_pre_norm", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.5', 'mlp_block', 'ffw_down', 'bias')": {"key_metadata": [{"key": "blocks.5", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_down", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.5', 'mlp_block', 'ffw_down', 'kernel')": {"key_metadata": [{"key": "blocks.5", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_down", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.5', 'mlp_block', 'ffw_up', 'b')": {"key_metadata": [{"key": "blocks.5", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_up", "key_type": 2}, {"key": "b", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.5', 'mlp_block', 'ffw_up', 'w')": {"key_metadata": [{"key": "blocks.5", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_up", "key_type": 2}, {"key": "w", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.5', 'temporal_pre_norm', 'scale')": {"key_metadata": [{"key": "blocks.5", "key_type": 2}, {"key": "temporal_pre_norm", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.6', 'channel_pre_norm', 'scale')": {"key_metadata": [{"key": "blocks.6", "key_type": 2}, {"key": "channel_pre_norm", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.6', 'mlp_block', 'ffw_down', 'bias')": {"key_metadata": [{"key": "blocks.6", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_down", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.6', 'mlp_block', 'ffw_down', 'kernel')": {"key_metadata": [{"key": "blocks.6", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_down", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.6', 'mlp_block', 'ffw_up', 'b')": {"key_metadata": [{"key": "blocks.6", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_up", "key_type": 2}, {"key": "b", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.6', 'mlp_block', 'ffw_up', 'w')": {"key_metadata": [{"key": "blocks.6", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_up", "key_type": 2}, {"key": "w", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.6', 'recurrent_block', 'conv_1d', 'b')": {"key_metadata": [{"key": "blocks.6", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "conv_1d", "key_type": 2}, {"key": "b", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.6', 'recurrent_block', 'conv_1d', 'w')": {"key_metadata": [{"key": "blocks.6", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "conv_1d", "key_type": 2}, {"key": "w", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.6', 'recurrent_block', 'linear_out', 'bias')": {"key_metadata": [{"key": "blocks.6", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_out", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.6', 'recurrent_block', 'linear_out', 'kernel')": {"key_metadata": [{"key": "blocks.6", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_out", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.6', 'recurrent_block', 'linear_x', 'bias')": {"key_metadata": [{"key": "blocks.6", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_x", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.6', 'recurrent_block', 'linear_x', 'kernel')": {"key_metadata": [{"key": "blocks.6", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_x", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.6', 'recurrent_block', 'linear_y', 'bias')": {"key_metadata": [{"key": "blocks.6", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_y", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.6', 'recurrent_block', 'linear_y', 'kernel')": {"key_metadata": [{"key": "blocks.6", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_y", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.6', 'recurrent_block', 'rg_lru', 'a_gate', 'b')": {"key_metadata": [{"key": "blocks.6", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "a_gate", "key_type": 2}, {"key": "b", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.6', 'recurrent_block', 'rg_lru', 'a_gate', 'w')": {"key_metadata": [{"key": "blocks.6", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "a_gate", "key_type": 2}, {"key": "w", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.6', 'recurrent_block', 'rg_lru', 'a_param')": {"key_metadata": [{"key": "blocks.6", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "a_param", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.6', 'recurrent_block', 'rg_lru', 'input_gate', 'b')": {"key_metadata": [{"key": "blocks.6", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "input_gate", "key_type": 2}, {"key": "b", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.6', 'recurrent_block', 'rg_lru', 'input_gate', 'w')": {"key_metadata": [{"key": "blocks.6", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "input_gate", "key_type": 2}, {"key": "w", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.6', 'temporal_pre_norm', 'scale')": {"key_metadata": [{"key": "blocks.6", "key_type": 2}, {"key": "temporal_pre_norm", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.7', 'channel_pre_norm', 'scale')": {"key_metadata": [{"key": "blocks.7", "key_type": 2}, {"key": "channel_pre_norm", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.7', 'mlp_block', 'ffw_down', 'bias')": {"key_metadata": [{"key": "blocks.7", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_down", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.7', 'mlp_block', 'ffw_down', 'kernel')": {"key_metadata": [{"key": "blocks.7", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_down", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.7', 'mlp_block', 'ffw_up', 'b')": {"key_metadata": [{"key": "blocks.7", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_up", "key_type": 2}, {"key": "b", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.7', 'mlp_block', 'ffw_up', 'w')": {"key_metadata": [{"key": "blocks.7", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_up", "key_type": 2}, {"key": "w", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.7', 'recurrent_block', 'conv_1d', 'b')": {"key_metadata": [{"key": "blocks.7", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "conv_1d", "key_type": 2}, {"key": "b", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.7', 'recurrent_block', 'conv_1d', 'w')": {"key_metadata": [{"key": "blocks.7", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "conv_1d", "key_type": 2}, {"key": "w", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.7', 'recurrent_block', 'linear_out', 'bias')": {"key_metadata": [{"key": "blocks.7", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_out", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.7', 'recurrent_block', 'linear_out', 'kernel')": {"key_metadata": [{"key": "blocks.7", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_out", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.7', 'recurrent_block', 'linear_x', 'bias')": {"key_metadata": [{"key": "blocks.7", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_x", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.7', 'recurrent_block', 'linear_x', 'kernel')": {"key_metadata": [{"key": "blocks.7", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_x", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.7', 'recurrent_block', 'linear_y', 'bias')": {"key_metadata": [{"key": "blocks.7", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_y", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.7', 'recurrent_block', 'linear_y', 'kernel')": {"key_metadata": [{"key": "blocks.7", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_y", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.7', 'recurrent_block', 'rg_lru', 'a_gate', 'b')": {"key_metadata": [{"key": "blocks.7", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "a_gate", "key_type": 2}, {"key": "b", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.7', 'recurrent_block', 'rg_lru', 'a_gate', 'w')": {"key_metadata": [{"key": "blocks.7", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "a_gate", "key_type": 2}, {"key": "w", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.7', 'recurrent_block', 'rg_lru', 'a_param')": {"key_metadata": [{"key": "blocks.7", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "a_param", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.7', 'recurrent_block', 'rg_lru', 'input_gate', 'b')": {"key_metadata": [{"key": "blocks.7", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "input_gate", "key_type": 2}, {"key": "b", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.7', 'recurrent_block', 'rg_lru', 'input_gate', 'w')": {"key_metadata": [{"key": "blocks.7", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "input_gate", "key_type": 2}, {"key": "w", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.7', 'temporal_pre_norm', 'scale')": {"key_metadata": [{"key": "blocks.7", "key_type": 2}, {"key": "temporal_pre_norm", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.8', 'attention_block', 'proj_final', 'bias')": {"key_metadata": [{"key": "blocks.8", "key_type": 2}, {"key": "attention_block", "key_type": 2}, {"key": "proj_final", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.8', 'attention_block', 'proj_final', 'kernel')": {"key_metadata": [{"key": "blocks.8", "key_type": 2}, {"key": "attention_block", "key_type": 2}, {"key": "proj_final", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.8', 'attention_block', 'proj_k', 'kernel')": {"key_metadata": [{"key": "blocks.8", "key_type": 2}, {"key": "attention_block", "key_type": 2}, {"key": "proj_k", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.8', 'attention_block', 'proj_q', 'kernel')": {"key_metadata": [{"key": "blocks.8", "key_type": 2}, {"key": "attention_block", "key_type": 2}, {"key": "proj_q", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.8', 'attention_block', 'proj_v', 'kernel')": {"key_metadata": [{"key": "blocks.8", "key_type": 2}, {"key": "attention_block", "key_type": 2}, {"key": "proj_v", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.8', 'channel_pre_norm', 'scale')": {"key_metadata": [{"key": "blocks.8", "key_type": 2}, {"key": "channel_pre_norm", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.8', 'mlp_block', 'ffw_down', 'bias')": {"key_metadata": [{"key": "blocks.8", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_down", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.8', 'mlp_block', 'ffw_down', 'kernel')": {"key_metadata": [{"key": "blocks.8", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_down", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.8', 'mlp_block', 'ffw_up', 'b')": {"key_metadata": [{"key": "blocks.8", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_up", "key_type": 2}, {"key": "b", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.8', 'mlp_block', 'ffw_up', 'w')": {"key_metadata": [{"key": "blocks.8", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_up", "key_type": 2}, {"key": "w", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.8', 'temporal_pre_norm', 'scale')": {"key_metadata": [{"key": "blocks.8", "key_type": 2}, {"key": "temporal_pre_norm", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.9', 'channel_pre_norm', 'scale')": {"key_metadata": [{"key": "blocks.9", "key_type": 2}, {"key": "channel_pre_norm", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.9', 'mlp_block', 'ffw_down', 'bias')": {"key_metadata": [{"key": "blocks.9", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_down", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.9', 'mlp_block', 'ffw_down', 'kernel')": {"key_metadata": [{"key": "blocks.9", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_down", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.9', 'mlp_block', 'ffw_up', 'b')": {"key_metadata": [{"key": "blocks.9", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_up", "key_type": 2}, {"key": "b", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.9', 'mlp_block', 'ffw_up', 'w')": {"key_metadata": [{"key": "blocks.9", "key_type": 2}, {"key": "mlp_block", "key_type": 2}, {"key": "ffw_up", "key_type": 2}, {"key": "w", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.9', 'recurrent_block', 'conv_1d', 'b')": {"key_metadata": [{"key": "blocks.9", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "conv_1d", "key_type": 2}, {"key": "b", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.9', 'recurrent_block', 'conv_1d', 'w')": {"key_metadata": [{"key": "blocks.9", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "conv_1d", "key_type": 2}, {"key": "w", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.9', 'recurrent_block', 'linear_out', 'bias')": {"key_metadata": [{"key": "blocks.9", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_out", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.9', 'recurrent_block', 'linear_out', 'kernel')": {"key_metadata": [{"key": "blocks.9", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_out", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.9', 'recurrent_block', 'linear_x', 'bias')": {"key_metadata": [{"key": "blocks.9", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_x", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.9', 'recurrent_block', 'linear_x', 'kernel')": {"key_metadata": [{"key": "blocks.9", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_x", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.9', 'recurrent_block', 'linear_y', 'bias')": {"key_metadata": [{"key": "blocks.9", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_y", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.9', 'recurrent_block', 'linear_y', 'kernel')": {"key_metadata": [{"key": "blocks.9", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "linear_y", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.9', 'recurrent_block', 'rg_lru', 'a_gate', 'b')": {"key_metadata": [{"key": "blocks.9", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "a_gate", "key_type": 2}, {"key": "b", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.9', 'recurrent_block', 'rg_lru', 'a_gate', 'w')": {"key_metadata": [{"key": "blocks.9", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "a_gate", "key_type": 2}, {"key": "w", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.9', 'recurrent_block', 'rg_lru', 'a_param')": {"key_metadata": [{"key": "blocks.9", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "a_param", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.9', 'recurrent_block', 'rg_lru', 'input_gate', 'b')": {"key_metadata": [{"key": "blocks.9", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "input_gate", "key_type": 2}, {"key": "b", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.9', 'recurrent_block', 'rg_lru', 'input_gate', 'w')": {"key_metadata": [{"key": "blocks.9", "key_type": 2}, {"key": "recurrent_block", "key_type": 2}, {"key": "rg_lru", "key_type": 2}, {"key": "input_gate", "key_type": 2}, {"key": "w", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('blocks.9', 'temporal_pre_norm', 'scale')": {"key_metadata": [{"key": "blocks.9", "key_type": 2}, {"key": "temporal_pre_norm", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('embedder', 'input_embedding')": {"key_metadata": [{"key": "embedder", "key_type": 2}, {"key": "input_embedding", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('final_norm', 'scale')": {"key_metadata": [{"key": "final_norm", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}}, "use_zarr3": false}