disable EMA weights for instructpix2pix model, whcih should get memor…

…y usage as well as image quality to what it was before d2ac95f
YenCheChangGoogle · Jan 30, 2023 · aa4688e · aa4688e
1 parent ab059b6
commit aa4688e
Showing 1 changed file with 1 addition and 2 deletions.
diff --git 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