config.json

{
  "beta_schedule": "sigmoid",
  "ddim_sampling_eta": 0.0,
  "min_snr_gamma": 5,
  "min_snr_loss_weight": false,
  "objective": "pred_v",
  "offset_noise_strength": 0.0,
  "sampling_timesteps": 20,
  "schedule_fn_kwargs": {},
  "stldm_param": {
    "base_ch": 64,
    "chs_mult": [
      1,
      2,
      4,
      8
    ],
    "dim_head": 32,
    "heads": 4,
    "in_ch": 32,
    "num_groups": 8,
    "patch_size": 16
  },
  "timesteps": 50,
  "vp_param": {
    "N_S": 4,
    "N_T": 8,
    "groups": 8,
    "hid_S": 32,
    "hid_T": 512,
    "last_activation": "sigmoid",
    "shape_in": [
      5,
      1,
      128,
      128
    ],
    "shape_out": [
      20,
      1,
      128,
      128
    ]
  }
}