From aa4688eb8345de583070ca9ddb4c6f585f06762b Mon Sep 17 00:00:00 2001
From: AUTOMATIC <16777216c@gmail.com>
Date: Mon, 30 Jan 2023 13:29:44 +0300
Subject: disable EMA weights for instructpix2pix model, whcih should get
 memory usage as well as image quality to what it was before
 d2ac95fa7b2a8d0bcc5361ee16dba9cbb81ff8b2

---
 configs/instruct-pix2pix.yaml | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

(limited to 'configs/instruct-pix2pix.yaml')

diff --git a/configs/instruct-pix2pix.yaml b/configs/instruct-pix2pix.yaml
index 437ddcef..4e896879 100644
--- 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
-- 
cgit v1.2.1