From 634a591701bd14d4de496b2075ed8a436b343002 Mon Sep 17 00:00:00 2001
From: rromb <robin.rombach@gmail.com>
Date: Wed, 1 Jun 2022 09:36:27 +0200
Subject: [PATCH] adapted sampling config for clip-encoder training

---
 .../txt2img-1p4B-multinode-clip-encoder.yaml          | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/configs/stable-diffusion/txt2img-1p4B-multinode-clip-encoder.yaml b/configs/stable-diffusion/txt2img-1p4B-multinode-clip-encoder.yaml
index da87af4..22db311 100644
--- a/configs/stable-diffusion/txt2img-1p4B-multinode-clip-encoder.yaml
+++ b/configs/stable-diffusion/txt2img-1p4B-multinode-clip-encoder.yaml
@@ -110,11 +110,16 @@ lightning:
     image_logger:
       target: main.ImageLogger
       params:
-        batch_frequency: 5000
-        max_images: 8
+        batch_frequency: 50
+        max_images: 4
         increase_log_steps: False
         log_first_step: False
-
+        log_images_kwargs:
+          plot_progressive_rows: False
+          plot_diffusion_rows: False
+          N: 4
+          unconditional_guidance_scale: 3.0
+          unconditional_guidance_label: ""
 
   trainer:
     #replace_sampler_ddp: False