Skip to content
This repository was archived by the owner on Feb 7, 2025. It is now read-only.

Commit c1ec4ed

Browse files
authored
Fix model zoo scheduler args (#398)
1 parent b1c1a45 commit c1ec4ed

File tree

2 files changed

+2
-2
lines changed
  • model-zoo/models

2 files changed

+2
-2
lines changed

model-zoo/models/brain_image_synthesis_latent_diffusion_model/configs/inference.json

+1-1
Original file line numberDiff line numberDiff line change
@@ -84,7 +84,7 @@
8484
"beta_start": 0.0015,
8585
"beta_end": 0.0205,
8686
"num_train_timesteps": 1000,
87-
"beta_schedule": "scaled_linear",
87+
"schedule": "scaled_linear_beta",
8888
"clip_sample": false
8989
},
9090
"noise": "$torch.randn((1, 3, 20, 28, 20)).to(@device)",

model-zoo/models/cxr_image_synthesis_latent_diffusion_model/configs/inference.json

+1-1
Original file line numberDiff line numberDiff line change
@@ -86,7 +86,7 @@
8686
"beta_start": 0.0015,
8787
"beta_end": 0.0205,
8888
"num_train_timesteps": 1000,
89-
"beta_schedule": "scaled_linear",
89+
"schedule": "scaled_linear_beta",
9090
"prediction_type": "v_prediction",
9191
"clip_sample": false
9292
},

0 commit comments

Comments
 (0)