File size: 546 Bytes
9a7fe1f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
########
# CoDi #
########
codi:
type: codi
symbol: codi
find_unused_parameters: true
args:
audioldm_cfg: MODEL(audioldm_autoencoder)
autokl_cfg: MODEL(sd_autoencoder)
optimus_cfg: MODEL(optimus_vae)
clip_cfg: MODEL(clip_frozen)
clap_cfg: MODEL(clap_audio)
unet_config: MODEL(openai_unet_codi)
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" |