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

finalize upscaler conf

Browse source
This commit is contained in:
Patrick Esser 2022-06-13 11:12:22 +00:00 committed by root
parent 6f604acf0a
commit 8ff9bdb78f
1 changed files with 5 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
configs/stable-diffusion/txt2img-upscale-clip-encoder-f16-1024.yaml
View file

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