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

update 512 config

Browse source
This commit is contained in:
Patrick Esser 2022-06-09 23:58:52 +02:00
parent 7ee2fc4310
commit d87b27805f
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
configs/stable-diffusion/txt2img-1p4B-multinode-clip-encoder-high-res-512.yaml
View file

@ -2,6 +2,7 @@ model:
base_learning_rate: 1.0e-04 base_learning_rate: 1.0e-04
target: ldm.models.diffusion.ddpm.LatentDiffusion target: ldm.models.diffusion.ddpm.LatentDiffusion
params: params:
ckpt_path: "/home/mchorse/stable-diffusion-ckpts/256pretrain-2022-06-09.ckpt"
linear_start: 0.00085 linear_start: 0.00085
linear_end: 0.0120 linear_end: 0.0120
num_timesteps_cond: 1 num_timesteps_cond: 1
@ -47,7 +48,6 @@ model:
params: params:
embed_dim: 4 embed_dim: 4
monitor: val/rec_loss monitor: val/rec_loss
ckpt_path: "models/first_stage_models/kl-f8/model.ckpt"
ddconfig: ddconfig:
double_z: true double_z: true
z_channels: 4 z_channels: 4
@ -74,7 +74,7 @@ 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: 50 batch_size: 4
num_workers: 4 num_workers: 4
multinode: True multinode: True
train: train: