Mahmoud7 commited on
Commit
2730869
·
1 Parent(s): 070a85b
Files changed (49) hide show
  1. checkpoint-93500/optimizer.bin +3 -0
  2. checkpoint-93500/random_states_0.pkl +3 -0
  3. checkpoint-93500/scaler.pt +3 -0
  4. checkpoint-93500/scheduler.bin +3 -0
  5. checkpoint-93500/unet/config.json +60 -0
  6. checkpoint-93500/unet/diffusion_pytorch_model.bin +3 -0
  7. checkpoint-93500/unet_ema/config.json +67 -0
  8. checkpoint-93500/unet_ema/diffusion_pytorch_model.bin +3 -0
  9. checkpoint-94000/optimizer.bin +3 -0
  10. checkpoint-94000/random_states_0.pkl +3 -0
  11. checkpoint-94000/scaler.pt +3 -0
  12. checkpoint-94000/scheduler.bin +3 -0
  13. checkpoint-94000/unet/config.json +60 -0
  14. checkpoint-94000/unet/diffusion_pytorch_model.bin +3 -0
  15. checkpoint-94000/unet_ema/config.json +67 -0
  16. checkpoint-94000/unet_ema/diffusion_pytorch_model.bin +3 -0
  17. checkpoint-94500/optimizer.bin +3 -0
  18. checkpoint-94500/random_states_0.pkl +3 -0
  19. checkpoint-94500/scaler.pt +3 -0
  20. checkpoint-94500/scheduler.bin +3 -0
  21. checkpoint-94500/unet/config.json +60 -0
  22. checkpoint-94500/unet/diffusion_pytorch_model.bin +3 -0
  23. checkpoint-94500/unet_ema/config.json +67 -0
  24. checkpoint-94500/unet_ema/diffusion_pytorch_model.bin +3 -0
  25. checkpoint-95000/optimizer.bin +3 -0
  26. checkpoint-95000/random_states_0.pkl +3 -0
  27. checkpoint-95000/scaler.pt +3 -0
  28. checkpoint-95000/scheduler.bin +3 -0
  29. checkpoint-95000/unet/config.json +60 -0
  30. checkpoint-95000/unet/diffusion_pytorch_model.bin +3 -0
  31. checkpoint-95000/unet_ema/config.json +67 -0
  32. checkpoint-95000/unet_ema/diffusion_pytorch_model.bin +3 -0
  33. checkpoint-95500/optimizer.bin +3 -0
  34. checkpoint-95500/random_states_0.pkl +3 -0
  35. checkpoint-95500/scaler.pt +3 -0
  36. checkpoint-95500/scheduler.bin +3 -0
  37. checkpoint-95500/unet/config.json +60 -0
  38. checkpoint-95500/unet/diffusion_pytorch_model.bin +3 -0
  39. checkpoint-95500/unet_ema/config.json +67 -0
  40. checkpoint-95500/unet_ema/diffusion_pytorch_model.bin +3 -0
  41. checkpoint-96000/optimizer.bin +3 -0
  42. checkpoint-96000/random_states_0.pkl +3 -0
  43. checkpoint-96000/scaler.pt +3 -0
  44. checkpoint-96000/scheduler.bin +3 -0
  45. checkpoint-96000/unet/config.json +60 -0
  46. checkpoint-96000/unet/diffusion_pytorch_model.bin +3 -0
  47. checkpoint-96000/unet_ema/config.json +67 -0
  48. checkpoint-96000/unet_ema/diffusion_pytorch_model.bin +3 -0
  49. unet/diffusion_pytorch_model.bin +1 -1
checkpoint-93500/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6eea9e5a0ae752ce405be82fdd4b8d416f32ad496daf4920ab23abbedcf38a39
3
+ size 73251413
checkpoint-93500/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:badee0a929ad7db67ff75acb3163d43c868030d2b1cd1f928caacab03f1d1781
3
+ size 14663
checkpoint-93500/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:958f7ceb3693c8be69d45896fd74cfafff857a58061dfe9869ab5597580bc794
3
+ size 557
checkpoint-93500/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:abdd2549d23f1d7b3db53684839df8e1ecf16cec7be31b9fe5d6a1a654ada362
3
+ size 563
checkpoint-93500/unet/config.json ADDED
@@ -0,0 +1,60 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "UNet2DConditionModel",
3
+ "_diffusers_version": "0.16.1",
4
+ "act_fn": "silu",
5
+ "addition_embed_type": null,
6
+ "addition_embed_type_num_heads": 64,
7
+ "attention_head_dim": 8,
8
+ "block_out_channels": [
9
+ 64,
10
+ 64,
11
+ 128,
12
+ 128
13
+ ],
14
+ "center_input_sample": false,
15
+ "class_embed_type": null,
16
+ "class_embeddings_concat": false,
17
+ "conv_in_kernel": 3,
18
+ "conv_out_kernel": 3,
19
+ "cross_attention_dim": 256,
20
+ "cross_attention_norm": null,
21
+ "down_block_types": [
22
+ "DownBlock2D",
23
+ "DownBlock2D",
24
+ "DownBlock2D",
25
+ "CrossAttnDownBlock2D"
26
+ ],
27
+ "downsample_padding": 1,
28
+ "dual_cross_attention": false,
29
+ "encoder_hid_dim": null,
30
+ "flip_sin_to_cos": true,
31
+ "freq_shift": 0,
32
+ "in_channels": 3,
33
+ "layers_per_block": 2,
34
+ "mid_block_only_cross_attention": null,
35
+ "mid_block_scale_factor": 1,
36
+ "mid_block_type": "UNetMidBlock2DCrossAttn",
37
+ "norm_eps": 1e-05,
38
+ "norm_num_groups": 32,
39
+ "num_class_embeds": null,
40
+ "only_cross_attention": false,
41
+ "out_channels": 3,
42
+ "projection_class_embeddings_input_dim": null,
43
+ "resnet_out_scale_factor": 1.0,
44
+ "resnet_skip_time_act": false,
45
+ "resnet_time_scale_shift": "default",
46
+ "sample_size": 256,
47
+ "time_cond_proj_dim": null,
48
+ "time_embedding_act_fn": null,
49
+ "time_embedding_dim": null,
50
+ "time_embedding_type": "positional",
51
+ "timestep_post_act": null,
52
+ "up_block_types": [
53
+ "CrossAttnUpBlock2D",
54
+ "UpBlock2D",
55
+ "UpBlock2D",
56
+ "UpBlock2D"
57
+ ],
58
+ "upcast_attention": false,
59
+ "use_linear_projection": false
60
+ }
checkpoint-93500/unet/diffusion_pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:095af4df93e3830512f09fb2f10a2c5120c9909a6623432d643d8f2bcaed12f2
3
+ size 36621969
checkpoint-93500/unet_ema/config.json ADDED
@@ -0,0 +1,67 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "UNet2DConditionModel",
3
+ "_diffusers_version": "0.16.1",
4
+ "act_fn": "silu",
5
+ "addition_embed_type": null,
6
+ "addition_embed_type_num_heads": 64,
7
+ "attention_head_dim": 8,
8
+ "block_out_channels": [
9
+ 64,
10
+ 64,
11
+ 128,
12
+ 128
13
+ ],
14
+ "center_input_sample": false,
15
+ "class_embed_type": null,
16
+ "class_embeddings_concat": false,
17
+ "conv_in_kernel": 3,
18
+ "conv_out_kernel": 3,
19
+ "cross_attention_dim": 256,
20
+ "cross_attention_norm": null,
21
+ "decay": 0.9999,
22
+ "down_block_types": [
23
+ "DownBlock2D",
24
+ "DownBlock2D",
25
+ "DownBlock2D",
26
+ "CrossAttnDownBlock2D"
27
+ ],
28
+ "downsample_padding": 1,
29
+ "dual_cross_attention": false,
30
+ "encoder_hid_dim": null,
31
+ "flip_sin_to_cos": true,
32
+ "freq_shift": 0,
33
+ "in_channels": 3,
34
+ "inv_gamma": 1.0,
35
+ "layers_per_block": 2,
36
+ "mid_block_only_cross_attention": null,
37
+ "mid_block_scale_factor": 1,
38
+ "mid_block_type": "UNetMidBlock2DCrossAttn",
39
+ "min_decay": 0.0,
40
+ "norm_eps": 1e-05,
41
+ "norm_num_groups": 32,
42
+ "num_class_embeds": null,
43
+ "only_cross_attention": false,
44
+ "optimization_step": 93500,
45
+ "out_channels": 3,
46
+ "power": 0.75,
47
+ "projection_class_embeddings_input_dim": null,
48
+ "resnet_out_scale_factor": 1.0,
49
+ "resnet_skip_time_act": false,
50
+ "resnet_time_scale_shift": "default",
51
+ "sample_size": 256,
52
+ "time_cond_proj_dim": null,
53
+ "time_embedding_act_fn": null,
54
+ "time_embedding_dim": null,
55
+ "time_embedding_type": "positional",
56
+ "timestep_post_act": null,
57
+ "up_block_types": [
58
+ "CrossAttnUpBlock2D",
59
+ "UpBlock2D",
60
+ "UpBlock2D",
61
+ "UpBlock2D"
62
+ ],
63
+ "upcast_attention": false,
64
+ "update_after_step": 0,
65
+ "use_ema_warmup": true,
66
+ "use_linear_projection": false
67
+ }
checkpoint-93500/unet_ema/diffusion_pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:89e4f366590eea2e445859b7fe69362759773071b0c0937d48a4b828f27e56a1
3
+ size 36616081
checkpoint-94000/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:35d9bb4b3485f6914b5cb33931efd1deab6282de1e51946387edb72b42712987
3
+ size 73251413
checkpoint-94000/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dde997bbb18848fe50adac090f995e4c5c60d1cee2c4e7855ed3e997ade762f5
3
+ size 14663
checkpoint-94000/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:99c77801b5fb7f6af45e35ed6adb472130c259a769e62c59d7f4c03eebea926f
3
+ size 557
checkpoint-94000/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f4a70174cc3de997f45bed589bcea62050d80a9bdf39f767750738fdeb00b991
3
+ size 563
checkpoint-94000/unet/config.json ADDED
@@ -0,0 +1,60 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "UNet2DConditionModel",
3
+ "_diffusers_version": "0.16.1",
4
+ "act_fn": "silu",
5
+ "addition_embed_type": null,
6
+ "addition_embed_type_num_heads": 64,
7
+ "attention_head_dim": 8,
8
+ "block_out_channels": [
9
+ 64,
10
+ 64,
11
+ 128,
12
+ 128
13
+ ],
14
+ "center_input_sample": false,
15
+ "class_embed_type": null,
16
+ "class_embeddings_concat": false,
17
+ "conv_in_kernel": 3,
18
+ "conv_out_kernel": 3,
19
+ "cross_attention_dim": 256,
20
+ "cross_attention_norm": null,
21
+ "down_block_types": [
22
+ "DownBlock2D",
23
+ "DownBlock2D",
24
+ "DownBlock2D",
25
+ "CrossAttnDownBlock2D"
26
+ ],
27
+ "downsample_padding": 1,
28
+ "dual_cross_attention": false,
29
+ "encoder_hid_dim": null,
30
+ "flip_sin_to_cos": true,
31
+ "freq_shift": 0,
32
+ "in_channels": 3,
33
+ "layers_per_block": 2,
34
+ "mid_block_only_cross_attention": null,
35
+ "mid_block_scale_factor": 1,
36
+ "mid_block_type": "UNetMidBlock2DCrossAttn",
37
+ "norm_eps": 1e-05,
38
+ "norm_num_groups": 32,
39
+ "num_class_embeds": null,
40
+ "only_cross_attention": false,
41
+ "out_channels": 3,
42
+ "projection_class_embeddings_input_dim": null,
43
+ "resnet_out_scale_factor": 1.0,
44
+ "resnet_skip_time_act": false,
45
+ "resnet_time_scale_shift": "default",
46
+ "sample_size": 256,
47
+ "time_cond_proj_dim": null,
48
+ "time_embedding_act_fn": null,
49
+ "time_embedding_dim": null,
50
+ "time_embedding_type": "positional",
51
+ "timestep_post_act": null,
52
+ "up_block_types": [
53
+ "CrossAttnUpBlock2D",
54
+ "UpBlock2D",
55
+ "UpBlock2D",
56
+ "UpBlock2D"
57
+ ],
58
+ "upcast_attention": false,
59
+ "use_linear_projection": false
60
+ }
checkpoint-94000/unet/diffusion_pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4d67cb7b7d8be80687905c54b0aed4a2f5b7d0b138d9cd6c248a4c92ed81ed65
3
+ size 36621969
checkpoint-94000/unet_ema/config.json ADDED
@@ -0,0 +1,67 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "UNet2DConditionModel",
3
+ "_diffusers_version": "0.16.1",
4
+ "act_fn": "silu",
5
+ "addition_embed_type": null,
6
+ "addition_embed_type_num_heads": 64,
7
+ "attention_head_dim": 8,
8
+ "block_out_channels": [
9
+ 64,
10
+ 64,
11
+ 128,
12
+ 128
13
+ ],
14
+ "center_input_sample": false,
15
+ "class_embed_type": null,
16
+ "class_embeddings_concat": false,
17
+ "conv_in_kernel": 3,
18
+ "conv_out_kernel": 3,
19
+ "cross_attention_dim": 256,
20
+ "cross_attention_norm": null,
21
+ "decay": 0.9999,
22
+ "down_block_types": [
23
+ "DownBlock2D",
24
+ "DownBlock2D",
25
+ "DownBlock2D",
26
+ "CrossAttnDownBlock2D"
27
+ ],
28
+ "downsample_padding": 1,
29
+ "dual_cross_attention": false,
30
+ "encoder_hid_dim": null,
31
+ "flip_sin_to_cos": true,
32
+ "freq_shift": 0,
33
+ "in_channels": 3,
34
+ "inv_gamma": 1.0,
35
+ "layers_per_block": 2,
36
+ "mid_block_only_cross_attention": null,
37
+ "mid_block_scale_factor": 1,
38
+ "mid_block_type": "UNetMidBlock2DCrossAttn",
39
+ "min_decay": 0.0,
40
+ "norm_eps": 1e-05,
41
+ "norm_num_groups": 32,
42
+ "num_class_embeds": null,
43
+ "only_cross_attention": false,
44
+ "optimization_step": 94000,
45
+ "out_channels": 3,
46
+ "power": 0.75,
47
+ "projection_class_embeddings_input_dim": null,
48
+ "resnet_out_scale_factor": 1.0,
49
+ "resnet_skip_time_act": false,
50
+ "resnet_time_scale_shift": "default",
51
+ "sample_size": 256,
52
+ "time_cond_proj_dim": null,
53
+ "time_embedding_act_fn": null,
54
+ "time_embedding_dim": null,
55
+ "time_embedding_type": "positional",
56
+ "timestep_post_act": null,
57
+ "up_block_types": [
58
+ "CrossAttnUpBlock2D",
59
+ "UpBlock2D",
60
+ "UpBlock2D",
61
+ "UpBlock2D"
62
+ ],
63
+ "upcast_attention": false,
64
+ "update_after_step": 0,
65
+ "use_ema_warmup": true,
66
+ "use_linear_projection": false
67
+ }
checkpoint-94000/unet_ema/diffusion_pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:628b7009137879c67ff50916d05c727260c9c8eb98da11e09b5158fad61b9512
3
+ size 36616081
checkpoint-94500/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:76ab9a31e88c2bb03ace2fca47b5b5a2bcf92165286a458f7dd0010562653f8a
3
+ size 73251413
checkpoint-94500/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4edb0bc53343cf673251a9e6f2a3f73e2e990b202a229315937a9bf281a0d017
3
+ size 14663
checkpoint-94500/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4118e3129796bc34691777743d68480a43bb4e4c8b1a87a4181a1f27702e22df
3
+ size 557
checkpoint-94500/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0ef56c0f7346b08127e592bc901bbec0d69412f0b37d28039f7b7e81f101bc74
3
+ size 563
checkpoint-94500/unet/config.json ADDED
@@ -0,0 +1,60 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "UNet2DConditionModel",
3
+ "_diffusers_version": "0.16.1",
4
+ "act_fn": "silu",
5
+ "addition_embed_type": null,
6
+ "addition_embed_type_num_heads": 64,
7
+ "attention_head_dim": 8,
8
+ "block_out_channels": [
9
+ 64,
10
+ 64,
11
+ 128,
12
+ 128
13
+ ],
14
+ "center_input_sample": false,
15
+ "class_embed_type": null,
16
+ "class_embeddings_concat": false,
17
+ "conv_in_kernel": 3,
18
+ "conv_out_kernel": 3,
19
+ "cross_attention_dim": 256,
20
+ "cross_attention_norm": null,
21
+ "down_block_types": [
22
+ "DownBlock2D",
23
+ "DownBlock2D",
24
+ "DownBlock2D",
25
+ "CrossAttnDownBlock2D"
26
+ ],
27
+ "downsample_padding": 1,
28
+ "dual_cross_attention": false,
29
+ "encoder_hid_dim": null,
30
+ "flip_sin_to_cos": true,
31
+ "freq_shift": 0,
32
+ "in_channels": 3,
33
+ "layers_per_block": 2,
34
+ "mid_block_only_cross_attention": null,
35
+ "mid_block_scale_factor": 1,
36
+ "mid_block_type": "UNetMidBlock2DCrossAttn",
37
+ "norm_eps": 1e-05,
38
+ "norm_num_groups": 32,
39
+ "num_class_embeds": null,
40
+ "only_cross_attention": false,
41
+ "out_channels": 3,
42
+ "projection_class_embeddings_input_dim": null,
43
+ "resnet_out_scale_factor": 1.0,
44
+ "resnet_skip_time_act": false,
45
+ "resnet_time_scale_shift": "default",
46
+ "sample_size": 256,
47
+ "time_cond_proj_dim": null,
48
+ "time_embedding_act_fn": null,
49
+ "time_embedding_dim": null,
50
+ "time_embedding_type": "positional",
51
+ "timestep_post_act": null,
52
+ "up_block_types": [
53
+ "CrossAttnUpBlock2D",
54
+ "UpBlock2D",
55
+ "UpBlock2D",
56
+ "UpBlock2D"
57
+ ],
58
+ "upcast_attention": false,
59
+ "use_linear_projection": false
60
+ }
checkpoint-94500/unet/diffusion_pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c72f636932a7333b28165482441c1a33fd1c229ca34c07e2805bc2e43d27eb36
3
+ size 36621969
checkpoint-94500/unet_ema/config.json ADDED
@@ -0,0 +1,67 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "UNet2DConditionModel",
3
+ "_diffusers_version": "0.16.1",
4
+ "act_fn": "silu",
5
+ "addition_embed_type": null,
6
+ "addition_embed_type_num_heads": 64,
7
+ "attention_head_dim": 8,
8
+ "block_out_channels": [
9
+ 64,
10
+ 64,
11
+ 128,
12
+ 128
13
+ ],
14
+ "center_input_sample": false,
15
+ "class_embed_type": null,
16
+ "class_embeddings_concat": false,
17
+ "conv_in_kernel": 3,
18
+ "conv_out_kernel": 3,
19
+ "cross_attention_dim": 256,
20
+ "cross_attention_norm": null,
21
+ "decay": 0.9999,
22
+ "down_block_types": [
23
+ "DownBlock2D",
24
+ "DownBlock2D",
25
+ "DownBlock2D",
26
+ "CrossAttnDownBlock2D"
27
+ ],
28
+ "downsample_padding": 1,
29
+ "dual_cross_attention": false,
30
+ "encoder_hid_dim": null,
31
+ "flip_sin_to_cos": true,
32
+ "freq_shift": 0,
33
+ "in_channels": 3,
34
+ "inv_gamma": 1.0,
35
+ "layers_per_block": 2,
36
+ "mid_block_only_cross_attention": null,
37
+ "mid_block_scale_factor": 1,
38
+ "mid_block_type": "UNetMidBlock2DCrossAttn",
39
+ "min_decay": 0.0,
40
+ "norm_eps": 1e-05,
41
+ "norm_num_groups": 32,
42
+ "num_class_embeds": null,
43
+ "only_cross_attention": false,
44
+ "optimization_step": 94500,
45
+ "out_channels": 3,
46
+ "power": 0.75,
47
+ "projection_class_embeddings_input_dim": null,
48
+ "resnet_out_scale_factor": 1.0,
49
+ "resnet_skip_time_act": false,
50
+ "resnet_time_scale_shift": "default",
51
+ "sample_size": 256,
52
+ "time_cond_proj_dim": null,
53
+ "time_embedding_act_fn": null,
54
+ "time_embedding_dim": null,
55
+ "time_embedding_type": "positional",
56
+ "timestep_post_act": null,
57
+ "up_block_types": [
58
+ "CrossAttnUpBlock2D",
59
+ "UpBlock2D",
60
+ "UpBlock2D",
61
+ "UpBlock2D"
62
+ ],
63
+ "upcast_attention": false,
64
+ "update_after_step": 0,
65
+ "use_ema_warmup": true,
66
+ "use_linear_projection": false
67
+ }
checkpoint-94500/unet_ema/diffusion_pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:be43131470b06991f4da8ca13896e1501d3a63777b5e2f595da80aff1cf43c81
3
+ size 36616081
checkpoint-95000/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d04644d9f19970b8c028a60cda69cd37c7bfd5e8df347ab6bbb227f65011a44d
3
+ size 73251413
checkpoint-95000/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2525776b798088bdc3c7e2c289f93f9096fd1e5bed9278834af57ad7bdfad576
3
+ size 14663
checkpoint-95000/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f841ee043736fec647bc1fe76a4e7128cd68bab88f86d7681292b0cedd66f15d
3
+ size 557
checkpoint-95000/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5022595f7644489b574270b1e6b1508aea007715d9c08ee1b291bd2fb9b54eef
3
+ size 563
checkpoint-95000/unet/config.json ADDED
@@ -0,0 +1,60 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "UNet2DConditionModel",
3
+ "_diffusers_version": "0.16.1",
4
+ "act_fn": "silu",
5
+ "addition_embed_type": null,
6
+ "addition_embed_type_num_heads": 64,
7
+ "attention_head_dim": 8,
8
+ "block_out_channels": [
9
+ 64,
10
+ 64,
11
+ 128,
12
+ 128
13
+ ],
14
+ "center_input_sample": false,
15
+ "class_embed_type": null,
16
+ "class_embeddings_concat": false,
17
+ "conv_in_kernel": 3,
18
+ "conv_out_kernel": 3,
19
+ "cross_attention_dim": 256,
20
+ "cross_attention_norm": null,
21
+ "down_block_types": [
22
+ "DownBlock2D",
23
+ "DownBlock2D",
24
+ "DownBlock2D",
25
+ "CrossAttnDownBlock2D"
26
+ ],
27
+ "downsample_padding": 1,
28
+ "dual_cross_attention": false,
29
+ "encoder_hid_dim": null,
30
+ "flip_sin_to_cos": true,
31
+ "freq_shift": 0,
32
+ "in_channels": 3,
33
+ "layers_per_block": 2,
34
+ "mid_block_only_cross_attention": null,
35
+ "mid_block_scale_factor": 1,
36
+ "mid_block_type": "UNetMidBlock2DCrossAttn",
37
+ "norm_eps": 1e-05,
38
+ "norm_num_groups": 32,
39
+ "num_class_embeds": null,
40
+ "only_cross_attention": false,
41
+ "out_channels": 3,
42
+ "projection_class_embeddings_input_dim": null,
43
+ "resnet_out_scale_factor": 1.0,
44
+ "resnet_skip_time_act": false,
45
+ "resnet_time_scale_shift": "default",
46
+ "sample_size": 256,
47
+ "time_cond_proj_dim": null,
48
+ "time_embedding_act_fn": null,
49
+ "time_embedding_dim": null,
50
+ "time_embedding_type": "positional",
51
+ "timestep_post_act": null,
52
+ "up_block_types": [
53
+ "CrossAttnUpBlock2D",
54
+ "UpBlock2D",
55
+ "UpBlock2D",
56
+ "UpBlock2D"
57
+ ],
58
+ "upcast_attention": false,
59
+ "use_linear_projection": false
60
+ }
checkpoint-95000/unet/diffusion_pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ab62ad6666bb21ccf6cef860f65aff514d2d63b9b39963298f32bef7fe5e72f6
3
+ size 36621969
checkpoint-95000/unet_ema/config.json ADDED
@@ -0,0 +1,67 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "UNet2DConditionModel",
3
+ "_diffusers_version": "0.16.1",
4
+ "act_fn": "silu",
5
+ "addition_embed_type": null,
6
+ "addition_embed_type_num_heads": 64,
7
+ "attention_head_dim": 8,
8
+ "block_out_channels": [
9
+ 64,
10
+ 64,
11
+ 128,
12
+ 128
13
+ ],
14
+ "center_input_sample": false,
15
+ "class_embed_type": null,
16
+ "class_embeddings_concat": false,
17
+ "conv_in_kernel": 3,
18
+ "conv_out_kernel": 3,
19
+ "cross_attention_dim": 256,
20
+ "cross_attention_norm": null,
21
+ "decay": 0.9999,
22
+ "down_block_types": [
23
+ "DownBlock2D",
24
+ "DownBlock2D",
25
+ "DownBlock2D",
26
+ "CrossAttnDownBlock2D"
27
+ ],
28
+ "downsample_padding": 1,
29
+ "dual_cross_attention": false,
30
+ "encoder_hid_dim": null,
31
+ "flip_sin_to_cos": true,
32
+ "freq_shift": 0,
33
+ "in_channels": 3,
34
+ "inv_gamma": 1.0,
35
+ "layers_per_block": 2,
36
+ "mid_block_only_cross_attention": null,
37
+ "mid_block_scale_factor": 1,
38
+ "mid_block_type": "UNetMidBlock2DCrossAttn",
39
+ "min_decay": 0.0,
40
+ "norm_eps": 1e-05,
41
+ "norm_num_groups": 32,
42
+ "num_class_embeds": null,
43
+ "only_cross_attention": false,
44
+ "optimization_step": 95000,
45
+ "out_channels": 3,
46
+ "power": 0.75,
47
+ "projection_class_embeddings_input_dim": null,
48
+ "resnet_out_scale_factor": 1.0,
49
+ "resnet_skip_time_act": false,
50
+ "resnet_time_scale_shift": "default",
51
+ "sample_size": 256,
52
+ "time_cond_proj_dim": null,
53
+ "time_embedding_act_fn": null,
54
+ "time_embedding_dim": null,
55
+ "time_embedding_type": "positional",
56
+ "timestep_post_act": null,
57
+ "up_block_types": [
58
+ "CrossAttnUpBlock2D",
59
+ "UpBlock2D",
60
+ "UpBlock2D",
61
+ "UpBlock2D"
62
+ ],
63
+ "upcast_attention": false,
64
+ "update_after_step": 0,
65
+ "use_ema_warmup": true,
66
+ "use_linear_projection": false
67
+ }
checkpoint-95000/unet_ema/diffusion_pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7064de5886a982c76536d1674bb4077946b2aefcc5ffdae77482fe8944a1542c
3
+ size 36616081
checkpoint-95500/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:66401a6cf4d67cc9a0c39b7c5a3ec3ae2be9495925d93c8dd46a524eee548799
3
+ size 73251413
checkpoint-95500/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9940cb9a6d2aba9d7e42e30b97e5be470c09e5459ee87a4e3165419100b77b3d
3
+ size 14663
checkpoint-95500/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4b0e64af040caf0fa5ed53c6944897813b59262c29ec937e78f962a61f458c4a
3
+ size 557
checkpoint-95500/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:718ebb7147d2985931efa711ae44613bc763a662f2a9b1f4f63e311254491ede
3
+ size 563
checkpoint-95500/unet/config.json ADDED
@@ -0,0 +1,60 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "UNet2DConditionModel",
3
+ "_diffusers_version": "0.16.1",
4
+ "act_fn": "silu",
5
+ "addition_embed_type": null,
6
+ "addition_embed_type_num_heads": 64,
7
+ "attention_head_dim": 8,
8
+ "block_out_channels": [
9
+ 64,
10
+ 64,
11
+ 128,
12
+ 128
13
+ ],
14
+ "center_input_sample": false,
15
+ "class_embed_type": null,
16
+ "class_embeddings_concat": false,
17
+ "conv_in_kernel": 3,
18
+ "conv_out_kernel": 3,
19
+ "cross_attention_dim": 256,
20
+ "cross_attention_norm": null,
21
+ "down_block_types": [
22
+ "DownBlock2D",
23
+ "DownBlock2D",
24
+ "DownBlock2D",
25
+ "CrossAttnDownBlock2D"
26
+ ],
27
+ "downsample_padding": 1,
28
+ "dual_cross_attention": false,
29
+ "encoder_hid_dim": null,
30
+ "flip_sin_to_cos": true,
31
+ "freq_shift": 0,
32
+ "in_channels": 3,
33
+ "layers_per_block": 2,
34
+ "mid_block_only_cross_attention": null,
35
+ "mid_block_scale_factor": 1,
36
+ "mid_block_type": "UNetMidBlock2DCrossAttn",
37
+ "norm_eps": 1e-05,
38
+ "norm_num_groups": 32,
39
+ "num_class_embeds": null,
40
+ "only_cross_attention": false,
41
+ "out_channels": 3,
42
+ "projection_class_embeddings_input_dim": null,
43
+ "resnet_out_scale_factor": 1.0,
44
+ "resnet_skip_time_act": false,
45
+ "resnet_time_scale_shift": "default",
46
+ "sample_size": 256,
47
+ "time_cond_proj_dim": null,
48
+ "time_embedding_act_fn": null,
49
+ "time_embedding_dim": null,
50
+ "time_embedding_type": "positional",
51
+ "timestep_post_act": null,
52
+ "up_block_types": [
53
+ "CrossAttnUpBlock2D",
54
+ "UpBlock2D",
55
+ "UpBlock2D",
56
+ "UpBlock2D"
57
+ ],
58
+ "upcast_attention": false,
59
+ "use_linear_projection": false
60
+ }
checkpoint-95500/unet/diffusion_pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c5e7aef252005cf8f93a52031eca3aa846e8c07abdfd6cf280fa71a78e15b98c
3
+ size 36621969
checkpoint-95500/unet_ema/config.json ADDED
@@ -0,0 +1,67 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "UNet2DConditionModel",
3
+ "_diffusers_version": "0.16.1",
4
+ "act_fn": "silu",
5
+ "addition_embed_type": null,
6
+ "addition_embed_type_num_heads": 64,
7
+ "attention_head_dim": 8,
8
+ "block_out_channels": [
9
+ 64,
10
+ 64,
11
+ 128,
12
+ 128
13
+ ],
14
+ "center_input_sample": false,
15
+ "class_embed_type": null,
16
+ "class_embeddings_concat": false,
17
+ "conv_in_kernel": 3,
18
+ "conv_out_kernel": 3,
19
+ "cross_attention_dim": 256,
20
+ "cross_attention_norm": null,
21
+ "decay": 0.9999,
22
+ "down_block_types": [
23
+ "DownBlock2D",
24
+ "DownBlock2D",
25
+ "DownBlock2D",
26
+ "CrossAttnDownBlock2D"
27
+ ],
28
+ "downsample_padding": 1,
29
+ "dual_cross_attention": false,
30
+ "encoder_hid_dim": null,
31
+ "flip_sin_to_cos": true,
32
+ "freq_shift": 0,
33
+ "in_channels": 3,
34
+ "inv_gamma": 1.0,
35
+ "layers_per_block": 2,
36
+ "mid_block_only_cross_attention": null,
37
+ "mid_block_scale_factor": 1,
38
+ "mid_block_type": "UNetMidBlock2DCrossAttn",
39
+ "min_decay": 0.0,
40
+ "norm_eps": 1e-05,
41
+ "norm_num_groups": 32,
42
+ "num_class_embeds": null,
43
+ "only_cross_attention": false,
44
+ "optimization_step": 95500,
45
+ "out_channels": 3,
46
+ "power": 0.75,
47
+ "projection_class_embeddings_input_dim": null,
48
+ "resnet_out_scale_factor": 1.0,
49
+ "resnet_skip_time_act": false,
50
+ "resnet_time_scale_shift": "default",
51
+ "sample_size": 256,
52
+ "time_cond_proj_dim": null,
53
+ "time_embedding_act_fn": null,
54
+ "time_embedding_dim": null,
55
+ "time_embedding_type": "positional",
56
+ "timestep_post_act": null,
57
+ "up_block_types": [
58
+ "CrossAttnUpBlock2D",
59
+ "UpBlock2D",
60
+ "UpBlock2D",
61
+ "UpBlock2D"
62
+ ],
63
+ "upcast_attention": false,
64
+ "update_after_step": 0,
65
+ "use_ema_warmup": true,
66
+ "use_linear_projection": false
67
+ }
checkpoint-95500/unet_ema/diffusion_pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:677e047878e3d4b0824498e6a995521d104ca9653e3cb749ea35dd48c6948a95
3
+ size 36616081
checkpoint-96000/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cf192b53c94fae5c756e978b727dbb331e4c70b664124e68be1db121246571b3
3
+ size 73251413
checkpoint-96000/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b576f956b5cdabe53b9ecce7567ecc39095bb583d5a97d5e50ced17759705747
3
+ size 14663
checkpoint-96000/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:484d90b46bc359f73d3d95c5701b3242325f45dbd9547ed21821f9ea28c2b618
3
+ size 557
checkpoint-96000/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3983644dacfce33c6f4bf3c2573a66f3aad5be13a8560caf9ac27f319c287921
3
+ size 563
checkpoint-96000/unet/config.json ADDED
@@ -0,0 +1,60 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "UNet2DConditionModel",
3
+ "_diffusers_version": "0.16.1",
4
+ "act_fn": "silu",
5
+ "addition_embed_type": null,
6
+ "addition_embed_type_num_heads": 64,
7
+ "attention_head_dim": 8,
8
+ "block_out_channels": [
9
+ 64,
10
+ 64,
11
+ 128,
12
+ 128
13
+ ],
14
+ "center_input_sample": false,
15
+ "class_embed_type": null,
16
+ "class_embeddings_concat": false,
17
+ "conv_in_kernel": 3,
18
+ "conv_out_kernel": 3,
19
+ "cross_attention_dim": 256,
20
+ "cross_attention_norm": null,
21
+ "down_block_types": [
22
+ "DownBlock2D",
23
+ "DownBlock2D",
24
+ "DownBlock2D",
25
+ "CrossAttnDownBlock2D"
26
+ ],
27
+ "downsample_padding": 1,
28
+ "dual_cross_attention": false,
29
+ "encoder_hid_dim": null,
30
+ "flip_sin_to_cos": true,
31
+ "freq_shift": 0,
32
+ "in_channels": 3,
33
+ "layers_per_block": 2,
34
+ "mid_block_only_cross_attention": null,
35
+ "mid_block_scale_factor": 1,
36
+ "mid_block_type": "UNetMidBlock2DCrossAttn",
37
+ "norm_eps": 1e-05,
38
+ "norm_num_groups": 32,
39
+ "num_class_embeds": null,
40
+ "only_cross_attention": false,
41
+ "out_channels": 3,
42
+ "projection_class_embeddings_input_dim": null,
43
+ "resnet_out_scale_factor": 1.0,
44
+ "resnet_skip_time_act": false,
45
+ "resnet_time_scale_shift": "default",
46
+ "sample_size": 256,
47
+ "time_cond_proj_dim": null,
48
+ "time_embedding_act_fn": null,
49
+ "time_embedding_dim": null,
50
+ "time_embedding_type": "positional",
51
+ "timestep_post_act": null,
52
+ "up_block_types": [
53
+ "CrossAttnUpBlock2D",
54
+ "UpBlock2D",
55
+ "UpBlock2D",
56
+ "UpBlock2D"
57
+ ],
58
+ "upcast_attention": false,
59
+ "use_linear_projection": false
60
+ }
checkpoint-96000/unet/diffusion_pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d733e0a1c67922b365df0c6c4c3bedcdb60dcefaf36c3eee077c3cf874f972ef
3
+ size 36621969
checkpoint-96000/unet_ema/config.json ADDED
@@ -0,0 +1,67 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "UNet2DConditionModel",
3
+ "_diffusers_version": "0.16.1",
4
+ "act_fn": "silu",
5
+ "addition_embed_type": null,
6
+ "addition_embed_type_num_heads": 64,
7
+ "attention_head_dim": 8,
8
+ "block_out_channels": [
9
+ 64,
10
+ 64,
11
+ 128,
12
+ 128
13
+ ],
14
+ "center_input_sample": false,
15
+ "class_embed_type": null,
16
+ "class_embeddings_concat": false,
17
+ "conv_in_kernel": 3,
18
+ "conv_out_kernel": 3,
19
+ "cross_attention_dim": 256,
20
+ "cross_attention_norm": null,
21
+ "decay": 0.9999,
22
+ "down_block_types": [
23
+ "DownBlock2D",
24
+ "DownBlock2D",
25
+ "DownBlock2D",
26
+ "CrossAttnDownBlock2D"
27
+ ],
28
+ "downsample_padding": 1,
29
+ "dual_cross_attention": false,
30
+ "encoder_hid_dim": null,
31
+ "flip_sin_to_cos": true,
32
+ "freq_shift": 0,
33
+ "in_channels": 3,
34
+ "inv_gamma": 1.0,
35
+ "layers_per_block": 2,
36
+ "mid_block_only_cross_attention": null,
37
+ "mid_block_scale_factor": 1,
38
+ "mid_block_type": "UNetMidBlock2DCrossAttn",
39
+ "min_decay": 0.0,
40
+ "norm_eps": 1e-05,
41
+ "norm_num_groups": 32,
42
+ "num_class_embeds": null,
43
+ "only_cross_attention": false,
44
+ "optimization_step": 96000,
45
+ "out_channels": 3,
46
+ "power": 0.75,
47
+ "projection_class_embeddings_input_dim": null,
48
+ "resnet_out_scale_factor": 1.0,
49
+ "resnet_skip_time_act": false,
50
+ "resnet_time_scale_shift": "default",
51
+ "sample_size": 256,
52
+ "time_cond_proj_dim": null,
53
+ "time_embedding_act_fn": null,
54
+ "time_embedding_dim": null,
55
+ "time_embedding_type": "positional",
56
+ "timestep_post_act": null,
57
+ "up_block_types": [
58
+ "CrossAttnUpBlock2D",
59
+ "UpBlock2D",
60
+ "UpBlock2D",
61
+ "UpBlock2D"
62
+ ],
63
+ "upcast_attention": false,
64
+ "update_after_step": 0,
65
+ "use_ema_warmup": true,
66
+ "use_linear_projection": false
67
+ }
checkpoint-96000/unet_ema/diffusion_pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c5a33e66d5c61ddc0758e346d7c95c72ac71460cfe073a5733927a8614caa914
3
+ size 36616081
unet/diffusion_pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:9c8573689d56ab60f61e857036656cf07cc5332603be72a59c4077d7f15686b3
3
  size 36621969
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:09452ab0c28b9d9b2a1919692b7188954970b55f7f1687dbd26dbc4c360b60d5
3
  size 36621969