Skip to content

Commit 0389c02

Browse files
Fix noise scheduler api for ldm inference (Project-MONAI#1419)
Fixes Project-MONAI#1418 . ### Description Update the argument for monai generative >=0.22 ### Checks <!--- Put an `x` in all the boxes that apply, and remove the not applicable items --> - [x] Avoid including large-size files in the PR. Signed-off-by: Mingxin <[email protected]>
1 parent 9c6e949 commit 0389c02

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

generative/2d_ldm/inference.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -78,7 +78,7 @@ def main():
7878

7979
scheduler = DDPMScheduler(
8080
num_train_timesteps=args.NoiseScheduler["num_train_timesteps"],
81-
beta_schedule="scaled_linear",
81+
schedule="scaled_linear_beta",
8282
beta_start=args.NoiseScheduler["beta_start"],
8383
beta_end=args.NoiseScheduler["beta_end"],
8484
)

generative/3d_ldm/inference.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -78,7 +78,7 @@ def main():
7878

7979
scheduler = DDPMScheduler(
8080
num_train_timesteps=args.NoiseScheduler["num_train_timesteps"],
81-
beta_schedule="scaled_linear",
81+
schedule="scaled_linear_beta",
8282
beta_start=args.NoiseScheduler["beta_start"],
8383
beta_end=args.NoiseScheduler["beta_end"],
8484
)

0 commit comments

Comments
 (0)