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

mindev

Browse source
This commit is contained in:
pesser 2022-05-29 20:52:31 +00:00
parent add2812306
commit ddb9ce74f1
1 changed files with 7 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
configs/stable-diffusion/dev.yaml
View file

@ -31,16 +31,16 @@ model:
image_size: 32 image_size: 32
in_channels: 4 in_channels: 4
out_channels: 4 out_channels: 4
model_channels: 128 # 320 # TODO increase model_channels: 32 # 320 # TODO increase
attention_resolutions: [ 4, 2, 1 ] # is equal to fixed spatial resolution: 32 , 16 , 8 attention_resolutions: [ ] # is equal to fixed spatial resolution: 32 , 16 , 8
num_res_blocks: 2 num_res_blocks: 2
channel_mult: [ 1,2,4,4 ] channel_mult: [ 1, ]
#num_head_channels: 32 #num_head_channels: 32
num_heads: 8 num_heads: 8
use_spatial_transformer: True use_spatial_transformer: True
transformer_depth: 1 transformer_depth: 1
context_dim: 1280 context_dim: 32
use_checkpoint: True use_checkpoint: False
first_stage_config: first_stage_config:
target: ldm.models.autoencoder.AutoencoderKL target: ldm.models.autoencoder.AutoencoderKL
@ -69,8 +69,8 @@ model:
cond_stage_config: cond_stage_config:
target: ldm.modules.encoders.modules.BERTEmbedder target: ldm.modules.encoders.modules.BERTEmbedder
params: params:
n_embed: 1280 n_embed: 32
n_layer: 3 #32 # TODO: increase n_layer: 1 #32 # TODO: increase
data: data: