antioom

2022-06-18 07:30:59 +00:00 · 2022-06-18 07:30:59 +00:00 · cace7f06aa
commit cace7f06aa
parent 9981c7209c
1 changed files with 1 additions and 1 deletions
--- 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: 8
+    batch_size: 6
    num_workers: 4
    multinode: True
    min_size: 384   # TODO: experiment. Note: for 2B, images are stored at max 384 resolution