gauenk commited on
Commit
126db33
1 Parent(s): 03b0dbd
Files changed (37) hide show
  1. checkpoint-1000/optimizer.bin +3 -0
  2. checkpoint-1000/random_states_0.pkl +3 -0
  3. checkpoint-1000/scheduler.bin +3 -0
  4. checkpoint-1000/unet/config.json +51 -0
  5. checkpoint-1000/unet/diffusion_pytorch_model.safetensors +3 -0
  6. checkpoint-1000/unet_ema/config.json +58 -0
  7. checkpoint-1000/unet_ema/diffusion_pytorch_model.safetensors +3 -0
  8. checkpoint-1500/optimizer.bin +3 -0
  9. checkpoint-1500/random_states_0.pkl +3 -0
  10. checkpoint-1500/scheduler.bin +3 -0
  11. checkpoint-1500/unet/config.json +51 -0
  12. checkpoint-1500/unet/diffusion_pytorch_model.safetensors +3 -0
  13. checkpoint-1500/unet_ema/config.json +58 -0
  14. checkpoint-1500/unet_ema/diffusion_pytorch_model.safetensors +3 -0
  15. checkpoint-2000/optimizer.bin +3 -0
  16. checkpoint-2000/random_states_0.pkl +3 -0
  17. checkpoint-2000/scheduler.bin +3 -0
  18. checkpoint-2000/unet/config.json +51 -0
  19. checkpoint-2000/unet/diffusion_pytorch_model.safetensors +3 -0
  20. checkpoint-2000/unet_ema/config.json +58 -0
  21. checkpoint-2000/unet_ema/diffusion_pytorch_model.safetensors +3 -0
  22. checkpoint-2500/optimizer.bin +3 -0
  23. checkpoint-2500/random_states_0.pkl +3 -0
  24. checkpoint-2500/scheduler.bin +3 -0
  25. checkpoint-2500/unet/config.json +51 -0
  26. checkpoint-2500/unet/diffusion_pytorch_model.safetensors +3 -0
  27. checkpoint-2500/unet_ema/config.json +58 -0
  28. checkpoint-2500/unet_ema/diffusion_pytorch_model.safetensors +3 -0
  29. checkpoint-500/optimizer.bin +3 -0
  30. checkpoint-500/random_states_0.pkl +3 -0
  31. checkpoint-500/scheduler.bin +3 -0
  32. checkpoint-500/unet/config.json +51 -0
  33. checkpoint-500/unet/diffusion_pytorch_model.safetensors +3 -0
  34. checkpoint-500/unet_ema/config.json +58 -0
  35. checkpoint-500/unet_ema/diffusion_pytorch_model.safetensors +3 -0
  36. logs/train_flowers/events.out.tfevents.1711144332.kent-pc.3621066.0 +2 -2
  37. unet/diffusion_pytorch_model.safetensors +1 -1
checkpoint-1000/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c7f9f2a7bcb3b6cf4f98c640a0238da6e3d6392f2f18d111e05f4c8bf9794dfb
3
+ size 909763736
checkpoint-1000/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d4c3fc8dadceae53c7f4a5c23ff10846d03518026878e9e37eb477cfec0e6b71
3
+ size 14604
checkpoint-1000/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4dc75ea621e0831b8362a0f4dc78058b75c83e1f04401e336eb47bc94acdbaee
3
+ size 1000
checkpoint-1000/unet/config.json ADDED
@@ -0,0 +1,51 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "UNet2DModel",
3
+ "_diffusers_version": "0.28.0.dev0",
4
+ "act_fn": "silu",
5
+ "add_attention": true,
6
+ "attention_head_dim": 8,
7
+ "attn_norm_num_groups": null,
8
+ "block_out_channels": [
9
+ 128,
10
+ 128,
11
+ 256,
12
+ 256,
13
+ 512,
14
+ 512
15
+ ],
16
+ "center_input_sample": false,
17
+ "class_embed_type": null,
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
+ "dropout": 0.0,
29
+ "flip_sin_to_cos": true,
30
+ "freq_shift": 0,
31
+ "in_channels": 3,
32
+ "layers_per_block": 2,
33
+ "mid_block_scale_factor": 1,
34
+ "norm_eps": 1e-05,
35
+ "norm_num_groups": 32,
36
+ "num_class_embeds": null,
37
+ "num_train_timesteps": null,
38
+ "out_channels": 3,
39
+ "resnet_time_scale_shift": "default",
40
+ "sample_size": 64,
41
+ "time_embedding_type": "positional",
42
+ "up_block_types": [
43
+ "UpBlock2D",
44
+ "AttnUpBlock2D",
45
+ "UpBlock2D",
46
+ "UpBlock2D",
47
+ "UpBlock2D",
48
+ "UpBlock2D"
49
+ ],
50
+ "upsample_type": "conv"
51
+ }
checkpoint-1000/unet/diffusion_pytorch_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4c2a42767edb98591123f7699e2d928afbb624c2e2cfdecd47d00f8f61c951c2
3
+ size 454741108
checkpoint-1000/unet_ema/config.json ADDED
@@ -0,0 +1,58 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "UNet2DModel",
3
+ "_diffusers_version": "0.28.0.dev0",
4
+ "act_fn": "silu",
5
+ "add_attention": true,
6
+ "attention_head_dim": 8,
7
+ "attn_norm_num_groups": null,
8
+ "block_out_channels": [
9
+ 128,
10
+ 128,
11
+ 256,
12
+ 256,
13
+ 512,
14
+ 512
15
+ ],
16
+ "center_input_sample": false,
17
+ "class_embed_type": null,
18
+ "decay": 0.9999,
19
+ "down_block_types": [
20
+ "DownBlock2D",
21
+ "DownBlock2D",
22
+ "DownBlock2D",
23
+ "DownBlock2D",
24
+ "AttnDownBlock2D",
25
+ "DownBlock2D"
26
+ ],
27
+ "downsample_padding": 1,
28
+ "downsample_type": "conv",
29
+ "dropout": 0.0,
30
+ "flip_sin_to_cos": true,
31
+ "freq_shift": 0,
32
+ "in_channels": 3,
33
+ "inv_gamma": 1.0,
34
+ "layers_per_block": 2,
35
+ "mid_block_scale_factor": 1,
36
+ "min_decay": 0.0,
37
+ "norm_eps": 1e-05,
38
+ "norm_num_groups": 32,
39
+ "num_class_embeds": null,
40
+ "num_train_timesteps": null,
41
+ "optimization_step": 1000,
42
+ "out_channels": 3,
43
+ "power": 0.75,
44
+ "resnet_time_scale_shift": "default",
45
+ "sample_size": 64,
46
+ "time_embedding_type": "positional",
47
+ "up_block_types": [
48
+ "UpBlock2D",
49
+ "AttnUpBlock2D",
50
+ "UpBlock2D",
51
+ "UpBlock2D",
52
+ "UpBlock2D",
53
+ "UpBlock2D"
54
+ ],
55
+ "update_after_step": 0,
56
+ "upsample_type": "conv",
57
+ "use_ema_warmup": true
58
+ }
checkpoint-1000/unet_ema/diffusion_pytorch_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a6c517b265723f91ad91ee28bacb6239d69ab7ea3904e8f14fcbb6fb94a2b4ef
3
+ size 454741108
checkpoint-1500/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ba1483aba2fe94d874067b658c501b493f44fe5ee7edd9ec2898421746e903fc
3
+ size 909763736
checkpoint-1500/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3b287f479247961ed5c4a2c3104c442d96a6efcb918db6b713657dc678f93582
3
+ size 14604
checkpoint-1500/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:25c88972e266e989ccbaf956fbbda289c98391723823c7f40dbe96883e25424d
3
+ size 1000
checkpoint-1500/unet/config.json ADDED
@@ -0,0 +1,51 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "UNet2DModel",
3
+ "_diffusers_version": "0.28.0.dev0",
4
+ "act_fn": "silu",
5
+ "add_attention": true,
6
+ "attention_head_dim": 8,
7
+ "attn_norm_num_groups": null,
8
+ "block_out_channels": [
9
+ 128,
10
+ 128,
11
+ 256,
12
+ 256,
13
+ 512,
14
+ 512
15
+ ],
16
+ "center_input_sample": false,
17
+ "class_embed_type": null,
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
+ "dropout": 0.0,
29
+ "flip_sin_to_cos": true,
30
+ "freq_shift": 0,
31
+ "in_channels": 3,
32
+ "layers_per_block": 2,
33
+ "mid_block_scale_factor": 1,
34
+ "norm_eps": 1e-05,
35
+ "norm_num_groups": 32,
36
+ "num_class_embeds": null,
37
+ "num_train_timesteps": null,
38
+ "out_channels": 3,
39
+ "resnet_time_scale_shift": "default",
40
+ "sample_size": 64,
41
+ "time_embedding_type": "positional",
42
+ "up_block_types": [
43
+ "UpBlock2D",
44
+ "AttnUpBlock2D",
45
+ "UpBlock2D",
46
+ "UpBlock2D",
47
+ "UpBlock2D",
48
+ "UpBlock2D"
49
+ ],
50
+ "upsample_type": "conv"
51
+ }
checkpoint-1500/unet/diffusion_pytorch_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2410a8f3baa9b3a73848ced15e73ce27313c92656b28289e89c0abb601e5edf1
3
+ size 454741108
checkpoint-1500/unet_ema/config.json ADDED
@@ -0,0 +1,58 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "UNet2DModel",
3
+ "_diffusers_version": "0.28.0.dev0",
4
+ "act_fn": "silu",
5
+ "add_attention": true,
6
+ "attention_head_dim": 8,
7
+ "attn_norm_num_groups": null,
8
+ "block_out_channels": [
9
+ 128,
10
+ 128,
11
+ 256,
12
+ 256,
13
+ 512,
14
+ 512
15
+ ],
16
+ "center_input_sample": false,
17
+ "class_embed_type": null,
18
+ "decay": 0.9999,
19
+ "down_block_types": [
20
+ "DownBlock2D",
21
+ "DownBlock2D",
22
+ "DownBlock2D",
23
+ "DownBlock2D",
24
+ "AttnDownBlock2D",
25
+ "DownBlock2D"
26
+ ],
27
+ "downsample_padding": 1,
28
+ "downsample_type": "conv",
29
+ "dropout": 0.0,
30
+ "flip_sin_to_cos": true,
31
+ "freq_shift": 0,
32
+ "in_channels": 3,
33
+ "inv_gamma": 1.0,
34
+ "layers_per_block": 2,
35
+ "mid_block_scale_factor": 1,
36
+ "min_decay": 0.0,
37
+ "norm_eps": 1e-05,
38
+ "norm_num_groups": 32,
39
+ "num_class_embeds": null,
40
+ "num_train_timesteps": null,
41
+ "optimization_step": 1500,
42
+ "out_channels": 3,
43
+ "power": 0.75,
44
+ "resnet_time_scale_shift": "default",
45
+ "sample_size": 64,
46
+ "time_embedding_type": "positional",
47
+ "up_block_types": [
48
+ "UpBlock2D",
49
+ "AttnUpBlock2D",
50
+ "UpBlock2D",
51
+ "UpBlock2D",
52
+ "UpBlock2D",
53
+ "UpBlock2D"
54
+ ],
55
+ "update_after_step": 0,
56
+ "upsample_type": "conv",
57
+ "use_ema_warmup": true
58
+ }
checkpoint-1500/unet_ema/diffusion_pytorch_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1cd717f3112200857d2556fc070093cad1eee8db1f7eb80e48ca95a4eae90a32
3
+ size 454741108
checkpoint-2000/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:29b5e60f031745f4464ef9ec1ca6ac7654e5f807d3d906f985ce7a69c52185f6
3
+ size 909763736
checkpoint-2000/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3ef502d35bf6cfd5200026d344619f6d8593ef1bdf9d325a93b5885cac208a89
3
+ size 14604
checkpoint-2000/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f1a8ff80a710537d924a74ed7127ac59dea114e7c77d9237940b1181e3ce5bfd
3
+ size 1000
checkpoint-2000/unet/config.json ADDED
@@ -0,0 +1,51 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "UNet2DModel",
3
+ "_diffusers_version": "0.28.0.dev0",
4
+ "act_fn": "silu",
5
+ "add_attention": true,
6
+ "attention_head_dim": 8,
7
+ "attn_norm_num_groups": null,
8
+ "block_out_channels": [
9
+ 128,
10
+ 128,
11
+ 256,
12
+ 256,
13
+ 512,
14
+ 512
15
+ ],
16
+ "center_input_sample": false,
17
+ "class_embed_type": null,
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
+ "dropout": 0.0,
29
+ "flip_sin_to_cos": true,
30
+ "freq_shift": 0,
31
+ "in_channels": 3,
32
+ "layers_per_block": 2,
33
+ "mid_block_scale_factor": 1,
34
+ "norm_eps": 1e-05,
35
+ "norm_num_groups": 32,
36
+ "num_class_embeds": null,
37
+ "num_train_timesteps": null,
38
+ "out_channels": 3,
39
+ "resnet_time_scale_shift": "default",
40
+ "sample_size": 64,
41
+ "time_embedding_type": "positional",
42
+ "up_block_types": [
43
+ "UpBlock2D",
44
+ "AttnUpBlock2D",
45
+ "UpBlock2D",
46
+ "UpBlock2D",
47
+ "UpBlock2D",
48
+ "UpBlock2D"
49
+ ],
50
+ "upsample_type": "conv"
51
+ }
checkpoint-2000/unet/diffusion_pytorch_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d7a5ce230da918b2239f9d484af3c9ff12ec4ed1ee02649ca2bb5e6c683f65d4
3
+ size 454741108
checkpoint-2000/unet_ema/config.json ADDED
@@ -0,0 +1,58 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "UNet2DModel",
3
+ "_diffusers_version": "0.28.0.dev0",
4
+ "act_fn": "silu",
5
+ "add_attention": true,
6
+ "attention_head_dim": 8,
7
+ "attn_norm_num_groups": null,
8
+ "block_out_channels": [
9
+ 128,
10
+ 128,
11
+ 256,
12
+ 256,
13
+ 512,
14
+ 512
15
+ ],
16
+ "center_input_sample": false,
17
+ "class_embed_type": null,
18
+ "decay": 0.9999,
19
+ "down_block_types": [
20
+ "DownBlock2D",
21
+ "DownBlock2D",
22
+ "DownBlock2D",
23
+ "DownBlock2D",
24
+ "AttnDownBlock2D",
25
+ "DownBlock2D"
26
+ ],
27
+ "downsample_padding": 1,
28
+ "downsample_type": "conv",
29
+ "dropout": 0.0,
30
+ "flip_sin_to_cos": true,
31
+ "freq_shift": 0,
32
+ "in_channels": 3,
33
+ "inv_gamma": 1.0,
34
+ "layers_per_block": 2,
35
+ "mid_block_scale_factor": 1,
36
+ "min_decay": 0.0,
37
+ "norm_eps": 1e-05,
38
+ "norm_num_groups": 32,
39
+ "num_class_embeds": null,
40
+ "num_train_timesteps": null,
41
+ "optimization_step": 2000,
42
+ "out_channels": 3,
43
+ "power": 0.75,
44
+ "resnet_time_scale_shift": "default",
45
+ "sample_size": 64,
46
+ "time_embedding_type": "positional",
47
+ "up_block_types": [
48
+ "UpBlock2D",
49
+ "AttnUpBlock2D",
50
+ "UpBlock2D",
51
+ "UpBlock2D",
52
+ "UpBlock2D",
53
+ "UpBlock2D"
54
+ ],
55
+ "update_after_step": 0,
56
+ "upsample_type": "conv",
57
+ "use_ema_warmup": true
58
+ }
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:7cd3c1042d79e40ffbb2ef09f456647953551497871ffd4bec5062410b4732c5
3
+ size 454741108
checkpoint-2500/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9e344775583fa4721f209ec133921353bbdd57e0a3757bd111f5de67e4ebfe28
3
+ size 909763736
checkpoint-2500/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a31f6de078e639ea5b7ac4ee400a3226664e13b17fb173a7710bd606df8d0b1b
3
+ size 14604
checkpoint-2500/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5a6e2ec6b6c7ee5a6b3f7f9b641cc3539bcd7abab20f91bf152c986a21e7fa26
3
+ size 1000
checkpoint-2500/unet/config.json ADDED
@@ -0,0 +1,51 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "UNet2DModel",
3
+ "_diffusers_version": "0.28.0.dev0",
4
+ "act_fn": "silu",
5
+ "add_attention": true,
6
+ "attention_head_dim": 8,
7
+ "attn_norm_num_groups": null,
8
+ "block_out_channels": [
9
+ 128,
10
+ 128,
11
+ 256,
12
+ 256,
13
+ 512,
14
+ 512
15
+ ],
16
+ "center_input_sample": false,
17
+ "class_embed_type": null,
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
+ "dropout": 0.0,
29
+ "flip_sin_to_cos": true,
30
+ "freq_shift": 0,
31
+ "in_channels": 3,
32
+ "layers_per_block": 2,
33
+ "mid_block_scale_factor": 1,
34
+ "norm_eps": 1e-05,
35
+ "norm_num_groups": 32,
36
+ "num_class_embeds": null,
37
+ "num_train_timesteps": null,
38
+ "out_channels": 3,
39
+ "resnet_time_scale_shift": "default",
40
+ "sample_size": 64,
41
+ "time_embedding_type": "positional",
42
+ "up_block_types": [
43
+ "UpBlock2D",
44
+ "AttnUpBlock2D",
45
+ "UpBlock2D",
46
+ "UpBlock2D",
47
+ "UpBlock2D",
48
+ "UpBlock2D"
49
+ ],
50
+ "upsample_type": "conv"
51
+ }
checkpoint-2500/unet/diffusion_pytorch_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6322a0b058bcfeef499bfc69ea901ef2ef9fd4595223cecffefbed0e4bf43539
3
+ size 454741108
checkpoint-2500/unet_ema/config.json ADDED
@@ -0,0 +1,58 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "UNet2DModel",
3
+ "_diffusers_version": "0.28.0.dev0",
4
+ "act_fn": "silu",
5
+ "add_attention": true,
6
+ "attention_head_dim": 8,
7
+ "attn_norm_num_groups": null,
8
+ "block_out_channels": [
9
+ 128,
10
+ 128,
11
+ 256,
12
+ 256,
13
+ 512,
14
+ 512
15
+ ],
16
+ "center_input_sample": false,
17
+ "class_embed_type": null,
18
+ "decay": 0.9999,
19
+ "down_block_types": [
20
+ "DownBlock2D",
21
+ "DownBlock2D",
22
+ "DownBlock2D",
23
+ "DownBlock2D",
24
+ "AttnDownBlock2D",
25
+ "DownBlock2D"
26
+ ],
27
+ "downsample_padding": 1,
28
+ "downsample_type": "conv",
29
+ "dropout": 0.0,
30
+ "flip_sin_to_cos": true,
31
+ "freq_shift": 0,
32
+ "in_channels": 3,
33
+ "inv_gamma": 1.0,
34
+ "layers_per_block": 2,
35
+ "mid_block_scale_factor": 1,
36
+ "min_decay": 0.0,
37
+ "norm_eps": 1e-05,
38
+ "norm_num_groups": 32,
39
+ "num_class_embeds": null,
40
+ "num_train_timesteps": null,
41
+ "optimization_step": 2500,
42
+ "out_channels": 3,
43
+ "power": 0.75,
44
+ "resnet_time_scale_shift": "default",
45
+ "sample_size": 64,
46
+ "time_embedding_type": "positional",
47
+ "up_block_types": [
48
+ "UpBlock2D",
49
+ "AttnUpBlock2D",
50
+ "UpBlock2D",
51
+ "UpBlock2D",
52
+ "UpBlock2D",
53
+ "UpBlock2D"
54
+ ],
55
+ "update_after_step": 0,
56
+ "upsample_type": "conv",
57
+ "use_ema_warmup": true
58
+ }
checkpoint-2500/unet_ema/diffusion_pytorch_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f05882cc089d7b6c5f8a1a8f9f3a128f5a8bda962e82bf137005e4d840b62ebd
3
+ size 454741108
checkpoint-500/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f8e0defc48c1b43ebd028db1855a5898ca1bbd98affe64565082b1b45379293c
3
+ size 909763736
checkpoint-500/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a6bf21702a7fe19b83f252f70717ee7436674188ad0338a7cdc6ed20d1e2f36b
3
+ size 14604
checkpoint-500/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e98739eb4b97fbd5416470b77110b9c750869c6c1abc89a792bd45eaa2c6aa48
3
+ size 1000
checkpoint-500/unet/config.json ADDED
@@ -0,0 +1,51 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "UNet2DModel",
3
+ "_diffusers_version": "0.28.0.dev0",
4
+ "act_fn": "silu",
5
+ "add_attention": true,
6
+ "attention_head_dim": 8,
7
+ "attn_norm_num_groups": null,
8
+ "block_out_channels": [
9
+ 128,
10
+ 128,
11
+ 256,
12
+ 256,
13
+ 512,
14
+ 512
15
+ ],
16
+ "center_input_sample": false,
17
+ "class_embed_type": null,
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
+ "dropout": 0.0,
29
+ "flip_sin_to_cos": true,
30
+ "freq_shift": 0,
31
+ "in_channels": 3,
32
+ "layers_per_block": 2,
33
+ "mid_block_scale_factor": 1,
34
+ "norm_eps": 1e-05,
35
+ "norm_num_groups": 32,
36
+ "num_class_embeds": null,
37
+ "num_train_timesteps": null,
38
+ "out_channels": 3,
39
+ "resnet_time_scale_shift": "default",
40
+ "sample_size": 64,
41
+ "time_embedding_type": "positional",
42
+ "up_block_types": [
43
+ "UpBlock2D",
44
+ "AttnUpBlock2D",
45
+ "UpBlock2D",
46
+ "UpBlock2D",
47
+ "UpBlock2D",
48
+ "UpBlock2D"
49
+ ],
50
+ "upsample_type": "conv"
51
+ }
checkpoint-500/unet/diffusion_pytorch_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c9366710e22002a538aaf3dd1346667f72ed31f5fe34b588a519ce60de7ea1cb
3
+ size 454741108
checkpoint-500/unet_ema/config.json ADDED
@@ -0,0 +1,58 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "UNet2DModel",
3
+ "_diffusers_version": "0.28.0.dev0",
4
+ "act_fn": "silu",
5
+ "add_attention": true,
6
+ "attention_head_dim": 8,
7
+ "attn_norm_num_groups": null,
8
+ "block_out_channels": [
9
+ 128,
10
+ 128,
11
+ 256,
12
+ 256,
13
+ 512,
14
+ 512
15
+ ],
16
+ "center_input_sample": false,
17
+ "class_embed_type": null,
18
+ "decay": 0.9999,
19
+ "down_block_types": [
20
+ "DownBlock2D",
21
+ "DownBlock2D",
22
+ "DownBlock2D",
23
+ "DownBlock2D",
24
+ "AttnDownBlock2D",
25
+ "DownBlock2D"
26
+ ],
27
+ "downsample_padding": 1,
28
+ "downsample_type": "conv",
29
+ "dropout": 0.0,
30
+ "flip_sin_to_cos": true,
31
+ "freq_shift": 0,
32
+ "in_channels": 3,
33
+ "inv_gamma": 1.0,
34
+ "layers_per_block": 2,
35
+ "mid_block_scale_factor": 1,
36
+ "min_decay": 0.0,
37
+ "norm_eps": 1e-05,
38
+ "norm_num_groups": 32,
39
+ "num_class_embeds": null,
40
+ "num_train_timesteps": null,
41
+ "optimization_step": 500,
42
+ "out_channels": 3,
43
+ "power": 0.75,
44
+ "resnet_time_scale_shift": "default",
45
+ "sample_size": 64,
46
+ "time_embedding_type": "positional",
47
+ "up_block_types": [
48
+ "UpBlock2D",
49
+ "AttnUpBlock2D",
50
+ "UpBlock2D",
51
+ "UpBlock2D",
52
+ "UpBlock2D",
53
+ "UpBlock2D"
54
+ ],
55
+ "update_after_step": 0,
56
+ "upsample_type": "conv",
57
+ "use_ema_warmup": true
58
+ }
checkpoint-500/unet_ema/diffusion_pytorch_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4eb4e971e37e0cedd549d8b40dd17da6fbfb92c75d70c9177cf8f3d83285d608
3
+ size 454741108
logs/train_flowers/events.out.tfevents.1711144332.kent-pc.3621066.0 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:e18245f95cf2ab0591eddefe60effd0c0f6eb6a3ba4ce9b0f495508b6ef73ce4
3
- size 191960
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d72eace8bed83c5ef1d905b1949b2063f53506a7f37a280c8b53a2215ae5440a
3
+ size 773236
unet/diffusion_pytorch_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:6524e0c226873adc2f8f3eeb960ecd153e8960cb8f62525d097d84cc9e481a9a
3
  size 454741108
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:464948435ef075544f91cb5863b7ed360e436b0c11626036e67ef94f64bf2982
3
  size 454741108