MedCoDi-M / configs /model /codi_2.yaml
dmolino's picture
Upload 276 files
168a510 verified
raw
history blame
477 Bytes
########
# CoDi #
########
codi_2:
type: codi_2
symbol: codi_2
find_unused_parameters: true
args:
autokl_cfg: MODEL(sd_autoencoder)
optimus_cfg: MODEL(optimus_vae)
clip_cfg: MODEL(clip_frozen)
unet_config: MODEL(openai_unet_codi_2)
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"