canberkb9 commited on
Commit
d6ad0ae
1 Parent(s): ffcc791
checkpoint-2000/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0b17d3caff7b8666069b56baab556a550039dd1960bf25022b43f54a89e6b0da
3
+ size 909855459
checkpoint-2000/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1473914b9eb632185662fed0c8ddef02a8a33ca2b575daad23af92d0271ba0ab
3
+ size 14727
checkpoint-2000/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:79d16e74604edb7a7ccabaaa01d9d43284dfa28b91285c71a373a5e9f7621754
3
+ size 563
checkpoint-2000/unet/config.json ADDED
@@ -0,0 +1,48 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "UNet2DModel",
3
+ "_diffusers_version": "0.21.0.dev0",
4
+ "act_fn": "silu",
5
+ "add_attention": true,
6
+ "attention_head_dim": 8,
7
+ "block_out_channels": [
8
+ 128,
9
+ 128,
10
+ 256,
11
+ 256,
12
+ 512,
13
+ 512
14
+ ],
15
+ "center_input_sample": false,
16
+ "class_embed_type": null,
17
+ "down_block_types": [
18
+ "DownBlock2D",
19
+ "DownBlock2D",
20
+ "DownBlock2D",
21
+ "DownBlock2D",
22
+ "AttnDownBlock2D",
23
+ "DownBlock2D"
24
+ ],
25
+ "downsample_padding": 1,
26
+ "downsample_type": "conv",
27
+ "flip_sin_to_cos": true,
28
+ "freq_shift": 0,
29
+ "in_channels": 8,
30
+ "layers_per_block": 2,
31
+ "mid_block_scale_factor": 1,
32
+ "norm_eps": 1e-05,
33
+ "norm_num_groups": 32,
34
+ "num_class_embeds": null,
35
+ "out_channels": 8,
36
+ "resnet_time_scale_shift": "default",
37
+ "sample_size": 32,
38
+ "time_embedding_type": "positional",
39
+ "up_block_types": [
40
+ "UpBlock2D",
41
+ "AttnUpBlock2D",
42
+ "UpBlock2D",
43
+ "UpBlock2D",
44
+ "UpBlock2D",
45
+ "UpBlock2D"
46
+ ],
47
+ "upsample_type": "conv"
48
+ }
checkpoint-2000/unet/diffusion_pytorch_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ae6fd255485006418fdf89ebc275e95f60e1019b1e89e5f0d844f83dcc2f1eef
3
+ size 454787208
checkpoint-2000/unet_ema/config.json ADDED
@@ -0,0 +1,55 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "UNet2DModel",
3
+ "_diffusers_version": "0.21.0.dev0",
4
+ "act_fn": "silu",
5
+ "add_attention": true,
6
+ "attention_head_dim": 8,
7
+ "block_out_channels": [
8
+ 128,
9
+ 128,
10
+ 256,
11
+ 256,
12
+ 512,
13
+ 512
14
+ ],
15
+ "center_input_sample": false,
16
+ "class_embed_type": null,
17
+ "decay": 0.9999,
18
+ "down_block_types": [
19
+ "DownBlock2D",
20
+ "DownBlock2D",
21
+ "DownBlock2D",
22
+ "DownBlock2D",
23
+ "AttnDownBlock2D",
24
+ "DownBlock2D"
25
+ ],
26
+ "downsample_padding": 1,
27
+ "downsample_type": "conv",
28
+ "flip_sin_to_cos": true,
29
+ "freq_shift": 0,
30
+ "in_channels": 8,
31
+ "inv_gamma": 1.0,
32
+ "layers_per_block": 2,
33
+ "mid_block_scale_factor": 1,
34
+ "min_decay": 0.0,
35
+ "norm_eps": 1e-05,
36
+ "norm_num_groups": 32,
37
+ "num_class_embeds": null,
38
+ "optimization_step": 2000,
39
+ "out_channels": 8,
40
+ "power": 0.75,
41
+ "resnet_time_scale_shift": "default",
42
+ "sample_size": 32,
43
+ "time_embedding_type": "positional",
44
+ "up_block_types": [
45
+ "UpBlock2D",
46
+ "AttnUpBlock2D",
47
+ "UpBlock2D",
48
+ "UpBlock2D",
49
+ "UpBlock2D",
50
+ "UpBlock2D"
51
+ ],
52
+ "update_after_step": 0,
53
+ "upsample_type": "conv",
54
+ "use_ema_warmup": true
55
+ }
checkpoint-2000/unet_ema/diffusion_pytorch_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5ff4932eebf1268460605f48e7bd8694dbd0f6894f433c0b2ab458c8ff4e9eec
3
+ size 454787208
checkpoint-4000/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a5ed3bad6ff1285676350a303829b9d117a82b90d4fd347e960036989fe752ea
3
+ size 909855459
checkpoint-4000/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7b0e5f97d79e63a98845bf54382ad02086f25220de5516705799a8b6033e8522
3
+ size 14727
checkpoint-4000/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c869f73638bdc17481d677dac81ab2b055c12be36db9616a079b0bc756e35f53
3
+ size 563
checkpoint-4000/unet/config.json ADDED
@@ -0,0 +1,48 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "UNet2DModel",
3
+ "_diffusers_version": "0.21.0.dev0",
4
+ "act_fn": "silu",
5
+ "add_attention": true,
6
+ "attention_head_dim": 8,
7
+ "block_out_channels": [
8
+ 128,
9
+ 128,
10
+ 256,
11
+ 256,
12
+ 512,
13
+ 512
14
+ ],
15
+ "center_input_sample": false,
16
+ "class_embed_type": null,
17
+ "down_block_types": [
18
+ "DownBlock2D",
19
+ "DownBlock2D",
20
+ "DownBlock2D",
21
+ "DownBlock2D",
22
+ "AttnDownBlock2D",
23
+ "DownBlock2D"
24
+ ],
25
+ "downsample_padding": 1,
26
+ "downsample_type": "conv",
27
+ "flip_sin_to_cos": true,
28
+ "freq_shift": 0,
29
+ "in_channels": 8,
30
+ "layers_per_block": 2,
31
+ "mid_block_scale_factor": 1,
32
+ "norm_eps": 1e-05,
33
+ "norm_num_groups": 32,
34
+ "num_class_embeds": null,
35
+ "out_channels": 8,
36
+ "resnet_time_scale_shift": "default",
37
+ "sample_size": 32,
38
+ "time_embedding_type": "positional",
39
+ "up_block_types": [
40
+ "UpBlock2D",
41
+ "AttnUpBlock2D",
42
+ "UpBlock2D",
43
+ "UpBlock2D",
44
+ "UpBlock2D",
45
+ "UpBlock2D"
46
+ ],
47
+ "upsample_type": "conv"
48
+ }
checkpoint-4000/unet/diffusion_pytorch_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b4fb8f14058929b93dacfbebef620ced10af3a997b9dcad43d5fe28967c10ec6
3
+ size 454787208
checkpoint-4000/unet_ema/config.json ADDED
@@ -0,0 +1,55 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "UNet2DModel",
3
+ "_diffusers_version": "0.21.0.dev0",
4
+ "act_fn": "silu",
5
+ "add_attention": true,
6
+ "attention_head_dim": 8,
7
+ "block_out_channels": [
8
+ 128,
9
+ 128,
10
+ 256,
11
+ 256,
12
+ 512,
13
+ 512
14
+ ],
15
+ "center_input_sample": false,
16
+ "class_embed_type": null,
17
+ "decay": 0.9999,
18
+ "down_block_types": [
19
+ "DownBlock2D",
20
+ "DownBlock2D",
21
+ "DownBlock2D",
22
+ "DownBlock2D",
23
+ "AttnDownBlock2D",
24
+ "DownBlock2D"
25
+ ],
26
+ "downsample_padding": 1,
27
+ "downsample_type": "conv",
28
+ "flip_sin_to_cos": true,
29
+ "freq_shift": 0,
30
+ "in_channels": 8,
31
+ "inv_gamma": 1.0,
32
+ "layers_per_block": 2,
33
+ "mid_block_scale_factor": 1,
34
+ "min_decay": 0.0,
35
+ "norm_eps": 1e-05,
36
+ "norm_num_groups": 32,
37
+ "num_class_embeds": null,
38
+ "optimization_step": 4000,
39
+ "out_channels": 8,
40
+ "power": 0.75,
41
+ "resnet_time_scale_shift": "default",
42
+ "sample_size": 32,
43
+ "time_embedding_type": "positional",
44
+ "up_block_types": [
45
+ "UpBlock2D",
46
+ "AttnUpBlock2D",
47
+ "UpBlock2D",
48
+ "UpBlock2D",
49
+ "UpBlock2D",
50
+ "UpBlock2D"
51
+ ],
52
+ "update_after_step": 0,
53
+ "upsample_type": "conv",
54
+ "use_ema_warmup": true
55
+ }
checkpoint-4000/unet_ema/diffusion_pytorch_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e6eeddfdb1ca3e411e3e876ba78d5131deec474cc2924f701b9af1948d2bbe97
3
+ size 454787208
unet/config.json CHANGED
@@ -26,13 +26,13 @@
26
  "downsample_type": "conv",
27
  "flip_sin_to_cos": true,
28
  "freq_shift": 0,
29
- "in_channels": 8,
30
  "layers_per_block": 2,
31
  "mid_block_scale_factor": 1,
32
  "norm_eps": 1e-05,
33
  "norm_num_groups": 32,
34
  "num_class_embeds": null,
35
- "out_channels": 8,
36
  "resnet_time_scale_shift": "default",
37
  "sample_size": 32,
38
  "time_embedding_type": "positional",
 
26
  "downsample_type": "conv",
27
  "flip_sin_to_cos": true,
28
  "freq_shift": 0,
29
+ "in_channels": 4,
30
  "layers_per_block": 2,
31
  "mid_block_scale_factor": 1,
32
  "norm_eps": 1e-05,
33
  "norm_num_groups": 32,
34
  "num_class_embeds": null,
35
+ "out_channels": 4,
36
  "resnet_time_scale_shift": "default",
37
  "sample_size": 32,
38
  "time_embedding_type": "positional",
unet/diffusion_pytorch_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:30179ffdf9302b4a878104d14da6f9366abff57b51c7b01ae8025066bf68ed03
3
- size 454787208
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1c19d0b66f812b28f09e0ce605dacb94c752e4effc9c3a8629799ae7f1f0c3f5
3
+ size 454750328