diff --git a/checkpoint-12000/optimizer.bin b/checkpoint-12000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..9583ddf12c1d29544f7bdd62628acbe3cab59488 --- /dev/null +++ b/checkpoint-12000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:974b5bc38acb57fe59ba5b701319a1bc03589a97daac1bafae7b46a0c67c29de +size 909763299 diff --git a/checkpoint-12000/random_states_0.pkl b/checkpoint-12000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..256d5ad42b4c0b85f1441e936353c10f7f72ba4d --- /dev/null +++ b/checkpoint-12000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e0e491630158259b9bf84ebb878ec5c1beea6442ae70b53376d5c6b6f98b255c +size 17811 diff --git a/checkpoint-12000/scaler.pt b/checkpoint-12000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..a5ba27272138cc678d110693e89eb04fec278c1c --- /dev/null +++ b/checkpoint-12000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7f42538cbf80d9d3d488f0fb2498336f0eea8018e2af8321bf8b20711c0ca289 +size 557 diff --git a/checkpoint-12000/scheduler.bin b/checkpoint-12000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..67f6486560c7165a380e5e42bbd9c4df6f9e6dec --- /dev/null +++ b/checkpoint-12000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:4bbe5d18dcc489a425353a98d7857f9b9f6f4bdbc0db6be62563be415a04c068 +size 563 diff --git a/checkpoint-12000/unet/config.json b/checkpoint-12000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..7a0c47e18dceccb3d9745c685fa44287c80c72d0 --- /dev/null +++ b/checkpoint-12000/unet/config.json @@ -0,0 +1,46 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.18.0.dev0", + "act_fn": "silu", + "add_attention": true, + "attention_head_dim": 8, + "block_out_channels": [ + 128, + 128, + 256, + 256, + 512, + 512 + ], + "center_input_sample": false, + "class_embed_type": null, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "AttnDownBlock2D", + "DownBlock2D" + ], + "downsample_padding": 1, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "layers_per_block": 2, + "mid_block_scale_factor": 1, + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "out_channels": 3, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_embedding_type": "positional", + "up_block_types": [ + "UpBlock2D", + "AttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ] +} diff --git a/checkpoint-12000/unet/diffusion_pytorch_model.bin b/checkpoint-12000/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..10a5ffd68ddf8e82f1f79e39df2d807f80231c0d --- /dev/null +++ b/checkpoint-12000/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:403c9621e78078408545fdb08caec193d508cb518533842532409e9d50633b09 +size 454867965 diff --git a/checkpoint-12000/unet_ema/config.json b/checkpoint-12000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..152612cc2dd0f85ec4c591b151e92981f6af793d --- /dev/null +++ b/checkpoint-12000/unet_ema/config.json @@ -0,0 +1,53 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.18.0.dev0", + "act_fn": "silu", + "add_attention": true, + "attention_head_dim": 8, + "block_out_channels": [ + 128, + 128, + 256, + 256, + 512, + 512 + ], + "center_input_sample": false, + "class_embed_type": null, + "decay": 0.9999, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "AttnDownBlock2D", + "DownBlock2D" + ], + "downsample_padding": 1, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "inv_gamma": 1.0, + "layers_per_block": 2, + "mid_block_scale_factor": 1, + "min_decay": 0.0, + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "optimization_step": 12000, + "out_channels": 3, + "power": 0.75, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_embedding_type": "positional", + "up_block_types": [ + "UpBlock2D", + "AttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "update_after_step": 0, + "use_ema_warmup": true +} diff --git a/checkpoint-12000/unet_ema/diffusion_pytorch_model.bin b/checkpoint-12000/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..4c145973d5fc4e018323296f4db2478cc222a6a6 --- /dev/null +++ b/checkpoint-12000/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:071247c4f8e51c794a114b0d1906f9ad2075e0de06ecda9dc50d12377fc9b898 +size 454861629 diff --git a/checkpoint-16000/optimizer.bin b/checkpoint-16000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..122267c452474c629bc5b74a2b704c995c9bd093 --- /dev/null +++ b/checkpoint-16000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:d5f3b8adbdc9e0ef0412dbaf7013e9ee3350dc8a80e64bebc2908f52b64d5809 +size 909763299 diff --git a/checkpoint-16000/random_states_0.pkl b/checkpoint-16000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..a179a84f89ebd6a1afb4f50d1abbd9349d50c0d5 --- /dev/null +++ b/checkpoint-16000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:0dc1f54c92720a0a7daeb4aedc414510e38e55073fc5fe971b4f03a2c5539902 +size 17811 diff --git a/checkpoint-16000/scaler.pt b/checkpoint-16000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..877376114c8b51d1f1304b6ee37d059b3e121b81 --- /dev/null +++ b/checkpoint-16000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c4c09d630f877940ceafcf5f685eb5ba5529b4047fdb34f343fe69daf3415c99 +size 557 diff --git a/checkpoint-16000/scheduler.bin b/checkpoint-16000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..5319806d1be7f5eba20f88ddd0e511697d8620c2 --- /dev/null +++ b/checkpoint-16000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:eacd9f89250dfb1f3316c16aa7bf9630f4c5acc497783eb15275e6c9d5dc4344 +size 563 diff --git a/checkpoint-16000/unet/config.json b/checkpoint-16000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..7a0c47e18dceccb3d9745c685fa44287c80c72d0 --- /dev/null +++ b/checkpoint-16000/unet/config.json @@ -0,0 +1,46 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.18.0.dev0", + "act_fn": "silu", + "add_attention": true, + "attention_head_dim": 8, + "block_out_channels": [ + 128, + 128, + 256, + 256, + 512, + 512 + ], + "center_input_sample": false, + "class_embed_type": null, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "AttnDownBlock2D", + "DownBlock2D" + ], + "downsample_padding": 1, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "layers_per_block": 2, + "mid_block_scale_factor": 1, + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "out_channels": 3, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_embedding_type": "positional", + "up_block_types": [ + "UpBlock2D", + "AttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ] +} diff --git a/checkpoint-16000/unet/diffusion_pytorch_model.bin b/checkpoint-16000/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..465e1f80da73875b8326fbcb56481945450d54d1 --- /dev/null +++ b/checkpoint-16000/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e0b8e086c677c25f3a548812700c0662bbfa009d9282bb9a9d90492092f1b5c1 +size 454867965 diff --git a/checkpoint-16000/unet_ema/config.json b/checkpoint-16000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..ea7f11b33addd7640d219d3986c4ba951f2aded8 --- /dev/null +++ b/checkpoint-16000/unet_ema/config.json @@ -0,0 +1,53 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.18.0.dev0", + "act_fn": "silu", + "add_attention": true, + "attention_head_dim": 8, + "block_out_channels": [ + 128, + 128, + 256, + 256, + 512, + 512 + ], + "center_input_sample": false, + "class_embed_type": null, + "decay": 0.9999, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "AttnDownBlock2D", + "DownBlock2D" + ], + "downsample_padding": 1, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "inv_gamma": 1.0, + "layers_per_block": 2, + "mid_block_scale_factor": 1, + "min_decay": 0.0, + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "optimization_step": 16000, + "out_channels": 3, + "power": 0.75, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_embedding_type": "positional", + "up_block_types": [ + "UpBlock2D", + "AttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "update_after_step": 0, + "use_ema_warmup": true +} diff --git a/checkpoint-16000/unet_ema/diffusion_pytorch_model.bin b/checkpoint-16000/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..c03d30471963b1289e8db18fa4dca891158eb010 --- /dev/null +++ b/checkpoint-16000/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:cf862e8850ce736bdecb232ca44f49c8bb9f5303fd4f2ce9f03baf66fcf1ba7a +size 454861629 diff --git a/checkpoint-20000/optimizer.bin b/checkpoint-20000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..cb72f0a270071404230e69d64be8dcb06427f71f --- /dev/null +++ b/checkpoint-20000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b0e3df79b6cf2538daec7707c440b9b31400db7ff50e115d699a3c67d4585548 +size 909763299 diff --git a/checkpoint-20000/random_states_0.pkl b/checkpoint-20000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..8d905a7cb26eebed50fd3d0c0a42fb27bb7f6bfa --- /dev/null +++ b/checkpoint-20000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:d99d1c8b109966000885bd7e93dcb84102407379c065ebecf81b295ea4468b89 +size 17811 diff --git a/checkpoint-20000/scaler.pt b/checkpoint-20000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..350a60fede3091e76f4e3bb1b25d65ed1fa36d6b --- /dev/null +++ b/checkpoint-20000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:906b8dab18dd4e790b865df1e27bfd905d5d060f24604e04674369f73e5ab51a +size 557 diff --git a/checkpoint-20000/scheduler.bin b/checkpoint-20000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..4f382031d71a35abaa2ec2e803399ab893abbee1 --- /dev/null +++ b/checkpoint-20000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:cebd295a62daf6f87a09c43fe2abb9a60346f3b70e1e565971874bff358abdf9 +size 563 diff --git a/checkpoint-20000/unet/config.json b/checkpoint-20000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..7a0c47e18dceccb3d9745c685fa44287c80c72d0 --- /dev/null +++ b/checkpoint-20000/unet/config.json @@ -0,0 +1,46 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.18.0.dev0", + "act_fn": "silu", + "add_attention": true, + "attention_head_dim": 8, + "block_out_channels": [ + 128, + 128, + 256, + 256, + 512, + 512 + ], + "center_input_sample": false, + "class_embed_type": null, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "AttnDownBlock2D", + "DownBlock2D" + ], + "downsample_padding": 1, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "layers_per_block": 2, + "mid_block_scale_factor": 1, + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "out_channels": 3, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_embedding_type": "positional", + "up_block_types": [ + "UpBlock2D", + "AttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ] +} diff --git a/checkpoint-20000/unet/diffusion_pytorch_model.bin b/checkpoint-20000/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..d64fa6210ab5c67a6718b2331dd8a39cbc5f704c --- /dev/null +++ b/checkpoint-20000/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:6abb642cce873a94c632c7d504b4a126732eb56a26bd22fcc38ccf81a7922033 +size 454867965 diff --git a/checkpoint-20000/unet_ema/config.json b/checkpoint-20000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..31c04015c331ca19d17af3fd054b852c31542891 --- /dev/null +++ b/checkpoint-20000/unet_ema/config.json @@ -0,0 +1,53 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.18.0.dev0", + "act_fn": "silu", + "add_attention": true, + "attention_head_dim": 8, + "block_out_channels": [ + 128, + 128, + 256, + 256, + 512, + 512 + ], + "center_input_sample": false, + "class_embed_type": null, + "decay": 0.9999, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "AttnDownBlock2D", + "DownBlock2D" + ], + "downsample_padding": 1, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "inv_gamma": 1.0, + "layers_per_block": 2, + "mid_block_scale_factor": 1, + "min_decay": 0.0, + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "optimization_step": 20000, + "out_channels": 3, + "power": 0.75, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_embedding_type": "positional", + "up_block_types": [ + "UpBlock2D", + "AttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "update_after_step": 0, + "use_ema_warmup": true +} diff --git a/checkpoint-20000/unet_ema/diffusion_pytorch_model.bin b/checkpoint-20000/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..79706dca20f82187b8068aa59548b13dbef5957f --- /dev/null +++ b/checkpoint-20000/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:6f108d2613f04ec72f8d95266b3b50f88d2492354ef25171cf873103b8da7b6f +size 454861629 diff --git a/checkpoint-24000/optimizer.bin b/checkpoint-24000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..990870437fd5c2ac1ea7765c4bac370f55e5ea6a --- /dev/null +++ b/checkpoint-24000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:aadb7dc0b99fd6413f260ae302109451046797ce1fbe68e479d3442a2b467c61 +size 909763299 diff --git a/checkpoint-24000/random_states_0.pkl b/checkpoint-24000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..55ad15d211809ba8e6757621ec265104e0ea9c6d --- /dev/null +++ b/checkpoint-24000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:68f13cdc1436466a9d51f6a67116721bb6dc83a89e66c8f18c15f4854ffd3948 +size 17811 diff --git a/checkpoint-24000/scaler.pt b/checkpoint-24000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..7b7a22bc8ca9395f22a3a90ecd90819794c68f87 --- /dev/null +++ b/checkpoint-24000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:6a37acc0b1866f021c0c02f4af78781bb91f42553e12daade08b3141b2805f3a +size 557 diff --git a/checkpoint-24000/scheduler.bin b/checkpoint-24000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..a8e751d8959a9dc5d1edef1c4f77cfe7b5c15286 --- /dev/null +++ b/checkpoint-24000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:466ce48c94a796b399038421aaa3fe4ba42876be480705a559e82871484b3541 +size 563 diff --git a/checkpoint-24000/unet/config.json b/checkpoint-24000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..7a0c47e18dceccb3d9745c685fa44287c80c72d0 --- /dev/null +++ b/checkpoint-24000/unet/config.json @@ -0,0 +1,46 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.18.0.dev0", + "act_fn": "silu", + "add_attention": true, + "attention_head_dim": 8, + "block_out_channels": [ + 128, + 128, + 256, + 256, + 512, + 512 + ], + "center_input_sample": false, + "class_embed_type": null, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "AttnDownBlock2D", + "DownBlock2D" + ], + "downsample_padding": 1, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "layers_per_block": 2, + "mid_block_scale_factor": 1, + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "out_channels": 3, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_embedding_type": "positional", + "up_block_types": [ + "UpBlock2D", + "AttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ] +} diff --git a/checkpoint-24000/unet/diffusion_pytorch_model.bin b/checkpoint-24000/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..7ff66997415479eeaeceb180fbe5b2b3c61be286 --- /dev/null +++ b/checkpoint-24000/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:a73a29cd2156a990409ffed906d6057ecbf411546f88647d20ee4ac852d58eb5 +size 454867965 diff --git a/checkpoint-24000/unet_ema/config.json b/checkpoint-24000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..ec1d7c8fd1ba9a21c7a352ddc2d37585434281fa --- /dev/null +++ b/checkpoint-24000/unet_ema/config.json @@ -0,0 +1,53 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.18.0.dev0", + "act_fn": "silu", + "add_attention": true, + "attention_head_dim": 8, + "block_out_channels": [ + 128, + 128, + 256, + 256, + 512, + 512 + ], + "center_input_sample": false, + "class_embed_type": null, + "decay": 0.9999, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "AttnDownBlock2D", + "DownBlock2D" + ], + "downsample_padding": 1, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "inv_gamma": 1.0, + "layers_per_block": 2, + "mid_block_scale_factor": 1, + "min_decay": 0.0, + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "optimization_step": 24000, + "out_channels": 3, + "power": 0.75, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_embedding_type": "positional", + "up_block_types": [ + "UpBlock2D", + "AttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "update_after_step": 0, + "use_ema_warmup": true +} diff --git a/checkpoint-24000/unet_ema/diffusion_pytorch_model.bin b/checkpoint-24000/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..b2f0ee62193ebf43d9f8c192b9918496169345e9 --- /dev/null +++ b/checkpoint-24000/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b142926c7cd4a67e2e35b248afe316f47ac27ff1bb91ebe65414ab03d04c52c2 +size 454861629 diff --git a/checkpoint-28000/optimizer.bin b/checkpoint-28000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..dbfe4e803b0206282addd14e5994ceb4f6e4ec78 --- /dev/null +++ b/checkpoint-28000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:73af7c0c101869a1feb2a6cc7605b7e36e48496e0a48f41f24b0dee3a89b8a9f +size 909763299 diff --git a/checkpoint-28000/random_states_0.pkl b/checkpoint-28000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..cb179d90b4f40e93d8bbf4f852ba514da5dc1557 --- /dev/null +++ b/checkpoint-28000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:6065f5607f45390340033d5851393bd784127d6694ac6bb97137b2a606d2c82b +size 17811 diff --git a/checkpoint-28000/scaler.pt b/checkpoint-28000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..fc4e7cf6fe6784d74408bb8b4e02d8925bc812ed --- /dev/null +++ b/checkpoint-28000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:ce75f7263a7f2cbcdd8cc2d633c72459a685fdd905798de1603be309149b1eae +size 557 diff --git a/checkpoint-28000/scheduler.bin b/checkpoint-28000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..28455777b627fe48ac12af65c296a4587ad29a3f --- /dev/null +++ b/checkpoint-28000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:d73acb0f508d98a760be42fa04bd15b1ac9d6c3b3d1e99e152de8682ccbd22dd +size 563 diff --git a/checkpoint-28000/unet/config.json b/checkpoint-28000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..7a0c47e18dceccb3d9745c685fa44287c80c72d0 --- /dev/null +++ b/checkpoint-28000/unet/config.json @@ -0,0 +1,46 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.18.0.dev0", + "act_fn": "silu", + "add_attention": true, + "attention_head_dim": 8, + "block_out_channels": [ + 128, + 128, + 256, + 256, + 512, + 512 + ], + "center_input_sample": false, + "class_embed_type": null, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "AttnDownBlock2D", + "DownBlock2D" + ], + "downsample_padding": 1, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "layers_per_block": 2, + "mid_block_scale_factor": 1, + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "out_channels": 3, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_embedding_type": "positional", + "up_block_types": [ + "UpBlock2D", + "AttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ] +} diff --git a/checkpoint-28000/unet/diffusion_pytorch_model.bin b/checkpoint-28000/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..8d8dd926fa676c46d1ca57f1c570a75dafd45b7a --- /dev/null +++ b/checkpoint-28000/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:f21671da83e9b1949562ec788baeeedc624ce8008b0067d54aded874548e10c4 +size 454867965 diff --git a/checkpoint-28000/unet_ema/config.json b/checkpoint-28000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..43d3e3ab7b59afb149b4359b8e9c07aa90ea5c05 --- /dev/null +++ b/checkpoint-28000/unet_ema/config.json @@ -0,0 +1,53 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.18.0.dev0", + "act_fn": "silu", + "add_attention": true, + "attention_head_dim": 8, + "block_out_channels": [ + 128, + 128, + 256, + 256, + 512, + 512 + ], + "center_input_sample": false, + "class_embed_type": null, + "decay": 0.9999, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "AttnDownBlock2D", + "DownBlock2D" + ], + "downsample_padding": 1, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "inv_gamma": 1.0, + "layers_per_block": 2, + "mid_block_scale_factor": 1, + "min_decay": 0.0, + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "optimization_step": 28000, + "out_channels": 3, + "power": 0.75, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_embedding_type": "positional", + "up_block_types": [ + "UpBlock2D", + "AttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "update_after_step": 0, + "use_ema_warmup": true +} diff --git a/checkpoint-28000/unet_ema/diffusion_pytorch_model.bin b/checkpoint-28000/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..156d948334584d33cbf8f38d1dae1c715f047671 --- /dev/null +++ b/checkpoint-28000/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:3d5694f0315b2ed031f38e6a1cc32fbcb13639f6922df8a4fc6636506f1da75d +size 454861629 diff --git a/checkpoint-32000/optimizer.bin b/checkpoint-32000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..472526acbcc70f50a14e34f3eb63dc26d917b3c9 --- /dev/null +++ b/checkpoint-32000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:01c52d95643f0bb212d9bfb4c8549b6e5d91b091fbd9bcfb359a5055546c0427 +size 909763299 diff --git a/checkpoint-32000/random_states_0.pkl b/checkpoint-32000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..4d00930e1ef4aabb0c733b7cf1dbc103e83bbec1 --- /dev/null +++ b/checkpoint-32000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:0631894c81431080b02664d081d2c2ef3ee68949429df53396aa13d27477d1d0 +size 17811 diff --git a/checkpoint-32000/scaler.pt b/checkpoint-32000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..9c2ab57cfcda21d3a1c92c6b8479a6952216ad88 --- /dev/null +++ b/checkpoint-32000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e78e7b9cb7a54cf14184edb98416395ba780edb0f6f836a0ec3b1cadb1c9dccd +size 557 diff --git a/checkpoint-32000/scheduler.bin b/checkpoint-32000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..298b9ce557ecf44d800d0e6a7f3ca38f670b6b06 --- /dev/null +++ b/checkpoint-32000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:390f6322fdc42893f967837f4aa5c56771f666e955f184deda124c2d1d6fbdb2 +size 563 diff --git a/checkpoint-32000/unet/config.json b/checkpoint-32000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..7a0c47e18dceccb3d9745c685fa44287c80c72d0 --- /dev/null +++ b/checkpoint-32000/unet/config.json @@ -0,0 +1,46 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.18.0.dev0", + "act_fn": "silu", + "add_attention": true, + "attention_head_dim": 8, + "block_out_channels": [ + 128, + 128, + 256, + 256, + 512, + 512 + ], + "center_input_sample": false, + "class_embed_type": null, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "AttnDownBlock2D", + "DownBlock2D" + ], + "downsample_padding": 1, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "layers_per_block": 2, + "mid_block_scale_factor": 1, + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "out_channels": 3, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_embedding_type": "positional", + "up_block_types": [ + "UpBlock2D", + "AttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ] +} diff --git a/checkpoint-32000/unet/diffusion_pytorch_model.bin b/checkpoint-32000/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..7de3e5ced5dfbada9de62aeedb39c6d4dccc05b9 --- /dev/null +++ b/checkpoint-32000/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:9c935b2c5f039e0fe0babed1537da6268b66c051058f73f18f1d5673e8d130ad +size 454867965 diff --git a/checkpoint-32000/unet_ema/config.json b/checkpoint-32000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..e5d54722907f83c0b06863b7dad3381b2dcab40b --- /dev/null +++ b/checkpoint-32000/unet_ema/config.json @@ -0,0 +1,53 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.18.0.dev0", + "act_fn": "silu", + "add_attention": true, + "attention_head_dim": 8, + "block_out_channels": [ + 128, + 128, + 256, + 256, + 512, + 512 + ], + "center_input_sample": false, + "class_embed_type": null, + "decay": 0.9999, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "AttnDownBlock2D", + "DownBlock2D" + ], + "downsample_padding": 1, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "inv_gamma": 1.0, + "layers_per_block": 2, + "mid_block_scale_factor": 1, + "min_decay": 0.0, + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "optimization_step": 32000, + "out_channels": 3, + "power": 0.75, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_embedding_type": "positional", + "up_block_types": [ + "UpBlock2D", + "AttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "update_after_step": 0, + "use_ema_warmup": true +} diff --git a/checkpoint-32000/unet_ema/diffusion_pytorch_model.bin b/checkpoint-32000/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..93034ab4ab7cee7c12dce815268216198f445683 --- /dev/null +++ b/checkpoint-32000/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:1d7424988409a5b37275e7cbdaec2564b37dd55b7b37c4b052d6cb9f609eb9c4 +size 454861629 diff --git a/checkpoint-36000/optimizer.bin b/checkpoint-36000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..0fc93caa7bdc75bd49263199e71e38bbd5c043ec --- /dev/null +++ b/checkpoint-36000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:20198161ef837a4abd0eaa16ec7ddc17cc76d4ee652cc69ae6a7d9b211c2017a +size 909763299 diff --git a/checkpoint-36000/random_states_0.pkl b/checkpoint-36000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..e6c65b3d50db04e25941ab7b59d18f23c6c002ba --- /dev/null +++ b/checkpoint-36000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:70e6bf0e7d4155f5d65598442180112aa1a3f7a9d8001cd45fb9125cb6cd67e6 +size 17811 diff --git a/checkpoint-36000/scaler.pt b/checkpoint-36000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..f5421385234f11146e70ef942e3ab4b0ba06e95d --- /dev/null +++ b/checkpoint-36000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:cda1251f1008cbd00481f10c7c064a6377f10272e5ef9bf936b598372a63d934 +size 557 diff --git a/checkpoint-36000/scheduler.bin b/checkpoint-36000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..c8b1f861c452a1c888eb47f8c52efab5923204a4 --- /dev/null +++ b/checkpoint-36000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:094c2356916b91645cd4f0bf155dd395376b04c3699e1222ccb95251c33ffcdd +size 563 diff --git a/checkpoint-36000/unet/config.json b/checkpoint-36000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..7a0c47e18dceccb3d9745c685fa44287c80c72d0 --- /dev/null +++ b/checkpoint-36000/unet/config.json @@ -0,0 +1,46 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.18.0.dev0", + "act_fn": "silu", + "add_attention": true, + "attention_head_dim": 8, + "block_out_channels": [ + 128, + 128, + 256, + 256, + 512, + 512 + ], + "center_input_sample": false, + "class_embed_type": null, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "AttnDownBlock2D", + "DownBlock2D" + ], + "downsample_padding": 1, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "layers_per_block": 2, + "mid_block_scale_factor": 1, + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "out_channels": 3, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_embedding_type": "positional", + "up_block_types": [ + "UpBlock2D", + "AttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ] +} diff --git a/checkpoint-36000/unet/diffusion_pytorch_model.bin b/checkpoint-36000/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..b196f0111ef9225c956fdcb28618d6b315d05d56 --- /dev/null +++ b/checkpoint-36000/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:d3ed2af4f158de209613e83acb0e9f145b8a36b7fdc3f5e1b91f626ced9bb485 +size 454867965 diff --git a/checkpoint-36000/unet_ema/config.json b/checkpoint-36000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..5fe3ef0dd0931d0cee948e0c45cccfaef0b7d329 --- /dev/null +++ b/checkpoint-36000/unet_ema/config.json @@ -0,0 +1,53 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.18.0.dev0", + "act_fn": "silu", + "add_attention": true, + "attention_head_dim": 8, + "block_out_channels": [ + 128, + 128, + 256, + 256, + 512, + 512 + ], + "center_input_sample": false, + "class_embed_type": null, + "decay": 0.9999, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "AttnDownBlock2D", + "DownBlock2D" + ], + "downsample_padding": 1, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "inv_gamma": 1.0, + "layers_per_block": 2, + "mid_block_scale_factor": 1, + "min_decay": 0.0, + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "optimization_step": 36000, + "out_channels": 3, + "power": 0.75, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_embedding_type": "positional", + "up_block_types": [ + "UpBlock2D", + "AttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "update_after_step": 0, + "use_ema_warmup": true +} diff --git a/checkpoint-36000/unet_ema/diffusion_pytorch_model.bin b/checkpoint-36000/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..2db182136cef44ade94cd3a96a700c02b59a97fa --- /dev/null +++ b/checkpoint-36000/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c26cf2daa84ccae750423086eaf32eaaa3b21953308acc932adcc3b8b3d84856 +size 454861629 diff --git a/checkpoint-4000/optimizer.bin b/checkpoint-4000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..f5b36468f75fafb36a85111a5bbbca695a4d3d12 --- /dev/null +++ b/checkpoint-4000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:228eade0c656012bac1323e5dd89dc95210623feedd9ad581acd2702e151b1c0 +size 909763299 diff --git a/checkpoint-4000/random_states_0.pkl b/checkpoint-4000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..45268454553a6a77532ec97e50a0dcfa697c2cbb --- /dev/null +++ b/checkpoint-4000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:41ee7d83bfdff621ff40d087610208cdafebd09d655aba0fd31d7a83758ff15d +size 17811 diff --git a/checkpoint-4000/scaler.pt b/checkpoint-4000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..e6dfc1dc5fe962ebf4423f5d584aeb4e90dc01be --- /dev/null +++ b/checkpoint-4000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:09a61ecdaa4a69ba865e0b9e19169660549c34c60a83537ab5ece413678520ab +size 557 diff --git a/checkpoint-4000/scheduler.bin b/checkpoint-4000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..f060474d900bf832ed3c3ab1011d2ec9405356e7 --- /dev/null +++ b/checkpoint-4000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:52d66767d84ae5f6e0ec8369be437c52becdd141973d6bbb8a55e0be5b3677a5 +size 563 diff --git a/checkpoint-4000/unet/config.json b/checkpoint-4000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..7a0c47e18dceccb3d9745c685fa44287c80c72d0 --- /dev/null +++ b/checkpoint-4000/unet/config.json @@ -0,0 +1,46 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.18.0.dev0", + "act_fn": "silu", + "add_attention": true, + "attention_head_dim": 8, + "block_out_channels": [ + 128, + 128, + 256, + 256, + 512, + 512 + ], + "center_input_sample": false, + "class_embed_type": null, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "AttnDownBlock2D", + "DownBlock2D" + ], + "downsample_padding": 1, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "layers_per_block": 2, + "mid_block_scale_factor": 1, + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "out_channels": 3, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_embedding_type": "positional", + "up_block_types": [ + "UpBlock2D", + "AttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ] +} diff --git a/checkpoint-4000/unet/diffusion_pytorch_model.bin b/checkpoint-4000/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..6a0568ed29c94be2aa12b64f8450845e59f3539c --- /dev/null +++ b/checkpoint-4000/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:41bc4c95559e818beb20cccf6e0e22ca172c2477ce40be0f8c424e9528be1aed +size 454867965 diff --git a/checkpoint-4000/unet_ema/config.json b/checkpoint-4000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..469615e486a259920dce98efe35f2dcc8b6d7b56 --- /dev/null +++ b/checkpoint-4000/unet_ema/config.json @@ -0,0 +1,53 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.18.0.dev0", + "act_fn": "silu", + "add_attention": true, + "attention_head_dim": 8, + "block_out_channels": [ + 128, + 128, + 256, + 256, + 512, + 512 + ], + "center_input_sample": false, + "class_embed_type": null, + "decay": 0.9999, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "AttnDownBlock2D", + "DownBlock2D" + ], + "downsample_padding": 1, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "inv_gamma": 1.0, + "layers_per_block": 2, + "mid_block_scale_factor": 1, + "min_decay": 0.0, + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "optimization_step": 4000, + "out_channels": 3, + "power": 0.75, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_embedding_type": "positional", + "up_block_types": [ + "UpBlock2D", + "AttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "update_after_step": 0, + "use_ema_warmup": true +} diff --git a/checkpoint-4000/unet_ema/diffusion_pytorch_model.bin b/checkpoint-4000/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..a16b5357ada222facde2fe931fcfb05c89396887 --- /dev/null +++ b/checkpoint-4000/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:84466f65426e7750e8e67dd8ac524704d963f30bb68bafd0e87848022e21e646 +size 454861629 diff --git a/checkpoint-8000/optimizer.bin b/checkpoint-8000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..8ca81f8e848555fb0f71cbf93480448c5e2f9f3f --- /dev/null +++ b/checkpoint-8000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:0025b6c3f8760ed7e94a53e7128669e8f2bc583cbcb938de746874919bc7c7d1 +size 909763299 diff --git a/checkpoint-8000/random_states_0.pkl b/checkpoint-8000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..de289e93d5185a3a0fbcb5d85ff326cf278d156f --- /dev/null +++ b/checkpoint-8000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:8806d0388c5926eb5ab6d9cf268542779a54afb4f6a43170d2bb5d6e7e864d4a +size 17811 diff --git a/checkpoint-8000/scaler.pt b/checkpoint-8000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..fd25dc8bd6a1a15851636b4ebafe829cb0ed1b0d --- /dev/null +++ b/checkpoint-8000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b278cd6e09360f31a3d837f80dee4c2ce4d9c9d186a939ecf157e1a0deb793f3 +size 557 diff --git a/checkpoint-8000/scheduler.bin b/checkpoint-8000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..cbbb7de9299e77d1f8a7d1c16e0ab1501a0747cf --- /dev/null +++ b/checkpoint-8000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:a355e459383e97d07fac2dfa1094db40f1744c276be3a43551c4e9e2f9e4c7b8 +size 563 diff --git a/checkpoint-8000/unet/config.json b/checkpoint-8000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..7a0c47e18dceccb3d9745c685fa44287c80c72d0 --- /dev/null +++ b/checkpoint-8000/unet/config.json @@ -0,0 +1,46 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.18.0.dev0", + "act_fn": "silu", + "add_attention": true, + "attention_head_dim": 8, + "block_out_channels": [ + 128, + 128, + 256, + 256, + 512, + 512 + ], + "center_input_sample": false, + "class_embed_type": null, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "AttnDownBlock2D", + "DownBlock2D" + ], + "downsample_padding": 1, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "layers_per_block": 2, + "mid_block_scale_factor": 1, + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "out_channels": 3, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_embedding_type": "positional", + "up_block_types": [ + "UpBlock2D", + "AttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ] +} diff --git a/checkpoint-8000/unet/diffusion_pytorch_model.bin b/checkpoint-8000/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..3463b664ce658f4fde757e7e81baa45f9ac4e3cc --- /dev/null +++ b/checkpoint-8000/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:fbdeba7a3e0e5b9dfbeca5e624f2750670830a4581dd3f0582b1a53b6fda7084 +size 454867965 diff --git a/checkpoint-8000/unet_ema/config.json b/checkpoint-8000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..49e1b205c7a4edccf0cb5bd9949744d13453913a --- /dev/null +++ b/checkpoint-8000/unet_ema/config.json @@ -0,0 +1,53 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.18.0.dev0", + "act_fn": "silu", + "add_attention": true, + "attention_head_dim": 8, + "block_out_channels": [ + 128, + 128, + 256, + 256, + 512, + 512 + ], + "center_input_sample": false, + "class_embed_type": null, + "decay": 0.9999, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "AttnDownBlock2D", + "DownBlock2D" + ], + "downsample_padding": 1, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "inv_gamma": 1.0, + "layers_per_block": 2, + "mid_block_scale_factor": 1, + "min_decay": 0.0, + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "optimization_step": 8000, + "out_channels": 3, + "power": 0.75, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_embedding_type": "positional", + "up_block_types": [ + "UpBlock2D", + "AttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "update_after_step": 0, + "use_ema_warmup": true +} diff --git a/checkpoint-8000/unet_ema/diffusion_pytorch_model.bin b/checkpoint-8000/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..7400c7c582f328552f210005f4ea42218ac7b3a9 --- /dev/null +++ b/checkpoint-8000/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:5822b992ff17ae969274a085f004df16680ef0f696d7a00af393b15c144216a7 +size 454861629 diff --git a/logs/train_unconditional/events.out.tfevents.1686614885.pop-os.1465109.0 b/logs/train_unconditional/events.out.tfevents.1686614885.pop-os.1465109.0 index be379e02da99b38a3b606799aa861e41bdca04f6..ff1e191d6995acfc22a7f26fa611b2daa38dfe52 100644 --- a/logs/train_unconditional/events.out.tfevents.1686614885.pop-os.1465109.0 +++ b/logs/train_unconditional/events.out.tfevents.1686614885.pop-os.1465109.0 @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:e8f3eb21388083813d14c49e34e84cb78938b55b916ab363712321d09ebf50d7 -size 875516 +oid sha256:4db6149b932302a09c993f4d17d1be548d3f8a030b3fcfcca08cb4dd76137b8c +size 7449159 diff --git a/unet/diffusion_pytorch_model.bin b/unet/diffusion_pytorch_model.bin index dc1ee5b43a31e03ad4fa9b7d79bc9b2e6ae1dd48..6d46e85b358ec88a6acaaf53532cd5e9279380f2 100644 --- a/unet/diffusion_pytorch_model.bin +++ b/unet/diffusion_pytorch_model.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:348d8fa8bb2cd800f798383622250f2887b71b2ce89f1fbcbac64624ad597e47 +oid sha256:fda2460ce738e5b161e4d0b71af2e5722a1023eb618d7a87ba6c2dc021d0a8b0 size 454867965