Add conditioning-tutorial checkpoint: context_model

.gitattributes

@@ -36,3 +36,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 conditioning_and_guidance/joint_model/1500/opt_state/ocdbt.process_0/d/04b2b6c8f458848a158e39ba5b736356 filter=lfs diff=lfs merge=lfs -text
 conditioning_and_guidance/context_model/1500/opt_state/ocdbt.process_0/d/07544e5937eab293eb793c6581913fc0 filter=lfs diff=lfs merge=lfs -text
 conditioning_and_guidance/joint_model/3000/opt_state/ocdbt.process_0/d/2be637e12a96f25ef46554e11f991918 filter=lfs diff=lfs merge=lfs -text
+conditioning_and_guidance/context_model/3000/opt_state/ocdbt.process_0/d/e155b6c9a22ebc2853d9a776c5e13f16 filter=lfs diff=lfs merge=lfs -text

conditioning_and_guidance/context_model/3000/_CHECKPOINT_METADATA

@@ -0,0 +1 @@
+{"item_handlers": {"dataset_state": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler", "ema_params": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler", "extra": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler", "opt_state": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler", "params": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1785787733516171000, "commit_timestamp_nsecs": 1785787733612746000, "custom_metadata": {"num_steps": 3000}}

conditioning_and_guidance/context_model/3000/dataset_state/_METADATA

@@ -0,0 +1 @@
+{"tree_metadata": {"('next_index',)": {"key_metadata": [{"key": "next_index", "key_type": 2}], "value_metadata": {"value_type": "scalar", "skip_deserialize": false}}}, "use_ocdbt": true, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}

conditioning_and_guidance/context_model/3000/ema_params/_METADATA

@@ -0,0 +1 @@
+{"tree_metadata": {"('network', 'context_encoder', 'context_encoders', 'label', 'layers', '0', 'bias', 'value')": {"key_metadata": [{"key": "network", "key_type": 2}, {"key": "context_encoder", "key_type": 2}, {"key": "context_encoders", "key_type": 2}, {"key": "label", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('network', 'context_encoder', 'context_encoders', 'label', 'layers', '0', 'kernel', 'value')": {"key_metadata": [{"key": "network", "key_type": 2}, {"key": "context_encoder", "key_type": 2}, {"key": "context_encoders", "key_type": 2}, {"key": "label", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('network', 'context_encoder', 'context_encoders', 'label', 'layers', '2', 'bias', 'value')": {"key_metadata": [{"key": "network", "key_type": 2}, {"key": "context_encoder", "key_type": 2}, {"key": "context_encoders", "key_type": 2}, {"key": "label", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "2", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('network', 'context_encoder', 'context_encoders', 'label', 'layers', '2', 'kernel', 'value')": {"key_metadata": [{"key": "network", "key_type": 2}, {"key": "context_encoder", "key_type": 2}, {"key": "context_encoders", "key_type": 2}, {"key": "label", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "2", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('network', 'context_encoder', 'time_encoder', 'linear_1', 'bias', 'value')": {"key_metadata": [{"key": "network", "key_type": 2}, {"key": "context_encoder", "key_type": 2}, {"key": "time_encoder", "key_type": 2}, {"key": "linear_1", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('network', 'context_encoder', 'time_encoder', 'linear_1', 'kernel', 'value')": {"key_metadata": [{"key": "network", "key_type": 2}, {"key": "context_encoder", "key_type": 2}, {"key": "time_encoder", "key_type": 2}, {"key": "linear_1", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('network', 'context_encoder', 'time_encoder', 'linear_2', 'bias', 'value')": {"key_metadata": [{"key": "network", "key_type": 2}, {"key": "context_encoder", "key_type": 2}, {"key": "time_encoder", "key_type": 2}, {"key": "linear_2", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('network', 'context_encoder', 'time_encoder', 'linear_2', 'kernel', 'value')": {"key_metadata": [{"key": "network", "key_type": 2}, {"key": "context_encoder", "key_type": 2}, {"key": "time_encoder", "key_type": 2}, {"key": "linear_2", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('network', 'context_encoder', 'time_encoder', 'noise_fourier', 'b', 'value')": {"key_metadata": [{"key": "network", "key_type": 2}, {"key": "context_encoder", "key_type": 2}, {"key": "time_encoder", "key_type": 2}, {"key": "noise_fourier", "key_type": 2}, {"key": "b", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('network', 'context_encoder', 'time_encoder', 'noise_fourier', 'w', 'value')": {"key_metadata": [{"key": "network", "key_type": 2}, {"key": "context_encoder", "key_type": 2}, {"key": "time_encoder", "key_type": 2}, {"key": "noise_fourier", "key_type": 2}, {"key": "w", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('network', 'context_encoder', 'time_encoder', 'time_fourier', 'b', 'value')": {"key_metadata": [{"key": "network", "key_type": 2}, {"key": "context_encoder", "key_type": 2}, {"key": "time_encoder", "key_type": 2}, {"key": "time_fourier", "key_type": 2}, {"key": "b", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('network', 'context_encoder', 'time_encoder', 'time_fourier', 'w', 'value')": {"key_metadata": [{"key": "network", "key_type": 2}, {"key": "context_encoder", "key_type": 2}, {"key": "time_encoder", "key_type": 2}, {"key": "time_fourier", "key_type": 2}, {"key": "w", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('network', 'heads', 'coordinates', 'bias', 'value')": {"key_metadata": [{"key": "network", "key_type": 2}, {"key": "heads", "key_type": 2}, {"key": "coordinates", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('network', 'heads', 'coordinates', 'kernel', 'value')": {"key_metadata": [{"key": "network", "key_type": 2}, {"key": "heads", "key_type": 2}, {"key": "coordinates", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('network', 'trunk', 'layers', '0', 'bias', 'value')": {"key_metadata": [{"key": "network", "key_type": 2}, {"key": "trunk", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('network', 'trunk', 'layers', '0', 'kernel', 'value')": {"key_metadata": [{"key": "network", "key_type": 2}, {"key": "trunk", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('network', 'trunk', 'layers', '2', 'bias', 'value')": {"key_metadata": [{"key": "network", "key_type": 2}, {"key": "trunk", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "2", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('network', 'trunk', 'layers', '2', 'kernel', 'value')": {"key_metadata": [{"key": "network", "key_type": 2}, {"key": "trunk", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "2", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}}, "use_ocdbt": true, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}

conditioning_and_guidance/context_model/3000/extra/_METADATA

@@ -0,0 +1 @@
+{"tree_metadata": {"('ema_decay',)": {"key_metadata": [{"key": "ema_decay", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('num_steps',)": {"key_metadata": [{"key": "num_steps", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('rng_key',)": {"key_metadata": [{"key": "rng_key", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}}, "use_ocdbt": true, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}

conditioning_and_guidance/context_model/3000/opt_state/_METADATA

@@ -0,0 +1 @@
+{"tree_metadata": {"('0', 'count')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "count", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('0', 'mu', 'network', 'context_encoder', 'context_encoders', 'label', 'layers', '0', 'bias', 'value')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "network", "key_type": 2}, {"key": "context_encoder", "key_type": 2}, {"key": "context_encoders", "key_type": 2}, {"key": "label", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('0', 'mu', 'network', 'context_encoder', 'context_encoders', 'label', 'layers', '0', 'kernel', 'value')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "network", "key_type": 2}, {"key": "context_encoder", "key_type": 2}, {"key": "context_encoders", "key_type": 2}, {"key": "label", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('0', 'mu', 'network', 'context_encoder', 'context_encoders', 'label', 'layers', '2', 'bias', 'value')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "network", "key_type": 2}, {"key": "context_encoder", "key_type": 2}, {"key": "context_encoders", "key_type": 2}, {"key": "label", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "2", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('0', 'mu', 'network', 'context_encoder', 'context_encoders', 'label', 'layers', '2', 'kernel', 'value')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "network", "key_type": 2}, {"key": "context_encoder", "key_type": 2}, {"key": "context_encoders", "key_type": 2}, {"key": "label", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "2", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('0', 'mu', 'network', 'context_encoder', 'time_encoder', 'linear_1', 'bias', 'value')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "network", "key_type": 2}, {"key": "context_encoder", "key_type": 2}, {"key": "time_encoder", "key_type": 2}, {"key": "linear_1", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('0', 'mu', 'network', 'context_encoder', 'time_encoder', 'linear_1', 'kernel', 'value')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "network", "key_type": 2}, {"key": "context_encoder", "key_type": 2}, {"key": "time_encoder", "key_type": 2}, {"key": "linear_1", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('0', 'mu', 'network', 'context_encoder', 'time_encoder', 'linear_2', 'bias', 'value')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "network", "key_type": 2}, {"key": "context_encoder", "key_type": 2}, {"key": "time_encoder", "key_type": 2}, {"key": "linear_2", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('0', 'mu', 'network', 'context_encoder', 'time_encoder', 'linear_2', 'kernel', 'value')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "network", "key_type": 2}, {"key": "context_encoder", "key_type": 2}, {"key": "time_encoder", "key_type": 2}, {"key": "linear_2", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('0', 'mu', 'network', 'context_encoder', 'time_encoder', 'noise_fourier', 'b', 'value')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "network", "key_type": 2}, {"key": "context_encoder", "key_type": 2}, {"key": "time_encoder", "key_type": 2}, {"key": "noise_fourier", "key_type": 2}, {"key": "b", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('0', 'mu', 'network', 'context_encoder', 'time_encoder', 'noise_fourier', 'w', 'value')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "network", "key_type": 2}, {"key": "context_encoder", "key_type": 2}, {"key": "time_encoder", "key_type": 2}, {"key": "noise_fourier", "key_type": 2}, {"key": "w", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('0', 'mu', 'network', 'context_encoder', 'time_encoder', 'time_fourier', 'b', 'value')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "network", "key_type": 2}, {"key": "context_encoder", "key_type": 2}, {"key": "time_encoder", "key_type": 2}, {"key": "time_fourier", "key_type": 2}, {"key": "b", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('0', 'mu', 'network', 'context_encoder', 'time_encoder', 'time_fourier', 'w', 'value')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "network", "key_type": 2}, {"key": "context_encoder", "key_type": 2}, {"key": "time_encoder", "key_type": 2}, {"key": "time_fourier", "key_type": 2}, {"key": "w", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('0', 'mu', 'network', 'heads', 'coordinates', 'bias', 'value')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "network", "key_type": 2}, {"key": "heads", "key_type": 2}, {"key": "coordinates", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('0', 'mu', 'network', 'heads', 'coordinates', 'kernel', 'value')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "network", "key_type": 2}, {"key": "heads", "key_type": 2}, {"key": "coordinates", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('0', 'mu', 'network', 'trunk', 'layers', '0', 'bias', 'value')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "network", "key_type": 2}, {"key": "trunk", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('0', 'mu', 'network', 'trunk', 'layers', '0', 'kernel', 'value')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "network", "key_type": 2}, {"key": "trunk", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('0', 'mu', 'network', 'trunk', 'layers', '2', 'bias', 'value')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "network", "key_type": 2}, {"key": "trunk", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "2", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('0', 'mu', 'network', 'trunk', 'layers', '2', 'kernel', 'value')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "network", "key_type": 2}, {"key": "trunk", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "2", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('0', 'nu', 'network', 'context_encoder', 'context_encoders', 'label', 'layers', '0', 'bias', 'value')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "network", "key_type": 2}, {"key": "context_encoder", "key_type": 2}, {"key": "context_encoders", "key_type": 2}, {"key": "label", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('0', 'nu', 'network', 'context_encoder', 'context_encoders', 'label', 'layers', '0', 'kernel', 'value')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "network", "key_type": 2}, {"key": "context_encoder", "key_type": 2}, {"key": "context_encoders", "key_type": 2}, {"key": "label", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('0', 'nu', 'network', 'context_encoder', 'context_encoders', 'label', 'layers', '2', 'bias', 'value')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "network", "key_type": 2}, {"key": "context_encoder", "key_type": 2}, {"key": "context_encoders", "key_type": 2}, {"key": "label", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "2", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('0', 'nu', 'network', 'context_encoder', 'context_encoders', 'label', 'layers', '2', 'kernel', 'value')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "network", "key_type": 2}, {"key": "context_encoder", "key_type": 2}, {"key": "context_encoders", "key_type": 2}, {"key": "label", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "2", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('0', 'nu', 'network', 'context_encoder', 'time_encoder', 'linear_1', 'bias', 'value')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "network", "key_type": 2}, {"key": "context_encoder", "key_type": 2}, {"key": "time_encoder", "key_type": 2}, {"key": "linear_1", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('0', 'nu', 'network', 'context_encoder', 'time_encoder', 'linear_1', 'kernel', 'value')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "network", "key_type": 2}, {"key": "context_encoder", "key_type": 2}, {"key": "time_encoder", "key_type": 2}, {"key": "linear_1", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('0', 'nu', 'network', 'context_encoder', 'time_encoder', 'linear_2', 'bias', 'value')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "network", "key_type": 2}, {"key": "context_encoder", "key_type": 2}, {"key": "time_encoder", "key_type": 2}, {"key": "linear_2", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('0', 'nu', 'network', 'context_encoder', 'time_encoder', 'linear_2', 'kernel', 'value')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "network", "key_type": 2}, {"key": "context_encoder", "key_type": 2}, {"key": "time_encoder", "key_type": 2}, {"key": "linear_2", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('0', 'nu', 'network', 'context_encoder', 'time_encoder', 'noise_fourier', 'b', 'value')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "network", "key_type": 2}, {"key": "context_encoder", "key_type": 2}, {"key": "time_encoder", "key_type": 2}, {"key": "noise_fourier", "key_type": 2}, {"key": "b", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('0', 'nu', 'network', 'context_encoder', 'time_encoder', 'noise_fourier', 'w', 'value')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "network", "key_type": 2}, {"key": "context_encoder", "key_type": 2}, {"key": "time_encoder", "key_type": 2}, {"key": "noise_fourier", "key_type": 2}, {"key": "w", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('0', 'nu', 'network', 'context_encoder', 'time_encoder', 'time_fourier', 'b', 'value')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "network", "key_type": 2}, {"key": "context_encoder", "key_type": 2}, {"key": "time_encoder", "key_type": 2}, {"key": "time_fourier", "key_type": 2}, {"key": "b", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('0', 'nu', 'network', 'context_encoder', 'time_encoder', 'time_fourier', 'w', 'value')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "network", "key_type": 2}, {"key": "context_encoder", "key_type": 2}, {"key": "time_encoder", "key_type": 2}, {"key": "time_fourier", "key_type": 2}, {"key": "w", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('0', 'nu', 'network', 'heads', 'coordinates', 'bias', 'value')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "network", "key_type": 2}, {"key": "heads", "key_type": 2}, {"key": "coordinates", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('0', 'nu', 'network', 'heads', 'coordinates', 'kernel', 'value')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "network", "key_type": 2}, {"key": "heads", "key_type": 2}, {"key": "coordinates", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('0', 'nu', 'network', 'trunk', 'layers', '0', 'bias', 'value')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "network", "key_type": 2}, {"key": "trunk", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('0', 'nu', 'network', 'trunk', 'layers', '0', 'kernel', 'value')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "network", "key_type": 2}, {"key": "trunk", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('0', 'nu', 'network', 'trunk', 'layers', '2', 'bias', 'value')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "network", "key_type": 2}, {"key": "trunk", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "2", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('0', 'nu', 'network', 'trunk', 'layers', '2', 'kernel', 'value')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "network", "key_type": 2}, {"key": "trunk", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "2", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('1', 'count')": {"key_metadata": [{"key": "1", "key_type": 1}, {"key": "count", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}}, "use_ocdbt": true, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}

conditioning_and_guidance/context_model/3000/opt_state/ocdbt.process_0/d/e155b6c9a22ebc2853d9a776c5e13f16

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:99494ebbeefd150acc93303373c8f7ac669b61e4fc57421af91f0c4b695d912b
+size 180769

conditioning_and_guidance/context_model/3000/params/_METADATA

@@ -0,0 +1 @@
+{"tree_metadata": {"('network', 'context_encoder', 'context_encoders', 'label', 'layers', '0', 'bias', 'value')": {"key_metadata": [{"key": "network", "key_type": 2}, {"key": "context_encoder", "key_type": 2}, {"key": "context_encoders", "key_type": 2}, {"key": "label", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('network', 'context_encoder', 'context_encoders', 'label', 'layers', '0', 'kernel', 'value')": {"key_metadata": [{"key": "network", "key_type": 2}, {"key": "context_encoder", "key_type": 2}, {"key": "context_encoders", "key_type": 2}, {"key": "label", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('network', 'context_encoder', 'context_encoders', 'label', 'layers', '2', 'bias', 'value')": {"key_metadata": [{"key": "network", "key_type": 2}, {"key": "context_encoder", "key_type": 2}, {"key": "context_encoders", "key_type": 2}, {"key": "label", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "2", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('network', 'context_encoder', 'context_encoders', 'label', 'layers', '2', 'kernel', 'value')": {"key_metadata": [{"key": "network", "key_type": 2}, {"key": "context_encoder", "key_type": 2}, {"key": "context_encoders", "key_type": 2}, {"key": "label", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "2", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('network', 'context_encoder', 'time_encoder', 'linear_1', 'bias', 'value')": {"key_metadata": [{"key": "network", "key_type": 2}, {"key": "context_encoder", "key_type": 2}, {"key": "time_encoder", "key_type": 2}, {"key": "linear_1", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('network', 'context_encoder', 'time_encoder', 'linear_1', 'kernel', 'value')": {"key_metadata": [{"key": "network", "key_type": 2}, {"key": "context_encoder", "key_type": 2}, {"key": "time_encoder", "key_type": 2}, {"key": "linear_1", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('network', 'context_encoder', 'time_encoder', 'linear_2', 'bias', 'value')": {"key_metadata": [{"key": "network", "key_type": 2}, {"key": "context_encoder", "key_type": 2}, {"key": "time_encoder", "key_type": 2}, {"key": "linear_2", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('network', 'context_encoder', 'time_encoder', 'linear_2', 'kernel', 'value')": {"key_metadata": [{"key": "network", "key_type": 2}, {"key": "context_encoder", "key_type": 2}, {"key": "time_encoder", "key_type": 2}, {"key": "linear_2", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('network', 'context_encoder', 'time_encoder', 'noise_fourier', 'b', 'value')": {"key_metadata": [{"key": "network", "key_type": 2}, {"key": "context_encoder", "key_type": 2}, {"key": "time_encoder", "key_type": 2}, {"key": "noise_fourier", "key_type": 2}, {"key": "b", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('network', 'context_encoder', 'time_encoder', 'noise_fourier', 'w', 'value')": {"key_metadata": [{"key": "network", "key_type": 2}, {"key": "context_encoder", "key_type": 2}, {"key": "time_encoder", "key_type": 2}, {"key": "noise_fourier", "key_type": 2}, {"key": "w", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('network', 'context_encoder', 'time_encoder', 'time_fourier', 'b', 'value')": {"key_metadata": [{"key": "network", "key_type": 2}, {"key": "context_encoder", "key_type": 2}, {"key": "time_encoder", "key_type": 2}, {"key": "time_fourier", "key_type": 2}, {"key": "b", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('network', 'context_encoder', 'time_encoder', 'time_fourier', 'w', 'value')": {"key_metadata": [{"key": "network", "key_type": 2}, {"key": "context_encoder", "key_type": 2}, {"key": "time_encoder", "key_type": 2}, {"key": "time_fourier", "key_type": 2}, {"key": "w", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('network', 'heads', 'coordinates', 'bias', 'value')": {"key_metadata": [{"key": "network", "key_type": 2}, {"key": "heads", "key_type": 2}, {"key": "coordinates", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('network', 'heads', 'coordinates', 'kernel', 'value')": {"key_metadata": [{"key": "network", "key_type": 2}, {"key": "heads", "key_type": 2}, {"key": "coordinates", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('network', 'trunk', 'layers', '0', 'bias', 'value')": {"key_metadata": [{"key": "network", "key_type": 2}, {"key": "trunk", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('network', 'trunk', 'layers', '0', 'kernel', 'value')": {"key_metadata": [{"key": "network", "key_type": 2}, {"key": "trunk", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('network', 'trunk', 'layers', '2', 'bias', 'value')": {"key_metadata": [{"key": "network", "key_type": 2}, {"key": "trunk", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "2", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('network', 'trunk', 'layers', '2', 'kernel', 'value')": {"key_metadata": [{"key": "network", "key_type": 2}, {"key": "trunk", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "2", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}}, "use_ocdbt": true, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}