diff --git a/checkpoint-107000/optimizer.bin b/checkpoint-107000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..c814e99d7ffa92d3bde856259ccc59b6b7ef44a4 --- /dev/null +++ b/checkpoint-107000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:21a96d180582bea7eaca794d42d88abc74d74870b21dc84b3d66a6c04f8733b6 +size 73251413 diff --git a/checkpoint-107000/random_states_0.pkl b/checkpoint-107000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..14eef72d99042983a1cfbaa32eb7471568713c85 --- /dev/null +++ b/checkpoint-107000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:bfd46df4279690dc9b17cb5c4c6ef5c25bdceb71d4c590c77bd138a1633de284 +size 14663 diff --git a/checkpoint-107000/scaler.pt b/checkpoint-107000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..c1ef49e17aeb22448ec7220d2d4f0299d3b8c333 --- /dev/null +++ b/checkpoint-107000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:303baed46be8e56d75b16ee5c38793643628a1afd47fc8fa8d0a0a22298b6404 +size 557 diff --git a/checkpoint-107000/scheduler.bin b/checkpoint-107000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..ee6e46ec93294d13301287940c000e61a4db3bdc --- /dev/null +++ b/checkpoint-107000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:ec283e5a2cd76a1f6ec30fc4a036ab93bdff8b580f33044723edf8606e734ae8 +size 563 diff --git a/checkpoint-107000/unet/config.json b/checkpoint-107000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..bb1cd664fc36d4dc821a4f0fdae4b94d7d99b356 --- /dev/null +++ b/checkpoint-107000/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-107000/unet/diffusion_pytorch_model.bin b/checkpoint-107000/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..e30917330a55b6aae747c9edf896b4abff30e49d --- /dev/null +++ b/checkpoint-107000/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:a265bdbabe7d58c9515075a29aab102c439ac37d3066becc83eb2c2a96d442b2 +size 36621969 diff --git a/checkpoint-107000/unet_ema/config.json b/checkpoint-107000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..ad6e74584c0a2139b451850251aefe9166755408 --- /dev/null +++ b/checkpoint-107000/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": 107000, + "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-107000/unet_ema/diffusion_pytorch_model.bin b/checkpoint-107000/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..ea7f24cbada05e5e8fb9e29d6322c74d010e6de9 --- /dev/null +++ b/checkpoint-107000/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:a111ad1f5463fb6fbcac63d08dcb8b4a9737282ce3a78439e676cfc0c5278ef3 +size 36616081 diff --git a/checkpoint-107500/optimizer.bin b/checkpoint-107500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..d2bf86237301d1d190ff6f703784d2ddedc2f075 --- /dev/null +++ b/checkpoint-107500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e2cdd5fab95f079a735ccef14d0e8883ebea8845a32e9c744a614be8331f5e8e +size 73251413 diff --git a/checkpoint-107500/random_states_0.pkl b/checkpoint-107500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..10f058c079ad3a92882b09e2cafa11e8fe664fbb --- /dev/null +++ b/checkpoint-107500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e29bdce8423f400c646e2b4c688bd306b924ad7b0b67ad77bba4ee7902956ab8 +size 14663 diff --git a/checkpoint-107500/scaler.pt b/checkpoint-107500/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..9e03d208cd49cef441618016960438d85d58564c --- /dev/null +++ b/checkpoint-107500/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:06059e8ba66d35b70f17ba8112120a6b9f78f19dd189c7c80a9a95a6f5a76956 +size 557 diff --git a/checkpoint-107500/scheduler.bin b/checkpoint-107500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..b93885d87353e037150c9599c161b10db718e33e --- /dev/null +++ b/checkpoint-107500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:881a1b8fbe4707bba7ae50a2eb18e76ebf798d9ff564983ea4d79398eda9afa2 +size 563 diff --git a/checkpoint-107500/unet/config.json b/checkpoint-107500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..bb1cd664fc36d4dc821a4f0fdae4b94d7d99b356 --- /dev/null +++ b/checkpoint-107500/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-107500/unet/diffusion_pytorch_model.bin b/checkpoint-107500/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..2aa746bfd7dbbd009e7c37771fcd06cd18b0783b --- /dev/null +++ b/checkpoint-107500/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:306882300b5a849d5e1c1d32089ad728c661e1f4e700245f10bc8011fc9b42f9 +size 36621969 diff --git a/checkpoint-107500/unet_ema/config.json b/checkpoint-107500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..9bcfb1f846680db989f7f4f9c44a6100632261d7 --- /dev/null +++ b/checkpoint-107500/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": 107500, + "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-107500/unet_ema/diffusion_pytorch_model.bin b/checkpoint-107500/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..1126518ad0cf635dd53ac9baf4a26a3635a3729e --- /dev/null +++ b/checkpoint-107500/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:28ff7ff5da66f17d872bec9dbc6c73a89b41ce71494b19b87c29c2be61877522 +size 36616081 diff --git a/checkpoint-108000/optimizer.bin b/checkpoint-108000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..5bcd786be299a78763a75ccc345e0d55cf050b8b --- /dev/null +++ b/checkpoint-108000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:1aced7d466465178a54b20927916ad744b0d79be123a777a25f2739818fe50cf +size 73251413 diff --git a/checkpoint-108000/random_states_0.pkl b/checkpoint-108000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..be2b47cf1e2ea462882df31878f1869b5ef459f9 --- /dev/null +++ b/checkpoint-108000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:8c826cdff8e0e4714017bb06a84feacdd0b0c5b25bf840f4ac49dd749c71575a +size 14663 diff --git a/checkpoint-108000/scaler.pt b/checkpoint-108000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..463f1901b9955bdaa37987dcd7ebf3823b193e78 --- /dev/null +++ b/checkpoint-108000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:08826b3877d2b1ddb4ab5fe8463d0a1fd7157d383aeefcb9eca286df911ee379 +size 557 diff --git a/checkpoint-108000/scheduler.bin b/checkpoint-108000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..eb425965ac5e31251f8d5cba131de739eaff91fe --- /dev/null +++ b/checkpoint-108000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:0d8ef7d95d8be5a12f53a3b0217958eb00b393055ebea3dd77ec16b5d81e3ac9 +size 563 diff --git a/checkpoint-108000/unet/config.json b/checkpoint-108000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..bb1cd664fc36d4dc821a4f0fdae4b94d7d99b356 --- /dev/null +++ b/checkpoint-108000/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-108000/unet/diffusion_pytorch_model.bin b/checkpoint-108000/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..800a2ed1b2f9f3c2bfee92cc3f8968b16ffd369d --- /dev/null +++ b/checkpoint-108000/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:933304891545beef0d388cf54a4ca0bc63b6f9f75eb5563a07e223e07d57ac03 +size 36621969 diff --git a/checkpoint-108000/unet_ema/config.json b/checkpoint-108000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..d46a76e3bd3f271ee14ac62e859364f6f492ffbd --- /dev/null +++ b/checkpoint-108000/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": 108000, + "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-108000/unet_ema/diffusion_pytorch_model.bin b/checkpoint-108000/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..df5d398d2a49bb4a1d78e26112c196dcadfd3c17 --- /dev/null +++ b/checkpoint-108000/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:724a509eff9fbe6fb7c024c7676b809e401a8f333052f090b99d6ca4cdaea6de +size 36616081 diff --git a/checkpoint-108500/optimizer.bin b/checkpoint-108500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..b0137a96489d3af7df91db3d1a654c3e5a59865c --- /dev/null +++ b/checkpoint-108500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:8381f05f8555060518718263a495a0623868d5d8fc09a63c89fa8e24f10de44d +size 73251413 diff --git a/checkpoint-108500/random_states_0.pkl b/checkpoint-108500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..b9ea25dc4f9be05ba633de9590c544c5f4addef3 --- /dev/null +++ b/checkpoint-108500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:426cef0bcc7c2eac81af777274e7e4f827b1e6f3a21c5dcaf897a0f432d1340a +size 14663 diff --git a/checkpoint-108500/scaler.pt b/checkpoint-108500/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..0ec9ef18d38351e91163d8c0772bc508410fead0 --- /dev/null +++ b/checkpoint-108500/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:38da948d114d1b576c9c69532a9db54a10f7bc04c4063e88a2b25bb0cd5579a0 +size 557 diff --git a/checkpoint-108500/scheduler.bin b/checkpoint-108500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..a54e592117e63be95c4e76fc25fa7924fa99ac6b --- /dev/null +++ b/checkpoint-108500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:a63b782052e195443c2aad333a11a44eb9f327d87ba6a6e15fa8b1e6ea8ace10 +size 563 diff --git a/checkpoint-108500/unet/config.json b/checkpoint-108500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..bb1cd664fc36d4dc821a4f0fdae4b94d7d99b356 --- /dev/null +++ b/checkpoint-108500/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-108500/unet/diffusion_pytorch_model.bin b/checkpoint-108500/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..eb603e5ebeb7687297e201c44bde316ca1c28bf2 --- /dev/null +++ b/checkpoint-108500/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:aad764a6091648e688a22dc5214b4c99b8b243aac16af594e8939481fc4d47c0 +size 36621969 diff --git a/checkpoint-108500/unet_ema/config.json b/checkpoint-108500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..7c0bb4c0ad25d6f07637d050e62de7788ca9b6a0 --- /dev/null +++ b/checkpoint-108500/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": 108500, + "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-108500/unet_ema/diffusion_pytorch_model.bin b/checkpoint-108500/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..25fd450d21877162c2bf8db448e354e0b6ebf6f8 --- /dev/null +++ b/checkpoint-108500/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:a725f0935c0cc8654b08d16ae6ba35cab7208bb6c8ba2ae304bd8589459acaf1 +size 36616081 diff --git a/checkpoint-109000/optimizer.bin b/checkpoint-109000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..579a3cb0af607b81782badde6dc38ccbf2bbdf2b --- /dev/null +++ b/checkpoint-109000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:cf61dc49ba13d0e0daf83d758f84677f4b7ffba1b3d6ba31b8166b70bc01ce8d +size 73251413 diff --git a/checkpoint-109000/random_states_0.pkl b/checkpoint-109000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..31315c74ba39e7783e4ab5ad02f60b8c863eeee7 --- /dev/null +++ b/checkpoint-109000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:9f2678b9863844cecbe0d0cdcdf07bbd473dbff4b3913eeecb262d3efba12f9e +size 14663 diff --git a/checkpoint-109000/scaler.pt b/checkpoint-109000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..50e3db85e3c0d096bbdb3660c3b9b0b386a82d3e --- /dev/null +++ b/checkpoint-109000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:eed37b42fd0b2349a3b4f4195cddbdcc61da15b1b05586d344864aeefc33d741 +size 557 diff --git a/checkpoint-109000/scheduler.bin b/checkpoint-109000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..67b1f5ea6194bf021308ec1c48f028bb2db4f2b6 --- /dev/null +++ b/checkpoint-109000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:f2c36e0f8eff49000074ec547d745f9c0a4ef1eed301109910a12cbf6d2ba1da +size 563 diff --git a/checkpoint-109000/unet/config.json b/checkpoint-109000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..bb1cd664fc36d4dc821a4f0fdae4b94d7d99b356 --- /dev/null +++ b/checkpoint-109000/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-109000/unet/diffusion_pytorch_model.bin b/checkpoint-109000/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..74cac74f2a7e824ca8b71bc2f758a6c4f89f1da8 --- /dev/null +++ b/checkpoint-109000/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:f82e15efc94203c6a3f3c1b08de347ab5a814ae0aeec26262073295ec3f16c77 +size 36621969 diff --git a/checkpoint-109000/unet_ema/config.json b/checkpoint-109000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..69023e73550c6981ab221b6d511d696a5c5c6354 --- /dev/null +++ b/checkpoint-109000/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": 109000, + "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-109000/unet_ema/diffusion_pytorch_model.bin b/checkpoint-109000/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..4af0eb2d6fa59147bfb461e50c85d293a32ffbb4 --- /dev/null +++ b/checkpoint-109000/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:bdf2c46fabe789c2b14823d66f6b19645a15dbb4977b60fbdfaf4e8ad9d62be6 +size 36616081 diff --git a/checkpoint-109500/optimizer.bin b/checkpoint-109500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..6bd703e7fdc9924cd5da0b56594b0df68f0565e7 --- /dev/null +++ b/checkpoint-109500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:2f27ab8dd6a86dc9ebe025f91e7155df6f42f3bf09dfebb99dc05c28262a0245 +size 73251413 diff --git a/checkpoint-109500/random_states_0.pkl b/checkpoint-109500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..e1a0ebdb229f0f097fb7dc5e6e33e45ac32927fa --- /dev/null +++ b/checkpoint-109500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7a8ba001aa231d59cb35a3106757f50f1e1be20a67dc50956f51c97b854a8276 +size 14663 diff --git a/checkpoint-109500/scaler.pt b/checkpoint-109500/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..7aa03ea309f68872a8bcaa72eb1e677d72a3cf09 --- /dev/null +++ b/checkpoint-109500/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:d694aa7edc3efcc35f68438ae852151addcff199329176bf93fe486881982409 +size 557 diff --git a/checkpoint-109500/scheduler.bin b/checkpoint-109500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..9a203f070e46672db5fa1e083d95e62bda5181a7 --- /dev/null +++ b/checkpoint-109500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:9986d8eedad41d1c14a6240c3923bb6646cb60600e2bf8116ffd8c06dc10a5d9 +size 563 diff --git a/checkpoint-109500/unet/config.json b/checkpoint-109500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..bb1cd664fc36d4dc821a4f0fdae4b94d7d99b356 --- /dev/null +++ b/checkpoint-109500/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-109500/unet/diffusion_pytorch_model.bin b/checkpoint-109500/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..b10a6157111ca980d9839ee78837af22cf007cfb --- /dev/null +++ b/checkpoint-109500/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:040dfd7c37548bf8c918de39ca1dcf47af32c248a48d3bea0923eb8e3ae5f9a9 +size 36621969 diff --git a/checkpoint-109500/unet_ema/config.json b/checkpoint-109500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..4f2b15359819d1bb382458fb7137750a0586fe4b --- /dev/null +++ b/checkpoint-109500/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": 109500, + "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-109500/unet_ema/diffusion_pytorch_model.bin b/checkpoint-109500/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..d17e2621738ac908ae82648fb59d959cd7b01eaf --- /dev/null +++ b/checkpoint-109500/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:44bd4520cb66a804ed88ae20bc6db5b599fa2c0233378722c86579eb621afee0 +size 36616081 diff --git a/checkpoint-110000/optimizer.bin b/checkpoint-110000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..f1d16c38e104b50308f0ad28b75a8eae18869d67 --- /dev/null +++ b/checkpoint-110000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:0679d7269a0e353bfa62300df7a10d6395489e37afb7e683947ae127eb7a7a91 +size 73251413 diff --git a/checkpoint-110000/random_states_0.pkl b/checkpoint-110000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..13d5a3aad18ebaa2e232dcdb6dde5db779887699 --- /dev/null +++ b/checkpoint-110000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:d9bacffb207bbaa815f759c001492fcf8be11378a1264c08725afee10e951ee5 +size 14663 diff --git a/checkpoint-110000/scaler.pt b/checkpoint-110000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..555cdbb3a1f5dfd5e6c2a70a164bc30fad960f13 --- /dev/null +++ b/checkpoint-110000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:d3daffdf2d2f5111b9cfbf7caa2a1fe2490879945d4a8a84564b4ec58575353a +size 557 diff --git a/checkpoint-110000/scheduler.bin b/checkpoint-110000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..bc0eeb8cb3ff33aff3dfb322fcc516f7e2e1bfc9 --- /dev/null +++ b/checkpoint-110000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:401667dc2885a298991ee5a536b7f32e30dd40b294a7d3cb892e4ee3426025b6 +size 563 diff --git a/checkpoint-110000/unet/config.json b/checkpoint-110000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..bb1cd664fc36d4dc821a4f0fdae4b94d7d99b356 --- /dev/null +++ b/checkpoint-110000/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-110000/unet/diffusion_pytorch_model.bin b/checkpoint-110000/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..6471fc04fdc44199b50f7512e2cdd3032c7673f4 --- /dev/null +++ b/checkpoint-110000/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:786140b4ae406441af6344eb724cff2f87cd8d5eb8507d009ecd91eb887395b4 +size 36621969 diff --git a/checkpoint-110000/unet_ema/config.json b/checkpoint-110000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..628ab3d10edad69ef69db392ae0e926a24488620 --- /dev/null +++ b/checkpoint-110000/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": 110000, + "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-110000/unet_ema/diffusion_pytorch_model.bin b/checkpoint-110000/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..93e51639ba1b72a29e3641388d4843c57cd36f4f --- /dev/null +++ b/checkpoint-110000/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:48e703e6ef500849b03a2b1d890594d17a0eedc818abce5824afef7da2917655 +size 36616081 diff --git a/unet/diffusion_pytorch_model.bin b/unet/diffusion_pytorch_model.bin index 5827754a723e67c2377e1cafb760653d0a4e21af..bf5cb83b0fdfa22f710e51586d363e4edb728a60 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:fc25f8f699f97e1725acf4457408a5c5ac685d72005809e11296eedf5a6bd22d +oid sha256:4d9eb8e1f582b69f1f0e0d12ced0b21327190f8ff0c87b93b6280235618af99f size 36621969