Browse Source

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

AUTOMATIC 2 years ago
parent
commit
aa4688eb83
1 changed files with 1 additions and 2 deletions
  1. 1 2
      configs/instruct-pix2pix.yaml

+ 1 - 2
configs/instruct-pix2pix.yaml

@@ -20,8 +20,7 @@ model:
     conditioning_key: hybrid
     conditioning_key: hybrid
     monitor: val/loss_simple_ema
     monitor: val/loss_simple_ema
     scale_factor: 0.18215
     scale_factor: 0.18215
-    use_ema: true
-    load_ema: true
+    use_ema: false
 
 
     scheduler_config: # 10000 warmup steps
     scheduler_config: # 10000 warmup steps
       target: ldm.lr_scheduler.LambdaLinearScheduler
       target: ldm.lr_scheduler.LambdaLinearScheduler