lightning:
  find_unused_parameters: false
  modelcheckpoint:
    params:
      every_n_train_steps: 5000
      save_top_k: -1
      monitor: null
  callbacks:
    image_logger:
      target: main.ImageLogger
      params:
        batch_frequency: 2000
        max_images: 4
        increase_log_steps: false
        log_first_step: true
        log_all_val: true
        log_images_kwargs:
          use_ema_scope: true
          inpaint: false
          plot_progressive_rows: false
          plot_diffusion_rows: false
          'N': 4
          unconditional_guidance_scale: 3.0
          unconditional_guidance_label:
          - ''
  trainer:
    benchmark: true
    num_sanity_val_steps: 0
    accumulate_grad_batches: 1
    accelerator: ddp
    check_val_every_n_epoch: 10
    gpus: 1