diff --git a/model-zoo/models/brain_image_synthesis_latent_diffusion_model/configs/inference.json b/model-zoo/models/brain_image_synthesis_latent_diffusion_model/configs/inference.json
index 9edac693..5e310dfa 100644
--- a/model-zoo/models/brain_image_synthesis_latent_diffusion_model/configs/inference.json
+++ b/model-zoo/models/brain_image_synthesis_latent_diffusion_model/configs/inference.json
@@ -84,7 +84,7 @@
         "beta_start": 0.0015,
         "beta_end": 0.0205,
         "num_train_timesteps": 1000,
-        "beta_schedule": "scaled_linear",
+        "schedule": "scaled_linear_beta",
         "clip_sample": false
     },
     "noise": "$torch.randn((1, 3, 20, 28, 20)).to(@device)",
diff --git a/model-zoo/models/cxr_image_synthesis_latent_diffusion_model/configs/inference.json b/model-zoo/models/cxr_image_synthesis_latent_diffusion_model/configs/inference.json
index 6f04c78b..a933098d 100644
--- a/model-zoo/models/cxr_image_synthesis_latent_diffusion_model/configs/inference.json
+++ b/model-zoo/models/cxr_image_synthesis_latent_diffusion_model/configs/inference.json
@@ -86,7 +86,7 @@
         "beta_start": 0.0015,
         "beta_end": 0.0205,
         "num_train_timesteps": 1000,
-        "beta_schedule": "scaled_linear",
+        "schedule": "scaled_linear_beta",
         "prediction_type": "v_prediction",
         "clip_sample": false
     },