From 8ff9bdb78f863e5942300caf363d8acafa613345 Mon Sep 17 00:00:00 2001
From: Patrick Esser <patrick.esser@gmx.net>
Date: Mon, 13 Jun 2022 11:12:22 +0000
Subject: [PATCH] finalize upscaler conf

---
 .../txt2img-upscale-clip-encoder-f16-1024.yaml         | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/configs/stable-diffusion/txt2img-upscale-clip-encoder-f16-1024.yaml b/configs/stable-diffusion/txt2img-upscale-clip-encoder-f16-1024.yaml
index 2f1ab09..1dab4d2 100644
--- a/configs/stable-diffusion/txt2img-upscale-clip-encoder-f16-1024.yaml
+++ b/configs/stable-diffusion/txt2img-upscale-clip-encoder-f16-1024.yaml
@@ -66,7 +66,7 @@ model:
         image_size: 64    # not really needed
         in_channels: 20
         out_channels: 16
-        model_channels: 32 # TODO: more
+        model_channels: 192
         attention_resolutions: [ 4, 2, 1 ]
         num_res_blocks: 2
         channel_mult: [ 1, 2, 4, 4 ]
@@ -105,8 +105,8 @@ data:
   target: ldm.data.laion.WebDataModuleFromConfig
   params:
     tar_base: "pipe:aws s3 cp s3://s-datasets/laion-high-resolution/"
-    batch_size: 4
-    num_workers: 1
+    batch_size: 8
+    num_workers: 4
     train:
       shards: '{00000..17279}.tar -'
       shuffle: 10000
@@ -147,7 +147,7 @@ lightning:
     image_logger:
       target: main.ImageLogger
       params:
-        batch_frequency: 10
+        batch_frequency: 5000
         max_images: 4
         increase_log_steps: False
         log_first_step: False
@@ -164,4 +164,4 @@ lightning:
     benchmark: True
     val_check_interval: 5000000  # really sorry
     num_sanity_val_steps: 0
-    accumulate_grad_batches: 1
+    accumulate_grad_batches: 4