config.json

{
  "arch": "hdit_ImageNet256Sp4",
  "betas": [
    0.9,
    0.95
  ],
  "colorization": false,
  "conditional": false,
  "ema_decay": 0.9999,
  "eps": 0.0,
  "lr": 0.0005,
  "mmse_model_arch": "swinir_L",
  "mmse_model_ckpt_path": null,
  "mmse_noise_std": 0.1,
  "num_flow_steps": 50,
  "stage": "flow",
  "t_schedule": "stratified_uniform",
  "weight_decay": 0.01
}