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 d324575..f901294 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: 8 + batch_size: 6 num_workers: 4 multinode: True min_size: 384 # TODO: experiment. Note: for 2B, images are stored at max 384 resolution