diff --git a/configs/stable-diffusion/txt2img-upscale-clip-encoder-f16-1024.yaml b/configs/stable-diffusion/txt2img-upscale-clip-encoder-f16-1024.yaml index 2f1ab09..1dab4d2 100644 --- a/configs/stable-diffusion/txt2img-upscale-clip-encoder-f16-1024.yaml +++ b/configs/stable-diffusion/txt2img-upscale-clip-encoder-f16-1024.yaml @@ -66,7 +66,7 @@ model: image_size: 64 # not really needed in_channels: 20 out_channels: 16 - model_channels: 32 # TODO: more + model_channels: 192 attention_resolutions: [ 4, 2, 1 ] num_res_blocks: 2 channel_mult: [ 1, 2, 4, 4 ] @@ -105,8 +105,8 @@ data: target: ldm.data.laion.WebDataModuleFromConfig params: tar_base: "pipe:aws s3 cp s3://s-datasets/laion-high-resolution/" - batch_size: 4 - num_workers: 1 + batch_size: 8 + num_workers: 4 train: shards: '{00000..17279}.tar -' shuffle: 10000 @@ -147,7 +147,7 @@ lightning: image_logger: target: main.ImageLogger params: - batch_frequency: 10 + batch_frequency: 5000 max_images: 4 increase_log_steps: False log_first_step: False @@ -164,4 +164,4 @@ lightning: benchmark: True val_check_interval: 5000000 # really sorry num_sanity_val_steps: 0 - accumulate_grad_batches: 1 + accumulate_grad_batches: 4