From 0eab5d03106327b1d597d9c73e5b8d68dd670f59 Mon Sep 17 00:00:00 2001 From: rromb Date: Tue, 31 May 2022 16:16:15 +0200 Subject: [PATCH] reduce batch size meh --- .../stable-diffusion/txt2img-1p4B-multinode-clip-encoder.yaml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/configs/stable-diffusion/txt2img-1p4B-multinode-clip-encoder.yaml b/configs/stable-diffusion/txt2img-1p4B-multinode-clip-encoder.yaml index 2a242f9..da87af4 100644 --- a/configs/stable-diffusion/txt2img-1p4B-multinode-clip-encoder.yaml +++ b/configs/stable-diffusion/txt2img-1p4B-multinode-clip-encoder.yaml @@ -74,7 +74,7 @@ data: target: ldm.data.laion.WebDataModuleFromConfig params: tar_base: "pipe:aws s3 cp s3://s-datasets/laion5b/laion2B-data/" - batch_size: 56 + batch_size: 50 num_workers: 4 multinode: True train: