patrickvonplaten commited on
Commit
07fd2eb
1 Parent(s): 88faf78

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
  16,
14
  32,
15
  64,
@@ -20,42 +20,42 @@
20
  512
21
  ],
22
  "center_input_sample": true,
23
- "conv_resample": true,
24
- "ddpm": false,
25
  "down_blocks": [
26
- "UNetResSkipDownBlock2D",
27
- "UNetResSkipDownBlock2D",
28
- "UNetResSkipDownBlock2D",
29
- "UNetResSkipDownBlock2D",
30
- "UNetResSkipDownBlock2D",
31
- "UNetResSkipDownBlock2D",
32
- "UNetResAttnSkipDownBlock2D",
33
- "UNetResSkipDownBlock2D"
34
  ],
35
  "downsample_padding": 1,
36
  "downscale_freq_shift": 0,
37
- "dropout": 0.0,
38
  "flip_sin_to_cos": true,
39
- "image_size": 1024,
 
40
  "in_channels": 3,
41
- "ldm": false,
42
  "mid_block_scale_factor": 1.41421356237,
43
- "name_or_path": "fusing/ffhq_ncsnpp",
44
- "num_head_channels": null,
45
- "num_res_blocks": 1,
 
46
  "out_channels": 3,
47
- "resnet_eps": 1e-06,
48
- "resnet_num_groups": null,
49
- "sde": false,
 
50
  "time_embedding_type": "fourier",
51
  "up_blocks": [
52
- "UNetResSkipUpBlock2D",
53
- "UNetResAttnSkipUpBlock2D",
54
- "UNetResSkipUpBlock2D",
55
- "UNetResSkipUpBlock2D",
56
- "UNetResSkipUpBlock2D",
57
- "UNetResSkipUpBlock2D",
58
- "UNetResSkipUpBlock2D",
59
- "UNetResSkipUpBlock2D"
60
  ]
61
  }
 
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
  16,
14
  32,
15
  64,
 
20
  512
21
  ],
22
  "center_input_sample": true,
 
 
23
  "down_blocks": [
24
+ "SkipDownBlock2D",
25
+ "SkipDownBlock2D",
26
+ "SkipDownBlock2D",
27
+ "SkipDownBlock2D",
28
+ "SkipDownBlock2D",
29
+ "SkipDownBlock2D",
30
+ "AttnSkipDownBlock2D",
31
+ "SkipDownBlock2D"
32
  ],
33
  "downsample_padding": 1,
34
  "downscale_freq_shift": 0,
 
35
  "flip_sin_to_cos": true,
36
+ "freq_shift": 0,
37
+ "image_size": null,
38
  "in_channels": 3,
39
+ "layers_per_block": 1,
40
  "mid_block_scale_factor": 1.41421356237,
41
+ "norm_eps": 1e-06,
42
+ "norm_num_groups": null,
43
+ "num_head_channels": 32,
44
+ "num_res_blocks": null,
45
  "out_channels": 3,
46
+ "resnet_act_fn": "silu",
47
+ "resnet_eps": 1e-05,
48
+ "resnet_num_groups": 32,
49
+ "sample_size": 1024,
50
  "time_embedding_type": "fourier",
51
  "up_blocks": [
52
+ "SkipUpBlock2D",
53
+ "AttnSkipUpBlock2D",
54
+ "SkipUpBlock2D",
55
+ "SkipUpBlock2D",
56
+ "SkipUpBlock2D",
57
+ "SkipUpBlock2D",
58
+ "SkipUpBlock2D",
59
+ "SkipUpBlock2D"
60
  ]
61
  }
diffusion_model.pt → diffusion_pytorch_model.bin RENAMED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:0d6d36a6e1bfa9deb80a8bc4d5976f1c94294812ed1780c6fbb01a0c19f58f18
3
  size 423345519
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:da9695f0e2be79b855c526edb4dcb99e5e25013738a416e04848ebc13647c6c6
3
  size 423345519
model_index.json CHANGED
@@ -3,7 +3,7 @@
3
  "_diffusers_version": "0.0.4",
4
  "model": [
5
  "diffusers",
6
- "UNetUnconditionalModel"
7
  ],
8
  "scheduler": [
9
  "diffusers",
 
3
  "_diffusers_version": "0.0.4",
4
  "model": [
5
  "diffusers",
6
+ "UNet2DModel"
7
  ],
8
  "scheduler": [
9
  "diffusers",