patrickvonplaten commited on
Commit
65181bb
1 Parent(s): 83feed9

update config

Browse files
config.json CHANGED
@@ -1 +1,55 @@
1
- {"_class_name": "UNetModel", "attn_resolutions": [16], "down_blocks": ["UNetResDownBlock2D", "UNetResDownBlock2D", "UNetResDownBlock2D", "UNetResDownBlock2D", "UNetResAttnDownBlock2D", "UNetResDownBlock2D"], "up_blocks": ["UNetResUpBlock2D", "UNetResAttnUpBlock2D", "UNetResUpBlock2D", "UNetResUpBlock2D", "UNetResUpBlock2D", "UNetResUpBlock2D"], "conv_resample": true, "downsample_padding": 0, "num_head_channels": null, "ch": 128, "ch_mult": [1, 1, 2, 2, 4, 4], "block_channels": [128, 128, 256, 256, 512, 512], "resnet_eps": 1e-06, "flip_sin_to_cos": false, "downscale_freq_shift": 1, "dropout": 0.0, "in_channels": 3, "name_or_path": "./ddpm-lsun-church/", "num_res_blocks": 2, "out_ch": 3, "out_channels": 3, "resamp_with_conv": true, "resolution": 256, "image_size": 256}
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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,
16
+ 256,
17
+ 512,
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
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:a1296e7d066e7c3c973dc3f3a27c6adca3a981646ede449f127db566da9abcd5
3
- size 470185705
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:03f5ad26b7b4ff473211661292178fdc152e630ad0b9d980b51c95e51101e1f9
3
+ size 454858877