| { |
| "format_version": 1, |
| "release": "FloodDiffusion2", |
| "checkpoints": [ |
| { |
| "config": "vae_263.yaml", |
| "global_step": 2250000, |
| "checkpoint": { |
| "path": "checkpoints/vae_263/model.ckpt", |
| "bytes": 350279555, |
| "sha256": "b80ce00b538b24b874dd15ac84fa4f34d9469fe7ac7f8d99e60bea276c3918d4" |
| }, |
| "configuration": { |
| "path": "checkpoints/vae_263/config.yaml", |
| "bytes": 2581, |
| "sha256": "7047f20640a97538d8629bf91cd12c667d7992740894e051501eb1badd0f098e" |
| }, |
| "assets": [ |
| { |
| "path": "checkpoints/vae_263/assets/Mean.npy", |
| "bytes": 1180, |
| "sha256": "bef8fea46b1c9ed21c378a378c66c5f6b55f7ecf2fdfa911670a7d020e7fffb0" |
| }, |
| { |
| "path": "checkpoints/vae_263/assets/Std.npy", |
| "bytes": 1180, |
| "sha256": "041dee3a9e09c97e5f66e10fb4b63b9410fb57ecc609a556ec0b65d8def2fb20" |
| } |
| ], |
| "tensor_count_verified": 638, |
| "preserved_training_state": [ |
| "state_dict", |
| "ema_state", |
| "optimizer_states", |
| "lr_schedulers", |
| "loops", |
| "epoch", |
| "global_step" |
| ], |
| "removed_metadata": [ |
| "callbacks" |
| ], |
| "normalization_dimension": 263, |
| "loss_matrix_validation": null |
| }, |
| { |
| "config": "ldf_263.yaml", |
| "global_step": 180000, |
| "checkpoint": { |
| "path": "checkpoints/ldf_263/model.ckpt", |
| "bytes": 2274793067, |
| "sha256": "94cc2ea9c0c29213a7f615e721c5fe5264e96d0a95f7f67adc0a5a5081f69437" |
| }, |
| "configuration": { |
| "path": "checkpoints/ldf_263/config.yaml", |
| "bytes": 4564, |
| "sha256": "4a252ff70dbd9a1864a9f0afb7e71f0842440d1cc8bcdf360ab1769cbcf65f41" |
| }, |
| "assets": [ |
| { |
| "path": "checkpoints/ldf_263/assets/Mean.npy", |
| "bytes": 144, |
| "sha256": "c8b1c3183972f4ba2e542eb8bc26945f5d13782d04b56345d6ab5c71e714dff6" |
| }, |
| { |
| "path": "checkpoints/ldf_263/assets/Std.npy", |
| "bytes": 144, |
| "sha256": "a4234b3c2ca19566cd74b749e1c1ede5cbc2ab245cfcc487891b14c056a2357d" |
| } |
| ], |
| "tensor_count_verified": 1388, |
| "preserved_training_state": [ |
| "state_dict", |
| "ema_state", |
| "optimizer_states", |
| "lr_schedulers", |
| "loops", |
| "epoch", |
| "global_step" |
| ], |
| "removed_metadata": [ |
| "callbacks" |
| ], |
| "normalization_dimension": 4, |
| "loss_matrix_validation": null |
| }, |
| { |
| "config": "df_humanml3d_263.yaml", |
| "global_step": 60000, |
| "checkpoint": { |
| "path": "checkpoints/humanml3d_fk_60k/model.ckpt", |
| "bytes": 2285681258, |
| "sha256": "d48332bcd88d06098627ed895e8d708702290d6ed518b20e5573b6b959725a63" |
| }, |
| "configuration": { |
| "path": "checkpoints/humanml3d_fk_60k/config.yaml", |
| "bytes": 3811, |
| "sha256": "983449db24beb87f9d63fbde21927e7119990bf10d6af6e32324f1a5526a7b8f" |
| }, |
| "assets": [ |
| { |
| "path": "checkpoints/humanml3d_fk_60k/assets/Mean.npy", |
| "bytes": 1180, |
| "sha256": "bef8fea46b1c9ed21c378a378c66c5f6b55f7ecf2fdfa911670a7d020e7fffb0" |
| }, |
| { |
| "path": "checkpoints/humanml3d_fk_60k/assets/Std.npy", |
| "bytes": 1180, |
| "sha256": "041dee3a9e09c97e5f66e10fb4b63b9410fb57ecc609a556ec0b65d8def2fb20" |
| }, |
| { |
| "path": "checkpoints/humanml3d_fk_60k/assets/W.npy", |
| "bytes": 553480, |
| "sha256": "4a8411b10882a74d09479e20e8ec10d0208473127f23e9da4d3192e2fea145d6" |
| } |
| ], |
| "tensor_count_verified": 1389, |
| "preserved_training_state": [ |
| "state_dict", |
| "ema_state", |
| "optimizer_states", |
| "lr_schedulers", |
| "loops", |
| "epoch", |
| "global_step" |
| ], |
| "removed_metadata": [ |
| "callbacks" |
| ], |
| "normalization_dimension": 263, |
| "loss_matrix_validation": { |
| "dimension": 263, |
| "coefficient": 1.0, |
| "raw_matrix_trace": 263.0000009536743, |
| "trace_normalization_required": false, |
| "loader_roundtrip_fp32_exact": true |
| } |
| }, |
| { |
| "config": "df_humanml3d_263_path.yaml", |
| "global_step": 55000, |
| "checkpoint": { |
| "path": "checkpoints/humanml3d_path_fk_55k/model.ckpt", |
| "bytes": 2285614312, |
| "sha256": "ee1239ee5458781ab84d21329d190bfcb45a23d12bec17df48924e000672d104" |
| }, |
| "configuration": { |
| "path": "checkpoints/humanml3d_path_fk_55k/config.yaml", |
| "bytes": 3877, |
| "sha256": "efcb66b0ec75c721b418887fa9824285dcc8934dc98550218348303bda273332" |
| }, |
| "assets": [ |
| { |
| "path": "checkpoints/humanml3d_path_fk_55k/assets/Mean.npy", |
| "bytes": 1180, |
| "sha256": "bef8fea46b1c9ed21c378a378c66c5f6b55f7ecf2fdfa911670a7d020e7fffb0" |
| }, |
| { |
| "path": "checkpoints/humanml3d_path_fk_55k/assets/Std.npy", |
| "bytes": 1180, |
| "sha256": "041dee3a9e09c97e5f66e10fb4b63b9410fb57ecc609a556ec0b65d8def2fb20" |
| }, |
| { |
| "path": "checkpoints/humanml3d_path_fk_55k/assets/W.npy", |
| "bytes": 540928, |
| "sha256": "ebc7243c5185ab6203af8e0b912e05d6bd15919f9eda7391b548a870eefea373" |
| } |
| ], |
| "tensor_count_verified": 1391, |
| "preserved_training_state": [ |
| "state_dict", |
| "ema_state", |
| "optimizer_states", |
| "lr_schedulers", |
| "loops", |
| "epoch", |
| "global_step" |
| ], |
| "removed_metadata": [ |
| "callbacks" |
| ], |
| "normalization_dimension": 263, |
| "loss_matrix_validation": { |
| "dimension": 260, |
| "coefficient": 1.0, |
| "raw_matrix_trace": 259.9999974966049, |
| "trace_normalization_required": true, |
| "loader_roundtrip_fp32_exact": true |
| } |
| }, |
| { |
| "config": "df_humanml3d_babel_263.yaml", |
| "global_step": 200000, |
| "checkpoint": { |
| "path": "checkpoints/humanml3d_babel_fk_200k/model.ckpt", |
| "bytes": 2285685418, |
| "sha256": "216a6168daca6552495ff8415a7cc31d896e9dfb116e4165e1b777ac609e6ae3" |
| }, |
| "configuration": { |
| "path": "checkpoints/humanml3d_babel_fk_200k/config.yaml", |
| "bytes": 4434, |
| "sha256": "69ee7e4642b5baa3bba4e9e89e14a69c020cb936075ced8683be98426488eac3" |
| }, |
| "assets": [ |
| { |
| "path": "checkpoints/humanml3d_babel_fk_200k/assets/Mean.npy", |
| "bytes": 1180, |
| "sha256": "bef8fea46b1c9ed21c378a378c66c5f6b55f7ecf2fdfa911670a7d020e7fffb0" |
| }, |
| { |
| "path": "checkpoints/humanml3d_babel_fk_200k/assets/Std.npy", |
| "bytes": 1180, |
| "sha256": "041dee3a9e09c97e5f66e10fb4b63b9410fb57ecc609a556ec0b65d8def2fb20" |
| }, |
| { |
| "path": "checkpoints/humanml3d_babel_fk_200k/assets/W.npy", |
| "bytes": 553480, |
| "sha256": "778981f8025de72f2f98121ae018e81f69a8ce54fc0a2a8615c33706eb38d534" |
| } |
| ], |
| "tensor_count_verified": 1389, |
| "preserved_training_state": [ |
| "state_dict", |
| "ema_state", |
| "optimizer_states", |
| "lr_schedulers", |
| "loops", |
| "epoch", |
| "global_step" |
| ], |
| "removed_metadata": [ |
| "callbacks" |
| ], |
| "normalization_dimension": 263, |
| "loss_matrix_validation": { |
| "dimension": 263, |
| "coefficient": 1.0, |
| "raw_matrix_trace": 263.0000002384186, |
| "trace_normalization_required": true, |
| "loader_roundtrip_fp32_exact": true |
| } |
| }, |
| { |
| "config": "df_humanml3d_babel_263_path.yaml", |
| "global_step": 200000, |
| "checkpoint": { |
| "path": "checkpoints/humanml3d_babel_path_200k/model.ckpt", |
| "bytes": 2285618728, |
| "sha256": "d2e6a28baf6e1cfcd8c2061425ebdd82b1024091e9714729b087b428485cac0e" |
| }, |
| "configuration": { |
| "path": "checkpoints/humanml3d_babel_path_200k/config.yaml", |
| "bytes": 4534, |
| "sha256": "a86c8536002233d68512a644c0affc21274d5f9610aff58479118ec0538d8e28" |
| }, |
| "assets": [ |
| { |
| "path": "checkpoints/humanml3d_babel_path_200k/assets/Mean.npy", |
| "bytes": 1180, |
| "sha256": "bef8fea46b1c9ed21c378a378c66c5f6b55f7ecf2fdfa911670a7d020e7fffb0" |
| }, |
| { |
| "path": "checkpoints/humanml3d_babel_path_200k/assets/Std.npy", |
| "bytes": 1180, |
| "sha256": "041dee3a9e09c97e5f66e10fb4b63b9410fb57ecc609a556ec0b65d8def2fb20" |
| }, |
| { |
| "path": "checkpoints/humanml3d_babel_path_200k/assets/W.npy", |
| "bytes": 540928, |
| "sha256": "ebc7243c5185ab6203af8e0b912e05d6bd15919f9eda7391b548a870eefea373" |
| } |
| ], |
| "tensor_count_verified": 1391, |
| "preserved_training_state": [ |
| "state_dict", |
| "ema_state", |
| "optimizer_states", |
| "lr_schedulers", |
| "loops", |
| "epoch", |
| "global_step" |
| ], |
| "removed_metadata": [ |
| "callbacks" |
| ], |
| "normalization_dimension": 263, |
| "loss_matrix_validation": { |
| "dimension": 260, |
| "coefficient": 1.0, |
| "raw_matrix_trace": 259.9999974966049, |
| "trace_normalization_required": true, |
| "loader_roundtrip_fp32_exact": true |
| } |
| }, |
| { |
| "config": "df_seed_138.yaml", |
| "global_step": 300000, |
| "checkpoint": { |
| "path": "checkpoints/seed_fk_300k/model.ckpt", |
| "bytes": 2280356842, |
| "sha256": "15caeff8faaa19f09437aee9f79088955708a32fd6ff609cc990338460dc130f" |
| }, |
| "configuration": { |
| "path": "checkpoints/seed_fk_300k/config.yaml", |
| "bytes": 3697, |
| "sha256": "41825321dce5aea7b3a384b4717b74aa7141c8e93988abc1987fd20bbdf6c686" |
| }, |
| "assets": [ |
| { |
| "path": "checkpoints/seed_fk_300k/assets/Mean.npy", |
| "bytes": 680, |
| "sha256": "98456f74f871856f48880c72d7340a776cb45dc88ba33e86491cfaf1f8bc9534" |
| }, |
| { |
| "path": "checkpoints/seed_fk_300k/assets/Std.npy", |
| "bytes": 680, |
| "sha256": "5ea97221c9a6d471d12b92d6562d4214d1080fb398e96e43f628b393a70b2b96" |
| }, |
| { |
| "path": "checkpoints/seed_fk_300k/assets/W.npy", |
| "bytes": 152480, |
| "sha256": "d5c81bdf26ff2347bb59840ab3bb52b179d9488fbb5ff88ea65f3e76fc9c5502" |
| } |
| ], |
| "tensor_count_verified": 1389, |
| "preserved_training_state": [ |
| "state_dict", |
| "ema_state", |
| "optimizer_states", |
| "lr_schedulers", |
| "loops", |
| "epoch", |
| "global_step" |
| ], |
| "removed_metadata": [ |
| "callbacks" |
| ], |
| "normalization_dimension": 138, |
| "loss_matrix_validation": { |
| "dimension": 138, |
| "coefficient": 1.0, |
| "raw_matrix_trace": 137.99999964237213, |
| "trace_normalization_required": true, |
| "loader_roundtrip_fp32_exact": true |
| } |
| }, |
| { |
| "config": "df_seed_138_path.yaml", |
| "global_step": 300000, |
| "checkpoint": { |
| "path": "checkpoints/seed_path_fk_300k/model.ckpt", |
| "bytes": 2280293224, |
| "sha256": "d60a085a280962f66669fbeb12568e6f6cd1b67304647c1b73fc847862498ef6" |
| }, |
| "configuration": { |
| "path": "checkpoints/seed_path_fk_300k/config.yaml", |
| "bytes": 3811, |
| "sha256": "2c414967a1e1ad2d6e959e4a1ce7fc22d43086a69340d245f16a5e41bcef850e" |
| }, |
| "assets": [ |
| { |
| "path": "checkpoints/seed_path_fk_300k/assets/Mean.npy", |
| "bytes": 680, |
| "sha256": "98456f74f871856f48880c72d7340a776cb45dc88ba33e86491cfaf1f8bc9534" |
| }, |
| { |
| "path": "checkpoints/seed_path_fk_300k/assets/Std.npy", |
| "bytes": 680, |
| "sha256": "5ea97221c9a6d471d12b92d6562d4214d1080fb398e96e43f628b393a70b2b96" |
| }, |
| { |
| "path": "checkpoints/seed_path_fk_300k/assets/W.npy", |
| "bytes": 145928, |
| "sha256": "0effaf2383a21f4e5a1bf03aed5841060cd45226ac93177e8e09ecd9dbf540b5" |
| } |
| ], |
| "tensor_count_verified": 1391, |
| "preserved_training_state": [ |
| "state_dict", |
| "ema_state", |
| "optimizer_states", |
| "lr_schedulers", |
| "loops", |
| "epoch", |
| "global_step" |
| ], |
| "removed_metadata": [ |
| "callbacks" |
| ], |
| "normalization_dimension": 138, |
| "loss_matrix_validation": { |
| "dimension": 135, |
| "coefficient": 1.0, |
| "raw_matrix_trace": 135.0000001192093, |
| "trace_normalization_required": true, |
| "loader_roundtrip_fp32_exact": true |
| } |
| } |
| ], |
| "total_checkpoint_bytes": 16328322404, |
| "complete": true |
| } |
|
|
