patrickvonplaten commited on
Commit
df8b410
1 Parent(s): cbb77b3
model_index.json ADDED
@@ -0,0 +1,12 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "DDPMPipeline",
3
+ "_diffusers_version": "0.0.4",
4
+ "scheduler": [
5
+ "diffusers",
6
+ "DDPMScheduler"
7
+ ],
8
+ "unet": [
9
+ "diffusers",
10
+ "UNetUnconditionalModel"
11
+ ]
12
+ }
scheduler/scheduler_config.json ADDED
@@ -0,0 +1,12 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "DDPMScheduler",
3
+ "_diffusers_version": "0.0.4",
4
+ "beta_end": 0.02,
5
+ "beta_schedule": "linear",
6
+ "beta_start": 0.0001,
7
+ "clip_sample": true,
8
+ "num_train_timesteps": 1000,
9
+ "timestep_values": null,
10
+ "trained_betas": null,
11
+ "variance_type": "fixed_small"
12
+ }
unet/config.json ADDED
@@ -0,0 +1,54 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2c00e8716c1174cde3db21115cf41a9380bf06ffc1232f6aa1dc67f8ff6b0806
3
+ size 454858877