diff --git a/logs/train_example/events.out.tfevents.1706690889.cf36f1bba5d4.400.0 b/logs/train_example/events.out.tfevents.1706690889.cf36f1bba5d4.400.0 new file mode 100644 index 0000000000000000000000000000000000000000..b08d0d93d251cea5db97ad07068c8254c6f55575 --- /dev/null +++ b/logs/train_example/events.out.tfevents.1706690889.cf36f1bba5d4.400.0 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b869756ba316ca81baac57b51452381da6bcfced86f1d40615936c37486bdbed +size 2562 diff --git a/logs/train_example/events.out.tfevents.1706690957.cf36f1bba5d4.400.1 b/logs/train_example/events.out.tfevents.1706690957.cf36f1bba5d4.400.1 new file mode 100644 index 0000000000000000000000000000000000000000..ceec7780f79eeeda118f46eb113d3ae4a531ebcb --- /dev/null +++ b/logs/train_example/events.out.tfevents.1706690957.cf36f1bba5d4.400.1 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:055c1bf2bdf59d87fb4978139aa087ff6027097460ca9eb837472cee283b1397 +size 126698 diff --git a/samples/0000.png b/samples/0000.png index 067502f917b46775f50317af083f5577bcbe05c0..e2f01b8eefbf7e08f0283fb04debd198bb194cb2 100644 --- a/samples/0000.png +++ b/samples/0000.png @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:52c2401a24c0de3c2688078a02224b756bbe61771f4c18d08ebe754416546287 -size 434224 +oid sha256:8ee8e4333992120c6664bb3caeba67ca31a91d501ed6bfda1b405be123e7a2a4 +size 997592 diff --git a/samples/0001.png b/samples/0001.png index 21d7686812d86d992a9416ff22a2f8694164315d..986b228cde9e24a062d1a63f3ddbb1d02ca120b2 100644 --- a/samples/0001.png +++ b/samples/0001.png @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:a69aaaabb63577d4b772ba7c1ca4a20f49859b71df9c06d81554bcac7cce5faf -size 402439 +oid sha256:0d23eac64f2070840d6c50930f94bf87766b0d13fc6db9faf75ef3b06bc790d8 +size 1051021 diff --git a/samples/0002.png b/samples/0002.png index fb8f8d07fc946644077d1b15baf9f7a82a9a2e3b..31036fa774a47bc0b8135c4c62fa853aa9d4b7c3 100644 --- a/samples/0002.png +++ b/samples/0002.png @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:c8b6bb3e837f846489438b47ae42f3960b820a17095ea66df3c5fa00fcd60665 -size 1097 +oid sha256:dd065071b493f8af85ee34e1773b6494cc2faaba6f970ea652a9a808b632da15 +size 817492 diff --git a/samples/0003.png b/samples/0003.png index fb8f8d07fc946644077d1b15baf9f7a82a9a2e3b..06d4c2bf88c49ffaf12883f8e204965b77d85c0e 100644 --- a/samples/0003.png +++ b/samples/0003.png @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:c8b6bb3e837f846489438b47ae42f3960b820a17095ea66df3c5fa00fcd60665 -size 1097 +oid sha256:49b396b71c229598658cd43f5d34e3abb00d4cc85ad90f2c3f7d3a5d12b5f39f +size 846094 diff --git a/samples/0004.png b/samples/0004.png index fb8f8d07fc946644077d1b15baf9f7a82a9a2e3b..a55ab0f2fab27be4d1855ed7f0ccef3b583646d3 100644 --- a/samples/0004.png +++ b/samples/0004.png @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:c8b6bb3e837f846489438b47ae42f3960b820a17095ea66df3c5fa00fcd60665 -size 1097 +oid sha256:1146700272fa30443678655b9f53e0d2365e01115848d1c09504bb2d40068b57 +size 797222 diff --git a/samples/0005.png b/samples/0005.png new file mode 100644 index 0000000000000000000000000000000000000000..0dd35b381a89d84866eb7f9ad2c63369e207b94a --- /dev/null +++ b/samples/0005.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:f68ce552d79aef2fe5acf5011aa047c72913eb4b1faf7185dc555fac488b5ee8 +size 756901 diff --git a/samples/0006.png b/samples/0006.png new file mode 100644 index 0000000000000000000000000000000000000000..811582537c0166fa49960df5a4fb91732d39a9de --- /dev/null +++ b/samples/0006.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:5a0593268b7bd8d469a90ccc2ffc8c069be71e3907a6f6fe7155de758f84be5c +size 749991 diff --git a/samples/0007.png b/samples/0007.png new file mode 100644 index 0000000000000000000000000000000000000000..e7d25736ac168b42f964287b3a4421428b8a818d --- /dev/null +++ b/samples/0007.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:3e61928ac3bca404be8298a95e29f3ce48ab8d99a28dbb14e18baa3a05b4e619 +size 758999 diff --git a/samples/0008.png b/samples/0008.png new file mode 100644 index 0000000000000000000000000000000000000000..775e66348a581c4bf3b060dc54b19e40c427cf67 --- /dev/null +++ b/samples/0008.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:28e5d9fa99647ea03552366963224d2c31de71cc85303bb13d1df39bcd93c10a +size 786318 diff --git a/samples/0009.png b/samples/0009.png new file mode 100644 index 0000000000000000000000000000000000000000..a94800f985f50e1fa0f7f55f3f716cd8b34bcf52 --- /dev/null +++ b/samples/0009.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:f428b1492b906042199e2c486d58c4b51e5dfc81d60c8375b523ffb0e98eb2cf +size 761633 diff --git a/samples/0010.png b/samples/0010.png new file mode 100644 index 0000000000000000000000000000000000000000..25407b02a590d18fcd2f8afd6b506fde1c6a3adc Binary files /dev/null and b/samples/0010.png differ diff --git a/samples/0011.png b/samples/0011.png new file mode 100644 index 0000000000000000000000000000000000000000..d5fcd3540d422923cde055e0026c58d3934d373c Binary files /dev/null and b/samples/0011.png differ diff --git a/samples/0012.png b/samples/0012.png new file mode 100644 index 0000000000000000000000000000000000000000..e0a502677b5ed2879890169bee7b18b17cd58aae Binary files /dev/null and b/samples/0012.png differ diff --git a/samples/0013.png b/samples/0013.png new file mode 100644 index 0000000000000000000000000000000000000000..cf46da3e47f68e7ed6bf899fd5e36885fc685d55 Binary files /dev/null and b/samples/0013.png differ diff --git a/samples/0014.png b/samples/0014.png new file mode 100644 index 0000000000000000000000000000000000000000..4bb8349103be32e948c4cecd310abea48fec305f Binary files /dev/null and b/samples/0014.png differ diff --git a/samples/0015.png b/samples/0015.png new file mode 100644 index 0000000000000000000000000000000000000000..7b26aeee8c4f469aa969c1cd6aa7014dc01cf00c Binary files /dev/null and b/samples/0015.png differ diff --git a/samples/0016.png b/samples/0016.png new file mode 100644 index 0000000000000000000000000000000000000000..e11c97be3455237c262bb73444884283bc65b6cd Binary files /dev/null and b/samples/0016.png differ diff --git a/samples/0017.png b/samples/0017.png new file mode 100644 index 0000000000000000000000000000000000000000..4a103f3631521b293090cddfa906afa10f82471c Binary files /dev/null and b/samples/0017.png differ diff --git a/samples/0018.png b/samples/0018.png new file mode 100644 index 0000000000000000000000000000000000000000..6ea2eff29f331d59a0a9153d1f8d55fd0be9bb31 Binary files /dev/null and b/samples/0018.png differ diff --git a/samples/0019.png b/samples/0019.png new file mode 100644 index 0000000000000000000000000000000000000000..74c92dbcfb55b1d53f987451bba91a00b3a48ac7 Binary files /dev/null and b/samples/0019.png differ diff --git a/samples/0020.png b/samples/0020.png new file mode 100644 index 0000000000000000000000000000000000000000..fdc563614726e27e9584a1eb883051e42b231029 Binary files /dev/null and b/samples/0020.png differ diff --git a/samples/0021.png b/samples/0021.png new file mode 100644 index 0000000000000000000000000000000000000000..ffa345e9ca9392dbf405cf24b3ac9b10459c58b7 Binary files /dev/null and b/samples/0021.png differ diff --git a/samples/0022.png b/samples/0022.png new file mode 100644 index 0000000000000000000000000000000000000000..3e1b4e669ad4b28ea3ec67f04f33c69c7ad2f520 Binary files /dev/null and b/samples/0022.png differ diff --git a/samples/0023.png b/samples/0023.png new file mode 100644 index 0000000000000000000000000000000000000000..de1debc74f83d4d72503669e2a5b9b107cf55b3a Binary files /dev/null and b/samples/0023.png differ diff --git a/samples/0024.png b/samples/0024.png new file mode 100644 index 0000000000000000000000000000000000000000..097e88e436429bbbcfa796fb56627c96c6e5d3ae Binary files /dev/null and b/samples/0024.png differ diff --git a/samples/0025.png b/samples/0025.png new file mode 100644 index 0000000000000000000000000000000000000000..33faded45ca82040f67f23aa925e9b47e2e7193e Binary files /dev/null and b/samples/0025.png differ diff --git a/samples/0026.png b/samples/0026.png new file mode 100644 index 0000000000000000000000000000000000000000..f70b3c1db1812e7fd1df02b62e4ec09aa0800d30 Binary files /dev/null and b/samples/0026.png differ diff --git a/samples/0027.png b/samples/0027.png new file mode 100644 index 0000000000000000000000000000000000000000..739ce57f8dd4bae8cb90689987e7ae068e031a44 Binary files /dev/null and b/samples/0027.png differ diff --git a/samples/0028.png b/samples/0028.png new file mode 100644 index 0000000000000000000000000000000000000000..99abb9bb48f6f8de907c35736c4124b2fc88999b Binary files /dev/null and b/samples/0028.png differ diff --git a/samples/0029.png b/samples/0029.png new file mode 100644 index 0000000000000000000000000000000000000000..d31f10ac5e0d9cb40391afd7b295a6103fc6be27 Binary files /dev/null and b/samples/0029.png differ diff --git a/samples/0030.png b/samples/0030.png new file mode 100644 index 0000000000000000000000000000000000000000..dd79652ff55b1d2988c97a1bd64681a1f13b060a Binary files /dev/null and b/samples/0030.png differ diff --git a/samples/0031.png b/samples/0031.png new file mode 100644 index 0000000000000000000000000000000000000000..c55824e25b0450fc522b36d5d79165d4c3c46564 Binary files /dev/null and b/samples/0031.png differ diff --git a/samples/0032.png b/samples/0032.png new file mode 100644 index 0000000000000000000000000000000000000000..29c3de38572d3dca50eb353e69f9c7003459174e Binary files /dev/null and b/samples/0032.png differ diff --git a/samples/0033.png b/samples/0033.png new file mode 100644 index 0000000000000000000000000000000000000000..5cfb7a02c800ce6150e264f4ad2c1e0b8ac1fa76 Binary files /dev/null and b/samples/0033.png differ diff --git a/samples/0034.png b/samples/0034.png new file mode 100644 index 0000000000000000000000000000000000000000..48fb3e03e0f7fa50fe44cb0bee64541b53041d5a Binary files /dev/null and b/samples/0034.png differ diff --git a/samples/0035.png b/samples/0035.png new file mode 100644 index 0000000000000000000000000000000000000000..29d2b334e89875842c348f93d1b7906b20161cc9 Binary files /dev/null and b/samples/0035.png differ diff --git a/samples/0036.png b/samples/0036.png new file mode 100644 index 0000000000000000000000000000000000000000..8ae7dc3e9476d09102f1341de348ae28b5a8b47a Binary files /dev/null and b/samples/0036.png differ diff --git a/samples/0037.png b/samples/0037.png new file mode 100644 index 0000000000000000000000000000000000000000..49f4523892ba5d4bd41e74bff6d55389ca5b0712 Binary files /dev/null and b/samples/0037.png differ diff --git a/samples/0038.png b/samples/0038.png new file mode 100644 index 0000000000000000000000000000000000000000..8a07ca279eebaf79397157dce7b76f7e1f00ed85 Binary files /dev/null and b/samples/0038.png differ diff --git a/samples/0039.png b/samples/0039.png new file mode 100644 index 0000000000000000000000000000000000000000..a9a6f24f83325ff5b925499ab4e5e0a9193f8d70 Binary files /dev/null and b/samples/0039.png differ diff --git a/samples/0040.png b/samples/0040.png new file mode 100644 index 0000000000000000000000000000000000000000..a4d6f538296f6d9665a7cee6dc2fa1e1b646eb2d Binary files /dev/null and b/samples/0040.png differ diff --git a/samples/0041.png b/samples/0041.png new file mode 100644 index 0000000000000000000000000000000000000000..bfa6d723cf94477f9216d0937752157a2b1a88b0 Binary files /dev/null and b/samples/0041.png differ diff --git a/samples/0042.png b/samples/0042.png new file mode 100644 index 0000000000000000000000000000000000000000..92b00254b823ea383c802fc190b44c46acb8c957 Binary files /dev/null and b/samples/0042.png differ diff --git a/samples/0043.png b/samples/0043.png new file mode 100644 index 0000000000000000000000000000000000000000..40e8612e5a08df2b8b807d9448be6ae06be7c63e Binary files /dev/null and b/samples/0043.png differ diff --git a/samples/0044.png b/samples/0044.png new file mode 100644 index 0000000000000000000000000000000000000000..455cb85412e3d2ef7fd9df5cc32f8e7e02d133e5 Binary files /dev/null and b/samples/0044.png differ diff --git a/samples/0045.png b/samples/0045.png new file mode 100644 index 0000000000000000000000000000000000000000..b3431ee5573c7fcf321a16a38dc3bf9897d73ff8 Binary files /dev/null and b/samples/0045.png differ diff --git a/samples/0046.png b/samples/0046.png new file mode 100644 index 0000000000000000000000000000000000000000..8a7e8525c13682d540a1f661eeb42cf537cf51b1 Binary files /dev/null and b/samples/0046.png differ diff --git a/samples/0047.png b/samples/0047.png new file mode 100644 index 0000000000000000000000000000000000000000..3f203ee4471ab0c80ad0d4c6e24f8d2c7942e140 Binary files /dev/null and b/samples/0047.png differ diff --git a/samples/0048.png b/samples/0048.png new file mode 100644 index 0000000000000000000000000000000000000000..a5ffc80694604e51a2fe34c6fec19721ad1b5fbd Binary files /dev/null and b/samples/0048.png differ diff --git a/samples/0049.png b/samples/0049.png new file mode 100644 index 0000000000000000000000000000000000000000..9036b7c856a25c8342c21bbbf61609c81703d2b3 Binary files /dev/null and b/samples/0049.png differ diff --git a/scheduler/scheduler_config.json b/scheduler/scheduler_config.json index 44e50b36ee12e93ee3604ce09665bcb4238d8259..9b16c7a4543460c5f3d8266cc8207e8db295ab9e 100644 --- a/scheduler/scheduler_config.json +++ b/scheduler/scheduler_config.json @@ -5,8 +5,15 @@ "beta_schedule": "linear", "beta_start": 0.0001, "clip_sample": true, - "num_train_timesteps": 100, + "clip_sample_range": 1.0, + "dynamic_thresholding_ratio": 0.995, + "num_train_timesteps": 500, "prediction_type": "epsilon", + "rescale_betas_zero_snr": false, + "sample_max_value": 1.0, + "steps_offset": 0, + "thresholding": false, + "timestep_spacing": "leading", "trained_betas": null, "variance_type": "fixed_small" } diff --git a/unet/config.json b/unet/config.json index db3616bd500cd0d6d217160500b372ab8a187121..1270d2a4dede0a46a864fe4873d3164bfcf0543c 100644 --- a/unet/config.json +++ b/unet/config.json @@ -4,6 +4,7 @@ "act_fn": "silu", "add_attention": true, "attention_head_dim": 8, + "attn_norm_num_groups": null, "block_out_channels": [ 128, 128, @@ -13,6 +14,7 @@ 512 ], "center_input_sample": false, + "class_embed_type": null, "down_block_types": [ "DownBlock2D", "DownBlock2D", @@ -22,6 +24,8 @@ "DownBlock2D" ], "downsample_padding": 1, + "downsample_type": "conv", + "dropout": 0.0, "flip_sin_to_cos": true, "freq_shift": 0, "in_channels": 1, @@ -29,6 +33,8 @@ "mid_block_scale_factor": 1, "norm_eps": 1e-05, "norm_num_groups": 32, + "num_class_embeds": null, + "num_train_timesteps": null, "out_channels": 1, "resnet_time_scale_shift": "default", "sample_size": 512, @@ -40,5 +46,6 @@ "UpBlock2D", "UpBlock2D", "UpBlock2D" - ] + ], + "upsample_type": "conv" } diff --git a/unet/diffusion_pytorch_model.safetensors b/unet/diffusion_pytorch_model.safetensors index c66cc217f612c1c225b0f761221fb1893fafbfc2..1ae1091c379d44832697e0990890410ba85c3fad 100644 --- a/unet/diffusion_pytorch_model.safetensors +++ b/unet/diffusion_pytorch_model.safetensors @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:88dd72b2fd3454ddb26a14d50d495fff0c69f928fa8e2131fe67e2a1863a0429 -size 454741132 +oid sha256:02cc6951fb7b74f84631f2e457003257c53506bf77666e4bec8b4b93f181d1fa +size 309463124