diff --git a/configs/stable-diffusion/txt2img-multinode-clip-encoder-f16-768.yaml b/configs/stable-diffusion/txt2img-multinode-clip-encoder-f16-768.yaml
index 707eda9..d324575 100644
--- a/configs/stable-diffusion/txt2img-multinode-clip-encoder-f16-768.yaml
+++ b/configs/stable-diffusion/txt2img-multinode-clip-encoder-f16-768.yaml
@@ -71,7 +71,7 @@ data:
   target: ldm.data.laion.WebDataModuleFromConfig
   params:
     tar_base: "pipe:aws s3 cp s3://s-datasets/laion5b/laion2B-data/"
-    batch_size: 10
+    batch_size: 8
     num_workers: 4
     multinode: True
     min_size: 384   # TODO: experiment. Note: for 2B, images are stored at max 384 resolution
@@ -104,6 +104,8 @@ data:
 
 
 lightning:
+  find_unused_parameters: False
+
   callbacks:
     image_logger:
       target: main.ImageLogger