patrickvonplaten commited on
Commit
5d168b7
·
1 Parent(s): c5f3587
Files changed (2) hide show
  1. config.json +37 -0
  2. diffusion_model.pt +3 -0
config.json ADDED
@@ -0,0 +1,37 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "UNetUnconditionalModel",
3
+ "_diffusers_version": "0.0.4",
4
+ "attention_resolutions": [
5
+ 16
6
+ ],
7
+ "attn_resolutions": null,
8
+ "block_channels": [
9
+ 32,
10
+ 64
11
+ ],
12
+ "center_input_sample": false,
13
+ "conv_resample": true,
14
+ "ddpm": false,
15
+ "down_blocks": [
16
+ "UNetResDownBlock2D",
17
+ "UNetResDownBlock2D"
18
+ ],
19
+ "downsample_padding": 1,
20
+ "downscale_freq_shift": 0,
21
+ "dropout": 0,
22
+ "flip_sin_to_cos": true,
23
+ "image_size": 32,
24
+ "in_channels": 4,
25
+ "mid_block_scale_factor": 1,
26
+ "num_head_channels": 32,
27
+ "num_res_blocks": 2,
28
+ "out_channels": 4,
29
+ "resnet_eps": 1e-05,
30
+ "resnet_num_groups": 32,
31
+ "sde": false,
32
+ "time_embedding_type": "positional",
33
+ "up_blocks": [
34
+ "UNetResUpBlock2D",
35
+ "UNetResUpBlock2D"
36
+ ]
37
+ }
diffusion_model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:07285ded74bc8c3642e455828c38bca4089432082ad7f336da8e0262d703a91d
3
+ size 3734853