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

2023-01-30 13:29:44 +03:00 · 2023-01-30 13:29:44 +03:00 · aa4688eb83
commit aa4688eb83
parent ab059b6e48
1 changed files with 1 additions and 2 deletions
--- a/configs/instruct-pix2pix.yaml
+++ b/configs/instruct-pix2pix.yaml
@ -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