config.json

{
  "_class_name": "UNet2DConditionModel",
  "num_train_timesteps": 1000,
  "beta_start": 0.0001,
  "beta_end": 0.02,
  "resolution": 512,
  "image_size": 512
}