version: 1.0

system: "base_tsr"

ddim:
  v_prediction: true
  diffusers:
    num_train_timesteps: 1000
    beta_schedule: 'scaled_linear'
    beta_start: 0.00085
    beta_end: 0.012
    prediction_type: 'v_prediction'
    rescale_betas_zero_snr: true
    timestep_spacing: 'trailing'
    clip_sample: false

diffwrap:
  UDiT:
    in_chans: 384
    out_chans: 128
    embed_dim: 768
    depth: 12
    num_heads: 12
    mlp_ratio: 4.0
    use_checkpoint: false