patrickvonplaten commited on
Commit
5e0ffdc
1 Parent(s): eb53667

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,
@@ -19,40 +19,40 @@
19
  256
20
  ],
21
  "center_input_sample": true,
22
- "conv_resample": true,
23
- "ddpm": false,
24
  "down_blocks": [
25
- "UNetResSkipDownBlock2D",
26
- "UNetResSkipDownBlock2D",
27
- "UNetResSkipDownBlock2D",
28
- "UNetResSkipDownBlock2D",
29
- "UNetResAttnSkipDownBlock2D",
30
- "UNetResSkipDownBlock2D",
31
- "UNetResSkipDownBlock2D"
32
  ],
33
  "downsample_padding": 1,
34
  "downscale_freq_shift": 0,
35
- "dropout": 0.0,
36
  "flip_sin_to_cos": true,
37
- "image_size": 256,
 
38
  "in_channels": 3,
39
- "ldm": false,
40
  "mid_block_scale_factor": 1.41421356237,
41
- "name_or_path": "fusing/celebahq_256-ncsnpp-ve",
42
- "num_head_channels": null,
43
- "num_res_blocks": 2,
 
44
  "out_channels": 3,
45
- "resnet_eps": 1e-06,
46
- "resnet_num_groups": null,
47
- "sde": false,
 
48
  "time_embedding_type": "fourier",
49
  "up_blocks": [
50
- "UNetResSkipUpBlock2D",
51
- "UNetResSkipUpBlock2D",
52
- "UNetResAttnSkipUpBlock2D",
53
- "UNetResSkipUpBlock2D",
54
- "UNetResSkipUpBlock2D",
55
- "UNetResSkipUpBlock2D",
56
- "UNetResSkipUpBlock2D"
57
  ]
58
  }
 
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,
 
19
  256
20
  ],
21
  "center_input_sample": true,
 
 
22
  "down_blocks": [
23
+ "SkipDownBlock2D",
24
+ "SkipDownBlock2D",
25
+ "SkipDownBlock2D",
26
+ "SkipDownBlock2D",
27
+ "AttnSkipDownBlock2D",
28
+ "SkipDownBlock2D",
29
+ "SkipDownBlock2D"
30
  ],
31
  "downsample_padding": 1,
32
  "downscale_freq_shift": 0,
 
33
  "flip_sin_to_cos": true,
34
+ "freq_shift": 0,
35
+ "image_size": null,
36
  "in_channels": 3,
37
+ "layers_per_block": 2,
38
  "mid_block_scale_factor": 1.41421356237,
39
+ "norm_eps": 1e-06,
40
+ "norm_num_groups": null,
41
+ "num_head_channels": 32,
42
+ "num_res_blocks": null,
43
  "out_channels": 3,
44
+ "resnet_act_fn": "silu",
45
+ "resnet_eps": 1e-05,
46
+ "resnet_num_groups": 32,
47
+ "sample_size": 256,
48
  "time_embedding_type": "fourier",
49
  "up_blocks": [
50
+ "SkipUpBlock2D",
51
+ "SkipUpBlock2D",
52
+ "AttnSkipUpBlock2D",
53
+ "SkipUpBlock2D",
54
+ "SkipUpBlock2D",
55
+ "SkipUpBlock2D",
56
+ "SkipUpBlock2D"
57
  ]
58
  }
diffusion_model.pt → diffusion_pytorch_model.bin RENAMED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:15c930d6831958fe22dff5a8d0704b64f6553fe98836f4e4cfc4dfc2196caac2
3
  size 262534071
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:74ef99631908256e592f05d3e71da8048bd28fc1e86e8e01f6f35798f2d6a285
3
  size 262534071
scheduler_config.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "VeSdeScheduler",
3
+ "_diffusers_version": "0.0.4",
4
+ "sampling_eps": 1e-05,
5
+ "sigma_max": 380,
6
+ "sigma_min": 0.01,
7
+ "snr": 0.075
8
+ }