Skip to content

Commit 438be11

Browse files
448 update LDM bundles with new noise_scheduler API (#449)
Fixes #448 . ### Description This PR is used to update LDM bundles with changed `DDPMScheduler` API. ### Status **Ready/Work in progress/Hold** ### Please ensure all the checkboxes: <!--- Put an `x` in all the boxes that apply, and remove the not applicable items --> - [x] Codeformat tests passed locally by running `./runtests.sh --codeformat`. - [ ] In-line docstrings updated. - [ ] Update `version` and `changelog` in `metadata.json` if changing an existing bundle. - [ ] Please ensure the naming rules in config files meet our requirements (please refer to: `CONTRIBUTING.md`). - [ ] Ensure versions of packages such as `monai`, `pytorch` and `numpy` are correct in `metadata.json`. - [ ] Descriptions should be consistent with the content, such as `eval_metrics` of the provided weights and TorchScript modules. - [ ] Files larger than 25MB are excluded and replaced by providing download links in `large_file.yml`. - [ ] Avoid using path that contains personal information within config files (such as use `/home/your_name/` for `"bundle_root"`). Signed-off-by: Yiheng Wang <[email protected]>
1 parent d3ea24a commit 438be11

File tree

4 files changed

+6
-4
lines changed

4 files changed

+6
-4
lines changed

models/brats_mri_axial_slices_generative_diffusion/configs/metadata.json

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,8 @@
11
{
22
"schema": "https://github.com/Project-MONAI/MONAI-extra-test-data/releases/download/0.8.1/meta_schema_generator_ldm_20230507.json",
3-
"version": "1.0.3",
3+
"version": "1.0.4",
44
"changelog": {
5+
"1.0.4": "update with new lr scheduler api",
56
"1.0.3": "update required packages",
67
"1.0.2": "remove unused saver in inference",
78
"1.0.1": "fix inference folder error",

models/brats_mri_axial_slices_generative_diffusion/configs/train_diffusion.json

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -60,7 +60,7 @@
6060
"_requires_": [
6161
"@load_autoencoder"
6262
],
63-
"beta_schedule": "scaled_linear",
63+
"schedule": "scaled_linear_beta",
6464
"num_train_timesteps": 1000,
6565
"beta_start": 0.0015,
6666
"beta_end": 0.0195

models/brats_mri_generative_diffusion/configs/metadata.json

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,8 @@
11
{
22
"schema": "https://github.com/Project-MONAI/MONAI-extra-test-data/releases/download/0.8.1/meta_schema_generator_ldm_20230507.json",
3-
"version": "1.0.3",
3+
"version": "1.0.4",
44
"changelog": {
5+
"1.0.4": "update with new lr scheduler api",
56
"1.0.3": "update required packages",
67
"1.0.2": "unify dataset dir in different configs",
78
"1.0.1": "update dependency, update trained model weights",

models/brats_mri_generative_diffusion/configs/train_diffusion.json

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -59,7 +59,7 @@
5959
"_requires_": [
6060
"@load_autoencoder"
6161
],
62-
"beta_schedule": "scaled_linear",
62+
"schedule": "scaled_linear_beta",
6363
"num_train_timesteps": 1000,
6464
"beta_start": 0.0015,
6565
"beta_end": 0.0195

0 commit comments

Comments
 (0)