diff --git a/checkpoint-12000/optimizer.bin b/checkpoint-12000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..b0e0d711e5a3c09f9badc73d6fa4d8d04b3a3015 --- /dev/null +++ b/checkpoint-12000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e3d55030a5e3c58f468b8ab69daec7fba61010fa6f190d25362cf3923518ec9a +size 909763299 diff --git a/checkpoint-12000/random_states_0.pkl b/checkpoint-12000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..53805057f8b9ba708f6a07e05b3265b37dc7f68b --- /dev/null +++ b/checkpoint-12000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e320c12bb901f705c68ae669440170e21c14d0d12147c6373fe3622b5e4de5c7 +size 17683 diff --git a/checkpoint-12000/scaler.pt b/checkpoint-12000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..8b37c8acb334cd79c5485ba60ac672a43126b172 --- /dev/null +++ b/checkpoint-12000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:26d6de29ea919977d83210dd033c3f425c21af0376a795c6537bdeb3983f2275 +size 557 diff --git a/checkpoint-12000/scheduler.bin b/checkpoint-12000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..8d45e490283a9b10b92f1ffb2b7666f0a947e8cf --- /dev/null +++ b/checkpoint-12000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:de090989ce3d18a5377f4ad473b21ca0e18ef0cce53c53fa2bb75b9b56810fd2 +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..8f87bcd3ecffc611d0aa45529e4bd25b6a817c0e --- /dev/null +++ b/checkpoint-12000/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:56753d6727c87c98913e2f8d7f5a26334b69c1b0fcc6e0a5b13ddfd7c533786b +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..ef48956074bcb6908f6a2171b75bc930e15ab52c --- /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:9bc71f4dc5cbcfadadbc9d571303e7c7a9397028021082aef8eaa1d833ee8b76 +size 454861629 diff --git a/checkpoint-16000/optimizer.bin b/checkpoint-16000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..fa99ffc5fc81ab9d5ca6b3c708a8473e51cbf2d6 --- /dev/null +++ b/checkpoint-16000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:267784f6d6fb840123069c61107f81ac0295bd062f0723a9e92e1c1cf14ed439 +size 909763299 diff --git a/checkpoint-16000/random_states_0.pkl b/checkpoint-16000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..31ea1512ced7bbe1dd5abbcfe9dfe231063e5e65 --- /dev/null +++ b/checkpoint-16000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:74070726c6c335df434f0c94df6326cc2aa22d5f3d90c1bfcb59f2c599504024 +size 17683 diff --git a/checkpoint-16000/scaler.pt b/checkpoint-16000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..6825e1eb88c50f56cbf39db48a6cbd69bfe396a4 --- /dev/null +++ b/checkpoint-16000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:01b42b13d2fcec7a742f245ee18dff35b2e9a09912fb4cc4024411cf01c63c72 +size 557 diff --git a/checkpoint-16000/scheduler.bin b/checkpoint-16000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..642dfea9f3d3f28c23da3996aad941071d21f553 --- /dev/null +++ b/checkpoint-16000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c3b070a1f4b1688a1891d5f0790ea5ac14af65467c99cd6b3f6b892ff61d3a6a +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..ae132f2a51d19ad65d3215e592e437ab6dab2ddc --- /dev/null +++ b/checkpoint-16000/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:91fcce1ec9e5ed312baed916d6395abc0a098924fb69fa7b2343bb1cf108fd74 +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..bd8bac780d5ff9bce858fa3a85cd75cc3494cf24 --- /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:8b14c4cc1a39ca42f36963a4a48777c672b4512ffebd2be7ac96b27fb7ac4d66 +size 454861629 diff --git a/checkpoint-20000/optimizer.bin b/checkpoint-20000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..00add3aa35bdc7fefa4e77ae137fa4a25384e52d --- /dev/null +++ b/checkpoint-20000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:0f4dc153588a7427fd43e11d9bac4b6c0b5e1913fa03dc94769c85ff4f676dd2 +size 909763299 diff --git a/checkpoint-20000/random_states_0.pkl b/checkpoint-20000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..f906c4fc76eae50a30457135d0d878a849fc89b7 --- /dev/null +++ b/checkpoint-20000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:f6e1f783e379b2c954bfca725606fe513f9fbbeceae128b4112c514da73eafc6 +size 17683 diff --git a/checkpoint-20000/scaler.pt b/checkpoint-20000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..dd15e6ef04befac935cacc59ffa81e6ea4b2d0d6 --- /dev/null +++ b/checkpoint-20000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:57cfa224356ca2e8f0d24e3b6a413a6d52de3029aec758bb162a30355e125313 +size 557 diff --git a/checkpoint-20000/scheduler.bin b/checkpoint-20000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..57e0f841c1a8eebdf416cbffe96735e16c4d3fb6 --- /dev/null +++ b/checkpoint-20000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:6b22ca5451a07ff82e824065d284607067c9196db903a8ebebbbdd42056f38ae +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..7c0b2fa07a6f9655bb2aeffc064ef384f131d80e --- /dev/null +++ b/checkpoint-20000/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:d988c6bf7b7979fee4537f555202a85238cbcca13ddf780b3c2ecceed83ff840 +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..3c91cf808d982d8020eb257c0d37dd4bda73be55 --- /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:e48d5f606a279dc0279c78ef63edf48dbf909709852e377fac7cd385184a1a98 +size 454861629 diff --git a/checkpoint-24000/optimizer.bin b/checkpoint-24000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..1e8dc1a735fccd707f79546191b753b58bfb8396 --- /dev/null +++ b/checkpoint-24000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:19ec15b94626fd81828ee35601fa8701a7d9b6bd5f4e37d22a126b8d9a0d988c +size 909763299 diff --git a/checkpoint-24000/random_states_0.pkl b/checkpoint-24000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..89b534251274d641694cd46725cfdbc9ac7d538c --- /dev/null +++ b/checkpoint-24000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:4d56b55545fd7cbdfec44ba9a62eb7d496e2358cd7603a968097e3be4a4f5a2c +size 17683 diff --git a/checkpoint-24000/scaler.pt b/checkpoint-24000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..3f17921c569855ee9b5a4889e8ca9ee62cdb1b34 --- /dev/null +++ b/checkpoint-24000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:a594c1aebd1597cb8a8c776f96c7123125e5a044c0d7da8a3ecd1d1351d04269 +size 557 diff --git a/checkpoint-24000/scheduler.bin b/checkpoint-24000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..c12d6346000122a8be4b9d0b6890344245da857f --- /dev/null +++ b/checkpoint-24000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:87635a58ae00fc8e9fad593e909b4448947282e68a523796e8dd5fafa6c06d02 +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..6ad1356290916360e7fecf2dd754506c8809866f --- /dev/null +++ b/checkpoint-24000/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:0979397fba99aca1fd4c1e33eb7465576f19cf672b2dde8a52b5aaa84140328e +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..637735eec9a4182b422dc3693cc62e4d25fe825b --- /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:f386d71354694ccc45a3e1da2fad7ab5376cf8d1b8339933ed56124b0a7dea6d +size 454861629 diff --git a/checkpoint-28000/optimizer.bin b/checkpoint-28000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..969982aa649fd4de09c106d3d4af98edb2426094 --- /dev/null +++ b/checkpoint-28000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:1245f6368fb6dc6ef66ed548f023615a1d4f000a12bfd95a2c72379bee85d1be +size 909763299 diff --git a/checkpoint-28000/random_states_0.pkl b/checkpoint-28000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..2feffaa0754dafc57eeeb39d091f3ffed4533a58 --- /dev/null +++ b/checkpoint-28000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:0c78a7c349f6b91ad5076b72b3c7e8e5db809f644bdfa9e9b1653fcc26518122 +size 17683 diff --git a/checkpoint-28000/scaler.pt b/checkpoint-28000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..e1f29e88a498204fc9e94f787250b428b9e492e8 --- /dev/null +++ b/checkpoint-28000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:d7a83b2bae3e0e43cd11d87e066423b02498413138aff149d12df632ef6db55f +size 557 diff --git a/checkpoint-28000/scheduler.bin b/checkpoint-28000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..ed2b76b4247034d7942b8d60766053c0d3073ebe --- /dev/null +++ b/checkpoint-28000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:4f39eebaa013b059f31e854d6301e4397e7f0739f719b1149a08b068b719e6de +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..5ec21b0190af695b9e33c2e966113ad2f863bc3c --- /dev/null +++ b/checkpoint-28000/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:4424cd6c7ccf224154e25e0c66babb66fe35637799f0b66645fb0eb08f5c3e2c +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..4fed8aaff6f817dbac2db445294885923be4b4a1 --- /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:6efacfb64b4d31604634b7df8ec114134b4a98f8665a1cd374d0766da68afb34 +size 454861629 diff --git a/checkpoint-32000/optimizer.bin b/checkpoint-32000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..37f977aea5ac72be74c4f75528eb0b480f577922 --- /dev/null +++ b/checkpoint-32000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e20e3b920c505b1b72bb0ca3081656685bbaba1036390af8b5e45d71ad40289b +size 909763299 diff --git a/checkpoint-32000/random_states_0.pkl b/checkpoint-32000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..b2bb5ebc76af1946557ed6a67b3ccd013c5b9559 --- /dev/null +++ b/checkpoint-32000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:2b7c7db84236c256dcaa6993882ee38dcb03a58912e4bc2ff20b6dba28adfd17 +size 17683 diff --git a/checkpoint-32000/scaler.pt b/checkpoint-32000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..2deb8c96802b75bce2cff389a46f35698e2a2e6e --- /dev/null +++ b/checkpoint-32000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:a6bc158da9d3dca629cc8183f6da1e9a910d3f11a68a7ed50a9fde1ddb4892a7 +size 557 diff --git a/checkpoint-32000/scheduler.bin b/checkpoint-32000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..70a42cad02e8198b42b20713b1e6e83b72964625 --- /dev/null +++ b/checkpoint-32000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c7e13b89af62e92078bbf74068ec2208856396852372ca71ab501101ebad72ea +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..de0bae77c2d23e41a60b9b4c3ec2da619945ea34 --- /dev/null +++ b/checkpoint-32000/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:11f862c0f89db432354046f155bad096b7e6809fe12bdb841ae7dd29868c2edc +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..05084592946064dc0daedb582e6920aa8e283340 --- /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:a23fb9642e33648765bc9d63334b5f94f9614701cf90c87c830ee92be0620888 +size 454861629 diff --git a/checkpoint-36000/optimizer.bin b/checkpoint-36000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..f5c0c5acdd9a421e60e4805083bcaac01d680c90 --- /dev/null +++ b/checkpoint-36000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:78f3907d0dae9f3db1a583db2c0264f606309ef427848232f595c49553fd4322 +size 909763299 diff --git a/checkpoint-36000/random_states_0.pkl b/checkpoint-36000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..e994ba2e4694425902534dc06171960681267ee8 --- /dev/null +++ b/checkpoint-36000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:2878d2b5e1fe46cb341aa22e8e16328c79e0860b71ceed61570dc4410db177a0 +size 17683 diff --git a/checkpoint-36000/scaler.pt b/checkpoint-36000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..aee15e064c75f6ad9b3888cf66f01e8fa44f3f6e --- /dev/null +++ b/checkpoint-36000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:a831ed5071b96dfc638a32b487378a687f5733c80b2147cab4b8789583061322 +size 557 diff --git a/checkpoint-36000/scheduler.bin b/checkpoint-36000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..65a4c7b97196e2c48a8c3b454a0d8c57e42af106 --- /dev/null +++ b/checkpoint-36000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:61787a6d79eb002f5606225fa984af1ddcfe1b30f5470922051959c2e11e28a8 +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..6c0f9bf47c6649be40364bed4fd42f47742167d0 --- /dev/null +++ b/checkpoint-36000/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:61ba8cdd795dd6818b68d8f816370a8bc2e634b134b5163cc3f7312544c8fba2 +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..1876a5aa6b5888a2df890ca83cbe38d4af1c36e1 --- /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:30180efba2d138c68aa0ef18bf1be0a3523d79a967f9d260c5b0dedccc848148 +size 454861629 diff --git a/checkpoint-4000/optimizer.bin b/checkpoint-4000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..7374282993b43de28c76f902bb760a62a237f79d --- /dev/null +++ b/checkpoint-4000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:9f95d8c513003e0112caf6b24c9f195584ffa852fe8f7913e6026235e436fdae +size 909763299 diff --git a/checkpoint-4000/random_states_0.pkl b/checkpoint-4000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..74329b328c1108986fcf445dde8fc5332f5dada8 --- /dev/null +++ b/checkpoint-4000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:567599e65ffe9606963eda6864b88e9abe96c75a46417bcced0819bf93dea6e3 +size 17683 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..92b90f66f51071cc9f59468606d60885b8401bca --- /dev/null +++ b/checkpoint-4000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:0b809e7eb931af5117dc2abc3bdef4505698e9835fcf92056faf7ded2c2e5323 +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..06b07dbd443d825de529aad84a2092a3c3a5b56a --- /dev/null +++ b/checkpoint-4000/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:a733b9ab51c6ded674e628d0a674f686c1ded3e14a29cf5789520fbdb59d75b9 +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..5b99d44bb6c85efbae78aeb2eca9deeb7cd99409 --- /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:bd2bef8426973078fa731affd4c3777ff3be03e9d834c71ad9f4db7a79e994bc +size 454861629 diff --git a/checkpoint-8000/optimizer.bin b/checkpoint-8000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..60bd28bb08b53e078086ebc961fce0827890c4a7 --- /dev/null +++ b/checkpoint-8000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:99b8396a67b358a3ed818be0c20b6d6db8b7330dc8d4b2da6e97b60d962466ee +size 909763299 diff --git a/checkpoint-8000/random_states_0.pkl b/checkpoint-8000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..3258bcdda9fd39ffca4500f783b523e844d8dfe5 --- /dev/null +++ b/checkpoint-8000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7ba7d3f1fc0fd2edfca51993915be1e4253e9f04baf556bea35848731cf01cfa +size 17683 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..9caec12755e3ec0a3501b9176af10f010339f400 --- /dev/null +++ b/checkpoint-8000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e917ca02ff6c8454dda0e169193c484c3354054a2427d812c2fa8b8cca21c6de +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..9af7c9afb68925e57307c4563f6145127c630af2 --- /dev/null +++ b/checkpoint-8000/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7fd0e84fe936ba7e0acf0a8cc75d5dc928dccacb2ea545d13146d9a7b8c77585 +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..fafe10e1c78365890209c3a8c7c742788e660c0f --- /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:a2d5ee81cf90fe6061e73721a1abaffbe04aadfe7126c0026437f8094e4a34be +size 454861629 diff --git a/logs/train_unconditional/events.out.tfevents.1687101863.pop-os.1521524.0 b/logs/train_unconditional/events.out.tfevents.1687101863.pop-os.1521524.0 index 50de914e4aa757c2066b7f50dbd4319a2916f365..87a0b644ea0c646257bf6b0a5caecbbf54565622 100644 --- a/logs/train_unconditional/events.out.tfevents.1687101863.pop-os.1521524.0 +++ b/logs/train_unconditional/events.out.tfevents.1687101863.pop-os.1521524.0 @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:305faff6aa1a7f69a87fd17d26e228392961094405d18215318d92f3216a06e7 -size 733484 +oid sha256:e7d0871b01e6d9ecd71142ab8911be63d88ae54c794862d1552b3a672bf20b17 +size 8140129 diff --git a/unet/diffusion_pytorch_model.bin b/unet/diffusion_pytorch_model.bin index b7505ceb4b09ce6a3e6c039365b93316281d9041..0b1eaeff43df8f6f3c47f3c4ebb2bd820d0c61ab 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:a9aec5d48cfe7cad383fd541b59fcb523856ad3bbb3fac2ed11b3c481cf49e83 +oid sha256:de3bbd1860a92c9753a03457f67bd3a39a2dfd944f0ad1b03b258e9bb2a4479a size 454867965