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

update ae config

Browse source
This commit is contained in:
rromb 2022-06-10 12:02:22 +02:00
parent 5e3f9cec53
commit dd10f44cdb
1 changed files with 4 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
configs/stable-diffusion/txt2img-multinode-clip-encoder-f16-256-pretraining.yaml
View file

@ -45,23 +45,18 @@ model:
first_stage_config: first_stage_config:
target: ldm.models.autoencoder.AutoencoderKL target: ldm.models.autoencoder.AutoencoderKL
params: params:
embed_dim: 4 embed_dim: 16
monitor: val/rec_loss monitor: val/rec_loss
ckpt_path: "models/first_stage_models/kl-f16/model.ckpt" ckpt_path: "models/first_stage_models/kl-f16/model.ckpt"
ddconfig: ddconfig:
double_z: true z_channels: 16
z_channels: 4
resolution: 256 resolution: 256
in_channels: 3 in_channels: 3
out_ch: 3 out_ch: 3
ch: 128 ch: 128
ch_mult: ch_mult: [ 1,1,2,2,4 ] # num_down = len(ch_mult)-1
- 1
- 2
- 4
- 4
num_res_blocks: 2 num_res_blocks: 2
attn_resolutions: [] attn_resolutions: [ 16 ]
dropout: 0.0 dropout: 0.0
lossconfig: lossconfig:
target: torch.nn.Identity target: torch.nn.Identity