Commit aa4688eb authored by AUTOMATIC's avatar AUTOMATIC
Browse files

disable EMA weights for instructpix2pix model, whcih should get memory usage... 

disable EMA weights for instructpix2pix model, whcih should get memory usage as well as image quality to what it was before d2ac95fa
parent ab059b6e

configs/instruct-pix2pix.yaml

+1 −2
Original line number Diff line number Diff line
@@ -20,8 +20,7 @@ model: 
    conditioning_key: hybrid

    monitor: val/loss_simple_ema

    scale_factor: 0.18215

    use_ema: true

    load_ema: true

    use_ema: false



    scheduler_config: # 10000 warmup steps

      target: ldm.lr_scheduler.LambdaLinearScheduler

CRA Git | Maintained and supported by SUSTech CRA and CCSE