########
# CoDi #
########

thesis_model:
  type: thesis_model
  symbol: thesis_model
  find_unused_parameters: true
  args:
    autokl_cfg: MODEL(sd_autoencoder)
    optimus_cfg: MODEL(optimus_vae)
    clip_cfg: MODEL(clip_frozen)
    unet_config: MODEL(prova)
    beta_linear_start: 0.00085
    beta_linear_end: 0.012
    timesteps: 1000
    vision_scale_factor: 0.18215
    text_scale_factor: 4.3108
    audio_scale_factor: 0.9228
    use_ema: false
    parameterization : "eps"