patrickvonplaten commited on
Commit
2f36b66
β€’
1 Parent(s): 24473e6

update config

Browse files
unet/config.json β†’ config.json RENAMED
@@ -1,15 +1,15 @@
1
  {
2
- "_class_name": "UNetUnconditionalModel",
3
  "_diffusers_version": "0.0.4",
4
- "attention_resolutions": [
5
- 8,
6
- 4,
7
- 2
8
- ],
9
- "attn_resolutions": [
10
- 16
11
- ],
12
  "block_channels": [
 
 
 
 
 
 
13
  128,
14
  128,
15
  256,
@@ -18,37 +18,38 @@
18
  512
19
  ],
20
  "center_input_sample": false,
21
- "conv_resample": true,
22
- "ddpm": false,
23
  "down_blocks": [
24
- "UNetResDownBlock2D",
25
- "UNetResDownBlock2D",
26
- "UNetResDownBlock2D",
27
- "UNetResDownBlock2D",
28
- "UNetResAttnDownBlock2D",
29
- "UNetResDownBlock2D"
30
  ],
31
  "downsample_padding": 0,
32
- "downscale_freq_shift": 1,
33
- "dropout": 0.0,
34
  "flip_sin_to_cos": false,
35
- "image_size": 256,
 
36
  "in_channels": 3,
37
- "ldm": false,
38
  "mid_block_scale_factor": 1,
39
- "num_head_channels": null,
40
- "num_res_blocks": 2,
 
 
41
  "out_channels": 3,
42
- "resnet_eps": 1e-06,
 
43
  "resnet_num_groups": 32,
44
- "sde": false,
45
  "time_embedding_type": "positional",
46
  "up_blocks": [
47
- "UNetResUpBlock2D",
48
- "UNetResAttnUpBlock2D",
49
- "UNetResUpBlock2D",
50
- "UNetResUpBlock2D",
51
- "UNetResUpBlock2D",
52
- "UNetResUpBlock2D"
53
  ]
54
  }
 
1
  {
2
+ "_class_name": "UNet2DModel",
3
  "_diffusers_version": "0.0.4",
4
+ "act_fn": "silu",
5
+ "attention_head_dim": null,
 
 
 
 
 
 
6
  "block_channels": [
7
+ 224,
8
+ 448,
9
+ 672,
10
+ 896
11
+ ],
12
+ "block_out_channels": [
13
  128,
14
  128,
15
  256,
 
18
  512
19
  ],
20
  "center_input_sample": false,
 
 
21
  "down_blocks": [
22
+ "DownBlock2D",
23
+ "DownBlock2D",
24
+ "DownBlock2D",
25
+ "DownBlock2D",
26
+ "AttnDownBlock2D",
27
+ "DownBlock2D"
28
  ],
29
  "downsample_padding": 0,
30
+ "downscale_freq_shift": 0,
 
31
  "flip_sin_to_cos": false,
32
+ "freq_shift": 1,
33
+ "image_size": null,
34
  "in_channels": 3,
35
+ "layers_per_block": 2,
36
  "mid_block_scale_factor": 1,
37
+ "norm_eps": 1e-06,
38
+ "norm_num_groups": 32,
39
+ "num_head_channels": 32,
40
+ "num_res_blocks": null,
41
  "out_channels": 3,
42
+ "resnet_act_fn": "silu",
43
+ "resnet_eps": 1e-05,
44
  "resnet_num_groups": 32,
45
+ "sample_size": 256,
46
  "time_embedding_type": "positional",
47
  "up_blocks": [
48
+ "UpBlock2D",
49
+ "AttnUpBlock2D",
50
+ "UpBlock2D",
51
+ "UpBlock2D",
52
+ "UpBlock2D",
53
+ "UpBlock2D"
54
  ]
55
  }
unet/diffusion_model.pt β†’ diffusion_pytorch_model.bin RENAMED
File without changes
model_index.json CHANGED
@@ -7,6 +7,6 @@
7
  ],
8
  "unet": [
9
  "diffusers",
10
- "UNetUnconditionalModel"
11
  ]
12
  }
 
7
  ],
8
  "unet": [
9
  "diffusers",
10
+ "UNet2DModel"
11
  ]
12
  }
scheduler/scheduler_config.json β†’ scheduler_config.json RENAMED
File without changes