From cace7f06aab36adf1d70d260cce4aad38ed114d6 Mon Sep 17 00:00:00 2001 From: Patrick Esser Date: Sat, 18 Jun 2022 07:30:59 +0000 Subject: [PATCH] antioom --- .../txt2img-multinode-clip-encoder-f16-768.yaml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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