diff --git a/checkpoint-103500/optimizer.bin b/checkpoint-103500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..179c75a5363acb964865744e58487982baa0f7fc --- /dev/null +++ b/checkpoint-103500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:0851ac81704583b883116696c085731aa9ce64e6f72a4ac47c9d1448415cd91a +size 73251413 diff --git a/checkpoint-103500/random_states_0.pkl b/checkpoint-103500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..a6476a1094fc172701e5394b321642427b35f615 --- /dev/null +++ b/checkpoint-103500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b694df11c87c7ec982dddc398660e594d9a39057ddeb8b8913a4aa203bfad408 +size 14663 diff --git a/checkpoint-103500/scaler.pt b/checkpoint-103500/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..1650f3aa99f4a9bd865fd18468ed18517f48b8bf --- /dev/null +++ b/checkpoint-103500/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:4b415639e39ec97e7d2a2b3aa82012619b036bcb9a48106bdf820005e7a5f5b6 +size 557 diff --git a/checkpoint-103500/scheduler.bin b/checkpoint-103500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..46cfeb48741a8877b5f71b60020899522c069971 --- /dev/null +++ b/checkpoint-103500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:94b55ca501ed904a0cd6ff5cf392253827369f685bc0604f588b862fb871ed8e +size 563 diff --git a/checkpoint-103500/unet/config.json b/checkpoint-103500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..bb1cd664fc36d4dc821a4f0fdae4b94d7d99b356 --- /dev/null +++ b/checkpoint-103500/unet/config.json @@ -0,0 +1,60 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "attention_head_dim": 8, + "block_out_channels": [ + 64, + 64, + 128, + 128 + ], + "center_input_sample": false, + "class_embed_type": null, + "class_embeddings_concat": false, + "conv_in_kernel": 3, + "conv_out_kernel": 3, + "cross_attention_dim": 256, + "cross_attention_norm": null, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "CrossAttnDownBlock2D" + ], + "downsample_padding": 1, + "dual_cross_attention": false, + "encoder_hid_dim": null, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "layers_per_block": 2, + "mid_block_only_cross_attention": null, + "mid_block_scale_factor": 1, + "mid_block_type": "UNetMidBlock2DCrossAttn", + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "only_cross_attention": false, + "out_channels": 3, + "projection_class_embeddings_input_dim": null, + "resnet_out_scale_factor": 1.0, + "resnet_skip_time_act": false, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_cond_proj_dim": null, + "time_embedding_act_fn": null, + "time_embedding_dim": null, + "time_embedding_type": "positional", + "timestep_post_act": null, + "up_block_types": [ + "CrossAttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "use_linear_projection": false +} diff --git a/checkpoint-103500/unet/diffusion_pytorch_model.bin b/checkpoint-103500/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..93745b0d9e3641077f2d8dbdafa36c50862399ca --- /dev/null +++ b/checkpoint-103500/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:30c0afa5c852dcbc6c7df789af4f98853b6dfb98f83387ddb7105aee3b53f5de +size 36621969 diff --git a/checkpoint-103500/unet_ema/config.json b/checkpoint-103500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..52568c9bdb2cf05d3bcb2933af317410a77fa663 --- /dev/null +++ b/checkpoint-103500/unet_ema/config.json @@ -0,0 +1,67 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "attention_head_dim": 8, + "block_out_channels": [ + 64, + 64, + 128, + 128 + ], + "center_input_sample": false, + "class_embed_type": null, + "class_embeddings_concat": false, + "conv_in_kernel": 3, + "conv_out_kernel": 3, + "cross_attention_dim": 256, + "cross_attention_norm": null, + "decay": 0.9999, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "CrossAttnDownBlock2D" + ], + "downsample_padding": 1, + "dual_cross_attention": false, + "encoder_hid_dim": null, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "inv_gamma": 1.0, + "layers_per_block": 2, + "mid_block_only_cross_attention": null, + "mid_block_scale_factor": 1, + "mid_block_type": "UNetMidBlock2DCrossAttn", + "min_decay": 0.0, + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "only_cross_attention": false, + "optimization_step": 103500, + "out_channels": 3, + "power": 0.75, + "projection_class_embeddings_input_dim": null, + "resnet_out_scale_factor": 1.0, + "resnet_skip_time_act": false, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_cond_proj_dim": null, + "time_embedding_act_fn": null, + "time_embedding_dim": null, + "time_embedding_type": "positional", + "timestep_post_act": null, + "up_block_types": [ + "CrossAttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "update_after_step": 0, + "use_ema_warmup": true, + "use_linear_projection": false +} diff --git a/checkpoint-103500/unet_ema/diffusion_pytorch_model.bin b/checkpoint-103500/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..925e59dc7b06c025c7a526aa4df737c31fff0468 --- /dev/null +++ b/checkpoint-103500/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:21b12e71eb685abce025920e09741e07d19434db4b9bb221fc8ed954b5b867e5 +size 36616081 diff --git a/checkpoint-104000/optimizer.bin b/checkpoint-104000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..f00a44fb5f3e271c68b3f48acdcdb788503572c1 --- /dev/null +++ b/checkpoint-104000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:151ed8856cc508ab8d710f180c303365a6905766726b576d9b1769b5cf07a1e6 +size 73251413 diff --git a/checkpoint-104000/random_states_0.pkl b/checkpoint-104000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..43933442dbfca9ac4950b98bf7f5b7610519db9a --- /dev/null +++ b/checkpoint-104000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:f5a4f6cfa19689afb597cedd473a7f843137ce376b4d27fca54d8db7dadf2171 +size 14663 diff --git a/checkpoint-104000/scaler.pt b/checkpoint-104000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..b3ff5d6c5330359fbdfa423ca21452b75e9c262f --- /dev/null +++ b/checkpoint-104000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:551a6e4985886e51a7b44d9954c6270b62f7989d1db7889a52d695e7ba1e90a7 +size 557 diff --git a/checkpoint-104000/scheduler.bin b/checkpoint-104000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..d5e48ebac0a0dc6e42f4dfff7e1680c3242ad848 --- /dev/null +++ b/checkpoint-104000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:1850b45a8280cb5f3abbf0cf0a8962201260a4d111fb84c7b114cdff6ac5db59 +size 563 diff --git a/checkpoint-104000/unet/config.json b/checkpoint-104000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..bb1cd664fc36d4dc821a4f0fdae4b94d7d99b356 --- /dev/null +++ b/checkpoint-104000/unet/config.json @@ -0,0 +1,60 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "attention_head_dim": 8, + "block_out_channels": [ + 64, + 64, + 128, + 128 + ], + "center_input_sample": false, + "class_embed_type": null, + "class_embeddings_concat": false, + "conv_in_kernel": 3, + "conv_out_kernel": 3, + "cross_attention_dim": 256, + "cross_attention_norm": null, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "CrossAttnDownBlock2D" + ], + "downsample_padding": 1, + "dual_cross_attention": false, + "encoder_hid_dim": null, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "layers_per_block": 2, + "mid_block_only_cross_attention": null, + "mid_block_scale_factor": 1, + "mid_block_type": "UNetMidBlock2DCrossAttn", + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "only_cross_attention": false, + "out_channels": 3, + "projection_class_embeddings_input_dim": null, + "resnet_out_scale_factor": 1.0, + "resnet_skip_time_act": false, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_cond_proj_dim": null, + "time_embedding_act_fn": null, + "time_embedding_dim": null, + "time_embedding_type": "positional", + "timestep_post_act": null, + "up_block_types": [ + "CrossAttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "use_linear_projection": false +} diff --git a/checkpoint-104000/unet/diffusion_pytorch_model.bin b/checkpoint-104000/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..80c5ca20898dd890e6b1958da675defad0d186b0 --- /dev/null +++ b/checkpoint-104000/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:cc35ec70dc8fc55ee64b440faaa4005a86fb78bc674bacf7ea800d664fd8f09b +size 36621969 diff --git a/checkpoint-104000/unet_ema/config.json b/checkpoint-104000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..4212f699c90e4c10fa61744a12463fbd6d63a22c --- /dev/null +++ b/checkpoint-104000/unet_ema/config.json @@ -0,0 +1,67 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "attention_head_dim": 8, + "block_out_channels": [ + 64, + 64, + 128, + 128 + ], + "center_input_sample": false, + "class_embed_type": null, + "class_embeddings_concat": false, + "conv_in_kernel": 3, + "conv_out_kernel": 3, + "cross_attention_dim": 256, + "cross_attention_norm": null, + "decay": 0.9999, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "CrossAttnDownBlock2D" + ], + "downsample_padding": 1, + "dual_cross_attention": false, + "encoder_hid_dim": null, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "inv_gamma": 1.0, + "layers_per_block": 2, + "mid_block_only_cross_attention": null, + "mid_block_scale_factor": 1, + "mid_block_type": "UNetMidBlock2DCrossAttn", + "min_decay": 0.0, + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "only_cross_attention": false, + "optimization_step": 104000, + "out_channels": 3, + "power": 0.75, + "projection_class_embeddings_input_dim": null, + "resnet_out_scale_factor": 1.0, + "resnet_skip_time_act": false, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_cond_proj_dim": null, + "time_embedding_act_fn": null, + "time_embedding_dim": null, + "time_embedding_type": "positional", + "timestep_post_act": null, + "up_block_types": [ + "CrossAttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "update_after_step": 0, + "use_ema_warmup": true, + "use_linear_projection": false +} diff --git a/checkpoint-104000/unet_ema/diffusion_pytorch_model.bin b/checkpoint-104000/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..16cdc230130d480f6d9c5d4a5b758b3e9febc9dd --- /dev/null +++ b/checkpoint-104000/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:73c294e1eb489a3388522434f5b64608d7439dd6ffef63e0b59de9db574b6bf7 +size 36616081 diff --git a/checkpoint-104500/optimizer.bin b/checkpoint-104500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..99befa27cf9081c41116482a4569dbc6b9e1d5a7 --- /dev/null +++ b/checkpoint-104500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:9dcb2b29968dc2e4195771427b046364c5450e3d30e42a1954cddd4baa6d4c6e +size 73251413 diff --git a/checkpoint-104500/random_states_0.pkl b/checkpoint-104500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..c1008821179f390dde564612ff60f01e672fe26b --- /dev/null +++ b/checkpoint-104500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:6485407e8df74a8755cd8d3bda410a23098f5174b928ee77d64c4347fe5dd13b +size 14663 diff --git a/checkpoint-104500/scaler.pt b/checkpoint-104500/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..f03b4d5f8c576798a3add72f194159e9df33fd65 --- /dev/null +++ b/checkpoint-104500/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:870d46056bf12c7aace1364140aa3392cc8365b45a89c94c475b64d6bfb72c2b +size 557 diff --git a/checkpoint-104500/scheduler.bin b/checkpoint-104500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..7033cfc65a4115ec4b69ae550d3a49bd056efa83 --- /dev/null +++ b/checkpoint-104500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:934a6f19b7ac0be21447c01f8d387ee11e5e7ed3b2fe790214cdc1929f4e8796 +size 563 diff --git a/checkpoint-104500/unet/config.json b/checkpoint-104500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..bb1cd664fc36d4dc821a4f0fdae4b94d7d99b356 --- /dev/null +++ b/checkpoint-104500/unet/config.json @@ -0,0 +1,60 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "attention_head_dim": 8, + "block_out_channels": [ + 64, + 64, + 128, + 128 + ], + "center_input_sample": false, + "class_embed_type": null, + "class_embeddings_concat": false, + "conv_in_kernel": 3, + "conv_out_kernel": 3, + "cross_attention_dim": 256, + "cross_attention_norm": null, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "CrossAttnDownBlock2D" + ], + "downsample_padding": 1, + "dual_cross_attention": false, + "encoder_hid_dim": null, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "layers_per_block": 2, + "mid_block_only_cross_attention": null, + "mid_block_scale_factor": 1, + "mid_block_type": "UNetMidBlock2DCrossAttn", + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "only_cross_attention": false, + "out_channels": 3, + "projection_class_embeddings_input_dim": null, + "resnet_out_scale_factor": 1.0, + "resnet_skip_time_act": false, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_cond_proj_dim": null, + "time_embedding_act_fn": null, + "time_embedding_dim": null, + "time_embedding_type": "positional", + "timestep_post_act": null, + "up_block_types": [ + "CrossAttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "use_linear_projection": false +} diff --git a/checkpoint-104500/unet/diffusion_pytorch_model.bin b/checkpoint-104500/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..8c86ea9864d41c4d589be2d2ba1e917396009ba1 --- /dev/null +++ b/checkpoint-104500/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7d4a2fb6bba7771244f60f430da8c9fa168853544c8f09150973c8d4c0bec99e +size 36621969 diff --git a/checkpoint-104500/unet_ema/config.json b/checkpoint-104500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..aa1f7abc81f08df004bf0ea62e543999bfe1b447 --- /dev/null +++ b/checkpoint-104500/unet_ema/config.json @@ -0,0 +1,67 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "attention_head_dim": 8, + "block_out_channels": [ + 64, + 64, + 128, + 128 + ], + "center_input_sample": false, + "class_embed_type": null, + "class_embeddings_concat": false, + "conv_in_kernel": 3, + "conv_out_kernel": 3, + "cross_attention_dim": 256, + "cross_attention_norm": null, + "decay": 0.9999, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "CrossAttnDownBlock2D" + ], + "downsample_padding": 1, + "dual_cross_attention": false, + "encoder_hid_dim": null, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "inv_gamma": 1.0, + "layers_per_block": 2, + "mid_block_only_cross_attention": null, + "mid_block_scale_factor": 1, + "mid_block_type": "UNetMidBlock2DCrossAttn", + "min_decay": 0.0, + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "only_cross_attention": false, + "optimization_step": 104500, + "out_channels": 3, + "power": 0.75, + "projection_class_embeddings_input_dim": null, + "resnet_out_scale_factor": 1.0, + "resnet_skip_time_act": false, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_cond_proj_dim": null, + "time_embedding_act_fn": null, + "time_embedding_dim": null, + "time_embedding_type": "positional", + "timestep_post_act": null, + "up_block_types": [ + "CrossAttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "update_after_step": 0, + "use_ema_warmup": true, + "use_linear_projection": false +} diff --git a/checkpoint-104500/unet_ema/diffusion_pytorch_model.bin b/checkpoint-104500/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..34b88e09c7c8d40d81130580716b26c835f44540 --- /dev/null +++ b/checkpoint-104500/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:538df79092af05913d2c5f286a45ba727765b6af0a72b54b14618202e57afc2d +size 36616081 diff --git a/checkpoint-105000/optimizer.bin b/checkpoint-105000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..3e75c60a778c523a5e69d4140787ce516a19b53f --- /dev/null +++ b/checkpoint-105000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:056d97f5c448e8ff5b3fc5ce5905dbea7b0c0f2a75e1ced8d254afea8db6079d +size 73251413 diff --git a/checkpoint-105000/random_states_0.pkl b/checkpoint-105000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..c725c0806f7b87204bc58f4581f545d400b248a2 --- /dev/null +++ b/checkpoint-105000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:2485825dce8d81e95d9a8cd0bd7cc9cb0258aad83bed63d03eefaeb746646fd6 +size 14663 diff --git a/checkpoint-105000/scaler.pt b/checkpoint-105000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..78cb0a9918c681d74775481a57093a4518b431e5 --- /dev/null +++ b/checkpoint-105000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:9f4d320cf851c3ff2fb1d46a634eb93afd5fca65dd4bf614b27d3347115b1ec4 +size 557 diff --git a/checkpoint-105000/scheduler.bin b/checkpoint-105000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..b55f293cad7c217dd652b6a06f1648d7f6164177 --- /dev/null +++ b/checkpoint-105000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:ba71c6cb251f04488f8b563427a00dc8c9a8ea2727be4f502d3e30977a78afeb +size 563 diff --git a/checkpoint-105000/unet/config.json b/checkpoint-105000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..bb1cd664fc36d4dc821a4f0fdae4b94d7d99b356 --- /dev/null +++ b/checkpoint-105000/unet/config.json @@ -0,0 +1,60 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "attention_head_dim": 8, + "block_out_channels": [ + 64, + 64, + 128, + 128 + ], + "center_input_sample": false, + "class_embed_type": null, + "class_embeddings_concat": false, + "conv_in_kernel": 3, + "conv_out_kernel": 3, + "cross_attention_dim": 256, + "cross_attention_norm": null, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "CrossAttnDownBlock2D" + ], + "downsample_padding": 1, + "dual_cross_attention": false, + "encoder_hid_dim": null, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "layers_per_block": 2, + "mid_block_only_cross_attention": null, + "mid_block_scale_factor": 1, + "mid_block_type": "UNetMidBlock2DCrossAttn", + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "only_cross_attention": false, + "out_channels": 3, + "projection_class_embeddings_input_dim": null, + "resnet_out_scale_factor": 1.0, + "resnet_skip_time_act": false, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_cond_proj_dim": null, + "time_embedding_act_fn": null, + "time_embedding_dim": null, + "time_embedding_type": "positional", + "timestep_post_act": null, + "up_block_types": [ + "CrossAttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "use_linear_projection": false +} diff --git a/checkpoint-105000/unet/diffusion_pytorch_model.bin b/checkpoint-105000/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..059861234cb408706d3182973386e4defcc67bec --- /dev/null +++ b/checkpoint-105000/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c4c4ae1abfb6481743f5801ca59cbfe8f099a4f675b316595d7c0a8d97050f8f +size 36621969 diff --git a/checkpoint-105000/unet_ema/config.json b/checkpoint-105000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..a97af47fe7c45cc4903c0275373a0fa872bc75f2 --- /dev/null +++ b/checkpoint-105000/unet_ema/config.json @@ -0,0 +1,67 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "attention_head_dim": 8, + "block_out_channels": [ + 64, + 64, + 128, + 128 + ], + "center_input_sample": false, + "class_embed_type": null, + "class_embeddings_concat": false, + "conv_in_kernel": 3, + "conv_out_kernel": 3, + "cross_attention_dim": 256, + "cross_attention_norm": null, + "decay": 0.9999, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "CrossAttnDownBlock2D" + ], + "downsample_padding": 1, + "dual_cross_attention": false, + "encoder_hid_dim": null, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "inv_gamma": 1.0, + "layers_per_block": 2, + "mid_block_only_cross_attention": null, + "mid_block_scale_factor": 1, + "mid_block_type": "UNetMidBlock2DCrossAttn", + "min_decay": 0.0, + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "only_cross_attention": false, + "optimization_step": 105000, + "out_channels": 3, + "power": 0.75, + "projection_class_embeddings_input_dim": null, + "resnet_out_scale_factor": 1.0, + "resnet_skip_time_act": false, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_cond_proj_dim": null, + "time_embedding_act_fn": null, + "time_embedding_dim": null, + "time_embedding_type": "positional", + "timestep_post_act": null, + "up_block_types": [ + "CrossAttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "update_after_step": 0, + "use_ema_warmup": true, + "use_linear_projection": false +} diff --git a/checkpoint-105000/unet_ema/diffusion_pytorch_model.bin b/checkpoint-105000/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..b0553c7665aec647e4869c20a82d4c2e275fd492 --- /dev/null +++ b/checkpoint-105000/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:88615a9d8ba7e9ce607c53cc73491ede20f0ec59d82ed7085f9fc2ab375475fc +size 36616081 diff --git a/checkpoint-105500/optimizer.bin b/checkpoint-105500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..3b3823521f1467b95dbb7b98a002c13af0e30dd7 --- /dev/null +++ b/checkpoint-105500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:d8e1df3c6a2775dca8f44d17c278f27a90d832d83cb9483a92b0a1833fca6684 +size 73251413 diff --git a/checkpoint-105500/random_states_0.pkl b/checkpoint-105500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..bd46634abb19ba4d62e52a9721515beff196b863 --- /dev/null +++ b/checkpoint-105500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:8dfebbec478aa38084d188f634a0dcf14dd26017a9d1f97e5934b32555f782f3 +size 14663 diff --git a/checkpoint-105500/scaler.pt b/checkpoint-105500/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..0a0d20c18b09640819e60df453ae9f59a069ac38 --- /dev/null +++ b/checkpoint-105500/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:6f184e03e79de051173f3d634acadf1745329b0bdfdfd2ad81e50118954ca37e +size 557 diff --git a/checkpoint-105500/scheduler.bin b/checkpoint-105500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..51961cc5d63f41efab012bc17ef7d32f6eb48fbf --- /dev/null +++ b/checkpoint-105500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:3e420eee92950db1e145d958e5be4733f0ead331535bb28408e3c8c41baa9b4c +size 563 diff --git a/checkpoint-105500/unet/config.json b/checkpoint-105500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..bb1cd664fc36d4dc821a4f0fdae4b94d7d99b356 --- /dev/null +++ b/checkpoint-105500/unet/config.json @@ -0,0 +1,60 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "attention_head_dim": 8, + "block_out_channels": [ + 64, + 64, + 128, + 128 + ], + "center_input_sample": false, + "class_embed_type": null, + "class_embeddings_concat": false, + "conv_in_kernel": 3, + "conv_out_kernel": 3, + "cross_attention_dim": 256, + "cross_attention_norm": null, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "CrossAttnDownBlock2D" + ], + "downsample_padding": 1, + "dual_cross_attention": false, + "encoder_hid_dim": null, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "layers_per_block": 2, + "mid_block_only_cross_attention": null, + "mid_block_scale_factor": 1, + "mid_block_type": "UNetMidBlock2DCrossAttn", + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "only_cross_attention": false, + "out_channels": 3, + "projection_class_embeddings_input_dim": null, + "resnet_out_scale_factor": 1.0, + "resnet_skip_time_act": false, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_cond_proj_dim": null, + "time_embedding_act_fn": null, + "time_embedding_dim": null, + "time_embedding_type": "positional", + "timestep_post_act": null, + "up_block_types": [ + "CrossAttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "use_linear_projection": false +} diff --git a/checkpoint-105500/unet/diffusion_pytorch_model.bin b/checkpoint-105500/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..d7605a2a9d5cbf89946aa42f4756e262c643d70f --- /dev/null +++ b/checkpoint-105500/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:f27a7b973ef1f84a7b54d89de1d511623bd8c1bdd370ce2941cbdfec95e7cc69 +size 36621969 diff --git a/checkpoint-105500/unet_ema/config.json b/checkpoint-105500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..4d79285f5594e666434b6b6805c48a0e19fd634f --- /dev/null +++ b/checkpoint-105500/unet_ema/config.json @@ -0,0 +1,67 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "attention_head_dim": 8, + "block_out_channels": [ + 64, + 64, + 128, + 128 + ], + "center_input_sample": false, + "class_embed_type": null, + "class_embeddings_concat": false, + "conv_in_kernel": 3, + "conv_out_kernel": 3, + "cross_attention_dim": 256, + "cross_attention_norm": null, + "decay": 0.9999, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "CrossAttnDownBlock2D" + ], + "downsample_padding": 1, + "dual_cross_attention": false, + "encoder_hid_dim": null, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "inv_gamma": 1.0, + "layers_per_block": 2, + "mid_block_only_cross_attention": null, + "mid_block_scale_factor": 1, + "mid_block_type": "UNetMidBlock2DCrossAttn", + "min_decay": 0.0, + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "only_cross_attention": false, + "optimization_step": 105500, + "out_channels": 3, + "power": 0.75, + "projection_class_embeddings_input_dim": null, + "resnet_out_scale_factor": 1.0, + "resnet_skip_time_act": false, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_cond_proj_dim": null, + "time_embedding_act_fn": null, + "time_embedding_dim": null, + "time_embedding_type": "positional", + "timestep_post_act": null, + "up_block_types": [ + "CrossAttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "update_after_step": 0, + "use_ema_warmup": true, + "use_linear_projection": false +} diff --git a/checkpoint-105500/unet_ema/diffusion_pytorch_model.bin b/checkpoint-105500/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..b1f589314f86a8dcfa407ff0a9ba8e229f722afb --- /dev/null +++ b/checkpoint-105500/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:da209ee6f8db646789060ddbf1f489beb4c999728d0e529fc74ea46bf0d3cd4c +size 36616081 diff --git a/checkpoint-106000/optimizer.bin b/checkpoint-106000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..339638f0c3de6f48f200ea34e00e16a988fec7ba --- /dev/null +++ b/checkpoint-106000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:d63faa6d478b60f5b2a4b847eae71167acfeaab833c233532835b979ed857370 +size 73251413 diff --git a/checkpoint-106000/random_states_0.pkl b/checkpoint-106000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..9c3612e581e64d8e5407426bc2bee917a9063ddd --- /dev/null +++ b/checkpoint-106000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:581909b006d9db2d28dc525b72791f31f0ec69e26ad4d82ee17ef3f6f68a3779 +size 14663 diff --git a/checkpoint-106000/scaler.pt b/checkpoint-106000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..a452ef94b511e49326e6e6dcfd106ee76c0cde98 --- /dev/null +++ b/checkpoint-106000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:02cc6d9ea4696b2e44416c8d82add17f97db3f714bd6c0708624fa37fed3f8a3 +size 557 diff --git a/checkpoint-106000/scheduler.bin b/checkpoint-106000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..8c918debbf043895c02e3b4ffdf9016fd0ca4dce --- /dev/null +++ b/checkpoint-106000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:4525c7a7bb58649a0413e2ec90a402b03184c56cb3cf7befc5b15d5070430178 +size 563 diff --git a/checkpoint-106000/unet/config.json b/checkpoint-106000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..bb1cd664fc36d4dc821a4f0fdae4b94d7d99b356 --- /dev/null +++ b/checkpoint-106000/unet/config.json @@ -0,0 +1,60 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "attention_head_dim": 8, + "block_out_channels": [ + 64, + 64, + 128, + 128 + ], + "center_input_sample": false, + "class_embed_type": null, + "class_embeddings_concat": false, + "conv_in_kernel": 3, + "conv_out_kernel": 3, + "cross_attention_dim": 256, + "cross_attention_norm": null, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "CrossAttnDownBlock2D" + ], + "downsample_padding": 1, + "dual_cross_attention": false, + "encoder_hid_dim": null, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "layers_per_block": 2, + "mid_block_only_cross_attention": null, + "mid_block_scale_factor": 1, + "mid_block_type": "UNetMidBlock2DCrossAttn", + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "only_cross_attention": false, + "out_channels": 3, + "projection_class_embeddings_input_dim": null, + "resnet_out_scale_factor": 1.0, + "resnet_skip_time_act": false, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_cond_proj_dim": null, + "time_embedding_act_fn": null, + "time_embedding_dim": null, + "time_embedding_type": "positional", + "timestep_post_act": null, + "up_block_types": [ + "CrossAttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "use_linear_projection": false +} diff --git a/checkpoint-106000/unet/diffusion_pytorch_model.bin b/checkpoint-106000/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..83637ffec865d22df2256b6c88511491b5479736 --- /dev/null +++ b/checkpoint-106000/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:757b2015684597849ff4708cd98b6d912616b2c50638f28488e1e765f9073115 +size 36621969 diff --git a/checkpoint-106000/unet_ema/config.json b/checkpoint-106000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..c076794f6e9205019534b77f7b810f81af8084a5 --- /dev/null +++ b/checkpoint-106000/unet_ema/config.json @@ -0,0 +1,67 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "attention_head_dim": 8, + "block_out_channels": [ + 64, + 64, + 128, + 128 + ], + "center_input_sample": false, + "class_embed_type": null, + "class_embeddings_concat": false, + "conv_in_kernel": 3, + "conv_out_kernel": 3, + "cross_attention_dim": 256, + "cross_attention_norm": null, + "decay": 0.9999, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "CrossAttnDownBlock2D" + ], + "downsample_padding": 1, + "dual_cross_attention": false, + "encoder_hid_dim": null, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "inv_gamma": 1.0, + "layers_per_block": 2, + "mid_block_only_cross_attention": null, + "mid_block_scale_factor": 1, + "mid_block_type": "UNetMidBlock2DCrossAttn", + "min_decay": 0.0, + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "only_cross_attention": false, + "optimization_step": 106000, + "out_channels": 3, + "power": 0.75, + "projection_class_embeddings_input_dim": null, + "resnet_out_scale_factor": 1.0, + "resnet_skip_time_act": false, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_cond_proj_dim": null, + "time_embedding_act_fn": null, + "time_embedding_dim": null, + "time_embedding_type": "positional", + "timestep_post_act": null, + "up_block_types": [ + "CrossAttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "update_after_step": 0, + "use_ema_warmup": true, + "use_linear_projection": false +} diff --git a/checkpoint-106000/unet_ema/diffusion_pytorch_model.bin b/checkpoint-106000/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..548115e6c02942226643bebd4e50d28bb4367ba1 --- /dev/null +++ b/checkpoint-106000/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:2fde268ffb08c6a9fc39afd249f4b96418e626c25c0a3aab0a3e1411d36d72c3 +size 36616081 diff --git a/checkpoint-106500/optimizer.bin b/checkpoint-106500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..00d3a91c5068250a045cd486f0a2cb6db8b4f5d9 --- /dev/null +++ b/checkpoint-106500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:fd29f4541443e109fd7bc8f52a6e9479b5215255e2f2a68340e198ed3c52db00 +size 73251413 diff --git a/checkpoint-106500/random_states_0.pkl b/checkpoint-106500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..bd6bcfdcfb3f7e7c701756698e3ab0f31ed724b7 --- /dev/null +++ b/checkpoint-106500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:de4f536deda3053cd531b0a28b29094aa787fc01fb11b312a2afe322768382b8 +size 14663 diff --git a/checkpoint-106500/scaler.pt b/checkpoint-106500/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..00b5a76f41a37c745823c7e4ef5af385e2cd35ee --- /dev/null +++ b/checkpoint-106500/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:1e815a4600150250089852bbcd22ad7b68d8d585d51296b87b50933faa835a54 +size 557 diff --git a/checkpoint-106500/scheduler.bin b/checkpoint-106500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..81a471939cc9041612e802cfdc8a62c6f22e7737 --- /dev/null +++ b/checkpoint-106500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:ea6f17023447c1f093bc0cabd2eed9c2e4b0679580936613dd97920a471f6c64 +size 563 diff --git a/checkpoint-106500/unet/config.json b/checkpoint-106500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..bb1cd664fc36d4dc821a4f0fdae4b94d7d99b356 --- /dev/null +++ b/checkpoint-106500/unet/config.json @@ -0,0 +1,60 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "attention_head_dim": 8, + "block_out_channels": [ + 64, + 64, + 128, + 128 + ], + "center_input_sample": false, + "class_embed_type": null, + "class_embeddings_concat": false, + "conv_in_kernel": 3, + "conv_out_kernel": 3, + "cross_attention_dim": 256, + "cross_attention_norm": null, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "CrossAttnDownBlock2D" + ], + "downsample_padding": 1, + "dual_cross_attention": false, + "encoder_hid_dim": null, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "layers_per_block": 2, + "mid_block_only_cross_attention": null, + "mid_block_scale_factor": 1, + "mid_block_type": "UNetMidBlock2DCrossAttn", + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "only_cross_attention": false, + "out_channels": 3, + "projection_class_embeddings_input_dim": null, + "resnet_out_scale_factor": 1.0, + "resnet_skip_time_act": false, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_cond_proj_dim": null, + "time_embedding_act_fn": null, + "time_embedding_dim": null, + "time_embedding_type": "positional", + "timestep_post_act": null, + "up_block_types": [ + "CrossAttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "use_linear_projection": false +} diff --git a/checkpoint-106500/unet/diffusion_pytorch_model.bin b/checkpoint-106500/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..c3525209b42d228e2924de24d40f2a50fdef7eb7 --- /dev/null +++ b/checkpoint-106500/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:287db3036fc5559e6369913c511a3daeed516c98d002a0582bea2fb4e2bcfcae +size 36621969 diff --git a/checkpoint-106500/unet_ema/config.json b/checkpoint-106500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..7d907443cff47da40daa61ccf3c62b193fc3866d --- /dev/null +++ b/checkpoint-106500/unet_ema/config.json @@ -0,0 +1,67 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "attention_head_dim": 8, + "block_out_channels": [ + 64, + 64, + 128, + 128 + ], + "center_input_sample": false, + "class_embed_type": null, + "class_embeddings_concat": false, + "conv_in_kernel": 3, + "conv_out_kernel": 3, + "cross_attention_dim": 256, + "cross_attention_norm": null, + "decay": 0.9999, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "CrossAttnDownBlock2D" + ], + "downsample_padding": 1, + "dual_cross_attention": false, + "encoder_hid_dim": null, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "inv_gamma": 1.0, + "layers_per_block": 2, + "mid_block_only_cross_attention": null, + "mid_block_scale_factor": 1, + "mid_block_type": "UNetMidBlock2DCrossAttn", + "min_decay": 0.0, + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "only_cross_attention": false, + "optimization_step": 106500, + "out_channels": 3, + "power": 0.75, + "projection_class_embeddings_input_dim": null, + "resnet_out_scale_factor": 1.0, + "resnet_skip_time_act": false, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_cond_proj_dim": null, + "time_embedding_act_fn": null, + "time_embedding_dim": null, + "time_embedding_type": "positional", + "timestep_post_act": null, + "up_block_types": [ + "CrossAttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "update_after_step": 0, + "use_ema_warmup": true, + "use_linear_projection": false +} diff --git a/checkpoint-106500/unet_ema/diffusion_pytorch_model.bin b/checkpoint-106500/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..906c04e07a247efed541ce198570ed767448c613 --- /dev/null +++ b/checkpoint-106500/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:d90f9c9e41adfc2a542368e57bcdd6095d164c8b03ca0d333ce306550efde411 +size 36616081 diff --git a/unet/diffusion_pytorch_model.bin b/unet/diffusion_pytorch_model.bin index 5e43bc17a35046d43a3603d689b8131853bf1fb1..5827754a723e67c2377e1cafb760653d0a4e21af 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:8a30ef2bdf16e85d0f22905473aff54bd0f5b795afed7f76d24e1fce8483cd9e +oid sha256:fc25f8f699f97e1725acf4457408a5c5ac685d72005809e11296eedf5a6bd22d size 36621969