diff --git a/checkpoint-3500/optimizer.bin b/checkpoint-3500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..32ad54acafb50c8dcf3d3c792195e945912c4bbc --- /dev/null +++ b/checkpoint-3500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:857c4160ebd038a89beae32464884e7ff4553374ff50342451340d6bbc4908b3 +size 57349893 diff --git a/checkpoint-3500/random_states_0.pkl b/checkpoint-3500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..953830e10e18e8a1a5e997f087e37eb1dc54c8ad --- /dev/null +++ b/checkpoint-3500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:5a39e36d6736323d7bd2d9d564fec12c4e6124fc416cdc0404832a279d62bc9b +size 14599 diff --git a/checkpoint-3500/scaler.pt b/checkpoint-3500/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..44e7bbfd017c097f419247a9c5de77cb216339ac --- /dev/null +++ b/checkpoint-3500/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:d4cfe72973031c5a482f6e9ed0cb664a6aa7f63116e294e2b1f72d360d9b033c +size 557 diff --git a/checkpoint-3500/scheduler.bin b/checkpoint-3500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..e4292e6a7c5a1d9fc1d593d22ed7489829279ef4 --- /dev/null +++ b/checkpoint-3500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:28c0470b823f736deebcede4713009f196c80a594c246bee581e19e274108e00 +size 563 diff --git a/checkpoint-3500/unet/config.json b/checkpoint-3500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..99ebec7e97d8731e19f09a61fd292f2b02845211 --- /dev/null +++ b/checkpoint-3500/unet/config.json @@ -0,0 +1,40 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "add_attention": true, + "attention_head_dim": 8, + "block_out_channels": [ + 64, + 64, + 128, + 128 + ], + "center_input_sample": false, + "class_embed_type": null, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "AttnDownBlock2D" + ], + "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": [ + "AttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ] +} diff --git a/checkpoint-3500/unet/diffusion_pytorch_model.bin b/checkpoint-3500/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..6214c759c1f479145dd0376812c6b065c84ef042 --- /dev/null +++ b/checkpoint-3500/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:a9a9fb4136efd63e31516405125f96c40f49eb4e089977a5042620688478b956 +size 28664529 diff --git a/checkpoint-3500/unet_ema/config.json b/checkpoint-3500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..75a80bca36877d5409a80138cd0057d7255e6784 --- /dev/null +++ b/checkpoint-3500/unet_ema/config.json @@ -0,0 +1,47 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "add_attention": true, + "attention_head_dim": 8, + "block_out_channels": [ + 64, + 64, + 128, + 128 + ], + "center_input_sample": false, + "class_embed_type": null, + "decay": 0.9999, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "AttnDownBlock2D" + ], + "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": 3500, + "out_channels": 3, + "power": 0.75, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_embedding_type": "positional", + "up_block_types": [ + "AttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "update_after_step": 0, + "use_ema_warmup": true +} diff --git a/checkpoint-3500/unet_ema/diffusion_pytorch_model.bin b/checkpoint-3500/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..a560a1f93e29390a9f32d0cf8357d4a7fd326c56 --- /dev/null +++ b/checkpoint-3500/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:577260592475a7f688e1f70b614ffb6eeb8b77058c02cbbb28f417b178a4114c +size 28659921 diff --git a/checkpoint-4000/optimizer.bin b/checkpoint-4000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..9c9a440c49c843f8dc48fd7f6562f62f1fbe9f7d --- /dev/null +++ b/checkpoint-4000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:453b618d6cdaab486d018cedfddd108bbf4e0954a4762a4f5ce9242c19c56261 +size 57349893 diff --git a/checkpoint-4000/random_states_0.pkl b/checkpoint-4000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..d1c0b06e17786dd4f7f589866af9aca365fb463d --- /dev/null +++ b/checkpoint-4000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:3d921ceb28cfcbd2bc680e2b675c8603e4ecb96020898d4d29b037f15bf35ee4 +size 14599 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..5e327c441807502689924cd9b0569516f11754d1 --- /dev/null +++ b/checkpoint-4000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:9252096a517fa7e94d8d4a4eec7a6cc10a01349e5774d2a0a74e2a7279c5a34e +size 563 diff --git a/checkpoint-4000/unet/config.json b/checkpoint-4000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..99ebec7e97d8731e19f09a61fd292f2b02845211 --- /dev/null +++ b/checkpoint-4000/unet/config.json @@ -0,0 +1,40 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "add_attention": true, + "attention_head_dim": 8, + "block_out_channels": [ + 64, + 64, + 128, + 128 + ], + "center_input_sample": false, + "class_embed_type": null, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "AttnDownBlock2D" + ], + "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": [ + "AttnUpBlock2D", + "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..f57740ea90839d2388aeb44ce6299e5844e52327 --- /dev/null +++ b/checkpoint-4000/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:a4b490fda7a440bb8b16aa5cb87745c7113d61f96ded5f1bff45e6bed46c472a +size 28664529 diff --git a/checkpoint-4000/unet_ema/config.json b/checkpoint-4000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..d5345a736fc75056544c49192a779f52d7358c07 --- /dev/null +++ b/checkpoint-4000/unet_ema/config.json @@ -0,0 +1,47 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "add_attention": true, + "attention_head_dim": 8, + "block_out_channels": [ + 64, + 64, + 128, + 128 + ], + "center_input_sample": false, + "class_embed_type": null, + "decay": 0.9999, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "AttnDownBlock2D" + ], + "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": [ + "AttnUpBlock2D", + "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..9626e7dae50d961b6d5e84ba64fc1b1892891f45 --- /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:a169796dca4f34b0fa4500be7932c043167536a4960d03e0ecdde7c7855d7f7f +size 28659921 diff --git a/checkpoint-4500/optimizer.bin b/checkpoint-4500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..45eb8c3e6c1987d8a56ba5ec9144db39e8ec1f53 --- /dev/null +++ b/checkpoint-4500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:69b03b582ff26f3aa8affd7ab316c0a813a935ac9edd57e5b84ded7c009b9486 +size 57349893 diff --git a/checkpoint-4500/random_states_0.pkl b/checkpoint-4500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..44b3648dba88b5bb32eb21f10b5f20a3473d3935 --- /dev/null +++ b/checkpoint-4500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:9155e72365b34a8ef13c09268ce028f442463e53a753c3ccbe24a06b6f86e215 +size 14599 diff --git a/checkpoint-4500/scaler.pt b/checkpoint-4500/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..8c98e356000f5210c0d0307e76e41b897f221bfa --- /dev/null +++ b/checkpoint-4500/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:945e2cc9e717c41b0b1286d550bee32d1e1a6cb3d9a42935c0294b123372c89f +size 557 diff --git a/checkpoint-4500/scheduler.bin b/checkpoint-4500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..7b007ca027e5a2da5bd4b0f8207f969e2cc5b8d7 --- /dev/null +++ b/checkpoint-4500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:214e859c89f066cadce348fc24c0fa8bbba31d427c88669e04fd79a2d19fae74 +size 563 diff --git a/checkpoint-4500/unet/config.json b/checkpoint-4500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..99ebec7e97d8731e19f09a61fd292f2b02845211 --- /dev/null +++ b/checkpoint-4500/unet/config.json @@ -0,0 +1,40 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "add_attention": true, + "attention_head_dim": 8, + "block_out_channels": [ + 64, + 64, + 128, + 128 + ], + "center_input_sample": false, + "class_embed_type": null, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "AttnDownBlock2D" + ], + "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": [ + "AttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ] +} diff --git a/checkpoint-4500/unet/diffusion_pytorch_model.bin b/checkpoint-4500/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..b3c7df9bbe1a0834ead7650c5ae8364ceaee54c4 --- /dev/null +++ b/checkpoint-4500/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:2d6f0f4489d803778053a05e444c617a868bea0483bfb9f6b70f622ca2810db7 +size 28664529 diff --git a/checkpoint-4500/unet_ema/config.json b/checkpoint-4500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..662cdf7cb10b5467b74f40df06b1bf3765126e99 --- /dev/null +++ b/checkpoint-4500/unet_ema/config.json @@ -0,0 +1,47 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "add_attention": true, + "attention_head_dim": 8, + "block_out_channels": [ + 64, + 64, + 128, + 128 + ], + "center_input_sample": false, + "class_embed_type": null, + "decay": 0.9999, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "AttnDownBlock2D" + ], + "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": 4500, + "out_channels": 3, + "power": 0.75, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_embedding_type": "positional", + "up_block_types": [ + "AttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "update_after_step": 0, + "use_ema_warmup": true +} diff --git a/checkpoint-4500/unet_ema/diffusion_pytorch_model.bin b/checkpoint-4500/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..2dce6e6b006ef8f0024e76c27e0f5b47f437768e --- /dev/null +++ b/checkpoint-4500/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:8bb2957f40a6de8a2d528fb2adaa12c0ddacbd1d086239b75635ae6f44d361b6 +size 28659921 diff --git a/checkpoint-5000/optimizer.bin b/checkpoint-5000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..c9d2ec4847a19bca7c8deb47ac1cde2b7844e5df --- /dev/null +++ b/checkpoint-5000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e884011e30b9bda249ceb69c53b189b17ffda00cdcc597fb27cf1333adb50ea0 +size 57349893 diff --git a/checkpoint-5000/random_states_0.pkl b/checkpoint-5000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..b7f9082791cda8602746ae979bbc6cf232910cd4 --- /dev/null +++ b/checkpoint-5000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:5bfd12af685259f32372f3b6c1110ca9abd4f68dddb9dce3389c0746e55d9b46 +size 14599 diff --git a/checkpoint-5000/scaler.pt b/checkpoint-5000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..9a0a5a2d6700d39df12b85e2a8ee02926333f46a --- /dev/null +++ b/checkpoint-5000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:a41313b6733b65c1acca917a7c2ced06503c11b898e6cd5de1e6a9f13bfb54ef +size 557 diff --git a/checkpoint-5000/scheduler.bin b/checkpoint-5000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..4e2691863dd8ca06c7f3480420df2e60abb63dec --- /dev/null +++ b/checkpoint-5000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:f590d7a6c15999ba9764947d8d26063e379b436d2ace22cb10c4ff28e5df3fd0 +size 563 diff --git a/checkpoint-5000/unet/config.json b/checkpoint-5000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..99ebec7e97d8731e19f09a61fd292f2b02845211 --- /dev/null +++ b/checkpoint-5000/unet/config.json @@ -0,0 +1,40 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "add_attention": true, + "attention_head_dim": 8, + "block_out_channels": [ + 64, + 64, + 128, + 128 + ], + "center_input_sample": false, + "class_embed_type": null, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "AttnDownBlock2D" + ], + "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": [ + "AttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ] +} diff --git a/checkpoint-5000/unet/diffusion_pytorch_model.bin b/checkpoint-5000/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..793069cebe84380f97eea9dab82270445b14da8f --- /dev/null +++ b/checkpoint-5000/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:d361722691ab26cee85ac47605ea72da7860c533b1ffaa9c50221292e6d16cc7 +size 28664529 diff --git a/checkpoint-5000/unet_ema/config.json b/checkpoint-5000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..80d6b0dfd18b82d4c28b1918982dff3a2728cb70 --- /dev/null +++ b/checkpoint-5000/unet_ema/config.json @@ -0,0 +1,47 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "add_attention": true, + "attention_head_dim": 8, + "block_out_channels": [ + 64, + 64, + 128, + 128 + ], + "center_input_sample": false, + "class_embed_type": null, + "decay": 0.9999, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "AttnDownBlock2D" + ], + "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": 5000, + "out_channels": 3, + "power": 0.75, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_embedding_type": "positional", + "up_block_types": [ + "AttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "update_after_step": 0, + "use_ema_warmup": true +} diff --git a/checkpoint-5000/unet_ema/diffusion_pytorch_model.bin b/checkpoint-5000/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..1a5aa9690e3e0f40398938d966755d4208e99ee4 --- /dev/null +++ b/checkpoint-5000/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:dc4aa2079f01ef456003060b3c5e6ac74546094cbfeb453fcc39386c711544d8 +size 28659921 diff --git a/checkpoint-5500/optimizer.bin b/checkpoint-5500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..155d68564d90e892b2c7ba12fa8c90f1cd44c26c --- /dev/null +++ b/checkpoint-5500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:4174d404753db17427ce124887307b1a6fb71cacbce8d6861670436b6f8ea5f6 +size 57349893 diff --git a/checkpoint-5500/random_states_0.pkl b/checkpoint-5500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..02fa2557624b44419ae188f625803728e1a20f79 --- /dev/null +++ b/checkpoint-5500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:1b42287ca8bb4094718842385d9081712f046ef9c42faf869378ef939bedba06 +size 14599 diff --git a/checkpoint-5500/scaler.pt b/checkpoint-5500/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..d7dc8ce86b334ba316696a99f82e1bc011bdcdcd --- /dev/null +++ b/checkpoint-5500/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:367928c873a79d90abcb66db0f8b320eea7346dc9bc779b4e7963dfc82cf2ada +size 557 diff --git a/checkpoint-5500/scheduler.bin b/checkpoint-5500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..3a919998d1d43ac0520975f8bec08b0827f66156 --- /dev/null +++ b/checkpoint-5500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:1f45136b619318b2f56c303c9f5da3e6ec4e6bd0719a0a44f8f2cbd769163605 +size 563 diff --git a/checkpoint-5500/unet/config.json b/checkpoint-5500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..99ebec7e97d8731e19f09a61fd292f2b02845211 --- /dev/null +++ b/checkpoint-5500/unet/config.json @@ -0,0 +1,40 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "add_attention": true, + "attention_head_dim": 8, + "block_out_channels": [ + 64, + 64, + 128, + 128 + ], + "center_input_sample": false, + "class_embed_type": null, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "AttnDownBlock2D" + ], + "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": [ + "AttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ] +} diff --git a/checkpoint-5500/unet/diffusion_pytorch_model.bin b/checkpoint-5500/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..891adadf708df3d6ef14b259d5460b742248b653 --- /dev/null +++ b/checkpoint-5500/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:6f0ab5d4eb81651535424168bc131182608e900f0cbed1a9596aa4cb47624c36 +size 28664529 diff --git a/checkpoint-5500/unet_ema/config.json b/checkpoint-5500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..d560d46788570fc22d2e43b637a51292c4b707f8 --- /dev/null +++ b/checkpoint-5500/unet_ema/config.json @@ -0,0 +1,47 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "add_attention": true, + "attention_head_dim": 8, + "block_out_channels": [ + 64, + 64, + 128, + 128 + ], + "center_input_sample": false, + "class_embed_type": null, + "decay": 0.9999, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "AttnDownBlock2D" + ], + "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": 5500, + "out_channels": 3, + "power": 0.75, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_embedding_type": "positional", + "up_block_types": [ + "AttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "update_after_step": 0, + "use_ema_warmup": true +} diff --git a/checkpoint-5500/unet_ema/diffusion_pytorch_model.bin b/checkpoint-5500/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..d7664140e74200dbb554ec64b0119d5a52a39598 --- /dev/null +++ b/checkpoint-5500/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:96cbd220a6fc6be917c959cf8d1a8d7875c25310fd08b4031faa41544b149752 +size 28659921 diff --git a/checkpoint-6000/optimizer.bin b/checkpoint-6000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..cbe5df1515b1170f06556f421886654920efda15 --- /dev/null +++ b/checkpoint-6000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:647d89841bc6b614e5bd5bdf891d600c940783b5d415d028a452f495a9b5d3e5 +size 57349893 diff --git a/checkpoint-6000/random_states_0.pkl b/checkpoint-6000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..498cd27b575591d5c1d45f6115ff3ec52a742638 --- /dev/null +++ b/checkpoint-6000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7a01c74585d4beda8eedb5381eec5d4b1f52a71a31d81ccc1603f5155b9da4e1 +size 14599 diff --git a/checkpoint-6000/scaler.pt b/checkpoint-6000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..2fd17ac3026715d5a68d70922df4742631319059 --- /dev/null +++ b/checkpoint-6000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:9a00ef97fc41c0d653df466126d855f16646dd10726d3ac99a23f55ca70ec19a +size 557 diff --git a/checkpoint-6000/scheduler.bin b/checkpoint-6000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..e86b029f3ec573cba40bd419e9dbfe7f55eeb34e --- /dev/null +++ b/checkpoint-6000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:cfd2e8e2a391136ab25982946d62f9b0898b21f357d4c2d18ca0a5194aa9cba7 +size 563 diff --git a/checkpoint-6000/unet/config.json b/checkpoint-6000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..99ebec7e97d8731e19f09a61fd292f2b02845211 --- /dev/null +++ b/checkpoint-6000/unet/config.json @@ -0,0 +1,40 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "add_attention": true, + "attention_head_dim": 8, + "block_out_channels": [ + 64, + 64, + 128, + 128 + ], + "center_input_sample": false, + "class_embed_type": null, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "AttnDownBlock2D" + ], + "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": [ + "AttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ] +} diff --git a/checkpoint-6000/unet/diffusion_pytorch_model.bin b/checkpoint-6000/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..c03f1d054209e604492efa06085f47555c736165 --- /dev/null +++ b/checkpoint-6000/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:d1f89c420437050e35dcef8d2934d2dddc054d1d675e5829048fda277ea16e3f +size 28664529 diff --git a/checkpoint-6000/unet_ema/config.json b/checkpoint-6000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..6db4c0df40f2abc306a0d5baf1302ca930a306c7 --- /dev/null +++ b/checkpoint-6000/unet_ema/config.json @@ -0,0 +1,47 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "add_attention": true, + "attention_head_dim": 8, + "block_out_channels": [ + 64, + 64, + 128, + 128 + ], + "center_input_sample": false, + "class_embed_type": null, + "decay": 0.9999, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "AttnDownBlock2D" + ], + "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": 6000, + "out_channels": 3, + "power": 0.75, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_embedding_type": "positional", + "up_block_types": [ + "AttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "update_after_step": 0, + "use_ema_warmup": true +} diff --git a/checkpoint-6000/unet_ema/diffusion_pytorch_model.bin b/checkpoint-6000/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..7deb80655ffc8b639006889ca97ffee8e9fc69c8 --- /dev/null +++ b/checkpoint-6000/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:1261bb60efc92b8b82e267af20739da71ca081d360707089392e23d5bc9ede08 +size 28659921 diff --git a/checkpoint-6500/optimizer.bin b/checkpoint-6500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..938c706709fdecbf8a139c3505b8ebf3f62774ff --- /dev/null +++ b/checkpoint-6500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:3d23feb24faf141d284acae3a5924d24f85f0ec5c8858d69fede6eafa9691aa7 +size 57349893 diff --git a/checkpoint-6500/random_states_0.pkl b/checkpoint-6500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..03b6f728414822ac633f0d5918444aaf5394562d --- /dev/null +++ b/checkpoint-6500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:75b77c3329d138d12edacc9b33420bad1b1bd8ae4cbb8f909a1d6d147ce5193f +size 14599 diff --git a/checkpoint-6500/scaler.pt b/checkpoint-6500/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..40275aa28a49ee92274a571543131266eb2b70e2 --- /dev/null +++ b/checkpoint-6500/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:443cc2fd4ac32450c4c0d394c84b6f1899943a2f8743462b4026e2d2fe3e5de6 +size 557 diff --git a/checkpoint-6500/scheduler.bin b/checkpoint-6500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..904c1f865a690d13b600e1af1f8f8e85c3864fda --- /dev/null +++ b/checkpoint-6500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:dc508bbc786f8beac5396d49b4539ef67516fd07df126018239457c755cdb0e0 +size 563 diff --git a/checkpoint-6500/unet/config.json b/checkpoint-6500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..99ebec7e97d8731e19f09a61fd292f2b02845211 --- /dev/null +++ b/checkpoint-6500/unet/config.json @@ -0,0 +1,40 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "add_attention": true, + "attention_head_dim": 8, + "block_out_channels": [ + 64, + 64, + 128, + 128 + ], + "center_input_sample": false, + "class_embed_type": null, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "AttnDownBlock2D" + ], + "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": [ + "AttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ] +} diff --git a/checkpoint-6500/unet/diffusion_pytorch_model.bin b/checkpoint-6500/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..94e79bebb98cb732160558071058f573d9b1fb6f --- /dev/null +++ b/checkpoint-6500/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:38799ebeef0ee067d023e6f1cdc8767b95fe5fec9152d7e50fe3c7140675fece +size 28664529 diff --git a/checkpoint-6500/unet_ema/config.json b/checkpoint-6500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..0c38dd0abf86c7c09ce9f1a8ddb062d05ce86569 --- /dev/null +++ b/checkpoint-6500/unet_ema/config.json @@ -0,0 +1,47 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "add_attention": true, + "attention_head_dim": 8, + "block_out_channels": [ + 64, + 64, + 128, + 128 + ], + "center_input_sample": false, + "class_embed_type": null, + "decay": 0.9999, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "AttnDownBlock2D" + ], + "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": 6500, + "out_channels": 3, + "power": 0.75, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_embedding_type": "positional", + "up_block_types": [ + "AttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "update_after_step": 0, + "use_ema_warmup": true +} diff --git a/checkpoint-6500/unet_ema/diffusion_pytorch_model.bin b/checkpoint-6500/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..05110a2c33aa1beb8f21b59c825bb795258045e2 --- /dev/null +++ b/checkpoint-6500/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:0129e3ea13ca9925094c40bbec18b78bdf89c6f9ad0b352528d3d192ea4d44e3 +size 28659921 diff --git a/unet/diffusion_pytorch_model.bin b/unet/diffusion_pytorch_model.bin index ab290db3936c1d4d9e9fe61185098277a33f6d68..101cba63865517197c5b6b487545601eb5353a0b 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:0df3c8a6c84a0db7ad7898d6446e5cd05c68405d54065e15bd992c29ac273d3d +oid sha256:04c1a1c3d41a832e5986bfab50f29688ae667f258e813b355771ec27ea42c13b size 28664529