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

Merge branch 'main' of github.com:pesser/stable-diffusion into main

Browse source
This commit is contained in:
rromb 2022-06-10 10:54:17 +02:00
commit 04bcc0f7de
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: