patrickvonplaten commited on
Commit
6ae21a8
•
1 Parent(s): 95752c1

update config

Browse files
config.json CHANGED
@@ -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
  }
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
unet/config.json DELETED
@@ -1,54 +0,0 @@
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,
16
- 256,
17
- 512,
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
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
unet/diffusion_model.pt DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:65835ead0bcaf5ba9f37d5920565018da0d2ff89ecd5ff22de8ee50a853e0696
3
- size 454858877