canberkb9 commited on
Commit
05f26f4
1 Parent(s): 2b32cdc
checkpoint-6000/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:99472668c44ad43974578ac9152a822c8ee8c58f8725b30f8b5b70974041e3aa
3
+ size 909800163
checkpoint-6000/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:618cb448cdea79073cd358e36978551f51352a4368e1dd3b0c2a47127694a0a8
3
+ size 14663
checkpoint-6000/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8ffffec83cb79e7c090c34eb26376f37f8b18579fd0937ce8c0fc513dfc14f07
3
+ size 563
checkpoint-6000/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": 5,
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": 5,
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-6000/unet/diffusion_pytorch_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f7c410e58478dbbe276b65a0499a553dc4964a7bde9547ec33797ce6f8050e81
3
+ size 454759548
checkpoint-6000/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": 5,
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": 6000,
39
+ "out_channels": 5,
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-6000/unet_ema/diffusion_pytorch_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c6694dda0607e8107e612d6b7fbf00b6781f34d786c1ca52e3e038a33acc603a
3
+ size 454759548
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": 5,
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": 5,
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": 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",
unet/diffusion_pytorch_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:55a3f0597b4bda8aea2734b12f8ec73b62a8c3c1e78ea5c10f179a3d7b53d2cb
3
- size 454759548
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ee1b24b3a12e48789829a1c4846e6ef8cb3a61089f9b6c5e2a479cab9105afab
3
+ size 454787208