r/stable-diffusion-finetune
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

bs and unused params

Browse source
This commit is contained in:
Patrick Esser 2022-06-17 23:20:03 +00:00 committed by root
parent 12d2aaf50c
commit 9981c7209c
1 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
configs/stable-diffusion/txt2img-multinode-clip-encoder-f16-768.yaml
View file

@ -71,7 +71,7 @@ data:
target: ldm.data.laion.WebDataModuleFromConfig target: ldm.data.laion.WebDataModuleFromConfig
params: params:
tar_base: "pipe:aws s3 cp s3://s-datasets/laion5b/laion2B-data/" tar_base: "pipe:aws s3 cp s3://s-datasets/laion5b/laion2B-data/"
batch_size: 10 batch_size: 8
num_workers: 4 num_workers: 4
multinode: True multinode: True
min_size: 384 # TODO: experiment. Note: for 2B, images are stored at max 384 resolution min_size: 384 # TODO: experiment. Note: for 2B, images are stored at max 384 resolution
@ -104,6 +104,8 @@ data:
lightning: lightning:
find_unused_parameters: False
callbacks: callbacks:
image_logger: image_logger:
target: main.ImageLogger target: main.ImageLogger