evison commited on
Commit
7fa698a
·
verified ·
1 Parent(s): cf15c39

Upload config.json

Browse files
Files changed (1) hide show
  1. config.json +32 -27
config.json CHANGED
@@ -1,32 +1,37 @@
1
  {
2
- "_class_name": "StableDiffusionPipeline",
3
- "_diffusers_version": "0.6.0",
4
- "feature_extractor": [
5
- "transformers",
6
- "CLIPImageProcessor"
 
 
 
 
7
  ],
8
- "safety_checker": [
9
- "stable_diffusion",
10
- "StableDiffusionSafetyChecker"
 
 
 
 
11
  ],
12
- "scheduler": [
13
- "diffusers",
14
- "PNDMScheduler"
 
 
 
 
 
 
 
 
 
 
 
15
  ],
16
- "text_encoder": [
17
- "transformers",
18
- "CLIPTextModel"
19
- ],
20
- "tokenizer": [
21
- "transformers",
22
- "CLIPTokenizer"
23
- ],
24
- "unet": [
25
- "diffusers",
26
- "UNet2DConditionModel"
27
- ],
28
- "vae": [
29
- "diffusers",
30
- "AutoencoderKL"
31
- ]
32
  }
 
1
  {
2
+ "_class_name": "UNet2DConditionModel",
3
+ "_diffusers_version": "0.10.2",
4
+ "act_fn": "silu",
5
+ "attention_head_dim": 64,
6
+ "block_out_channels": [
7
+ 128,
8
+ 256,
9
+ 512,
10
+ 1024
11
  ],
12
+ "center_input_sample": false,
13
+ "cross_attention_dim": 2048,
14
+ "down_block_types": [
15
+ "CrossAttnDownBlock2D",
16
+ "CrossAttnDownBlock2D",
17
+ "CrossAttnDownBlock2D",
18
+ "DownBlock2D"
19
  ],
20
+ "dropout": 0.0,
21
+ "flip_sin_to_cos": true,
22
+ "freq_shift": 0,
23
+ "in_channels": 4,
24
+ "layers_per_block": 2,
25
+ "norm_eps": 1e-05,
26
+ "norm_num_groups": 32,
27
+ "out_channels": 4,
28
+ "sample_size": 1024,
29
+ "up_block_types": [
30
+ "UpBlock2D",
31
+ "CrossAttnUpBlock2D",
32
+ "CrossAttnUpBlock2D",
33
+ "CrossAttnUpBlock2D"
34
  ],
35
+ "use_linear_projection": true,
36
+ "use_memory_efficient_attention": true
 
 
 
 
 
 
 
 
 
 
 
 
 
 
37
  }