From e9e2df0dfa21e20e8feca4153ef7e1b7202875e5 Mon Sep 17 00:00:00 2001 From: Mark Graham Date: Thu, 25 May 2023 15:45:15 -0600 Subject: [PATCH] Fixed model zoo scheduler args --- .../configs/inference.json | 2 +- .../configs/inference.json | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) 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 },