-
Notifications
You must be signed in to change notification settings - Fork 0
/
config.yaml
36 lines (36 loc) · 733 Bytes
/
config.yaml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
_class_name: UNet2DConditionModel
_diffusers_version: 0.9.0.dev0
_name_or_path: /content/stable-diffusion-custom
act_fn: silu
attention_head_dim: 8
block_out_channels:
- 320
- 640
- 1280
- 1280
center_input_sample: false
cross_attention_dim: 768
down_block_types:
- CrossAttnDownBlock2D
- CrossAttnDownBlock2D
- CrossAttnDownBlock2D
- DownBlock2D
downsample_padding: 1
dual_cross_attention: false
flip_sin_to_cos: true
freq_shift: 0
in_channels: 4
layers_per_block: 2
mid_block_scale_factor: 1
norm_eps: 1e-05
norm_num_groups: 32
num_class_embeds: null
only_cross_attention: false
out_channels: 4
sample_size: 32
up_block_types:
- UpBlock2D
- CrossAttnUpBlock2D
- CrossAttnUpBlock2D
- CrossAttnUpBlock2D
use_linear_projection: false