diff --git a/checkpoint-10500/optimizer.bin b/checkpoint-10500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..e83929d88e6827989d5adfd4b9f777fc5256f88d --- /dev/null +++ b/checkpoint-10500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:d069752d553e844a0d9fd312651dde9d39701257def98551fb6e765b81b5183e +size 73251413 diff --git a/checkpoint-10500/random_states_0.pkl b/checkpoint-10500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..d6906ed72f69e569d3b6150c98c66f69d5f38271 --- /dev/null +++ b/checkpoint-10500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c176434a90a69e6daa24223250f4dc3b4425ada8c2bd31dccf851497b8fc57a8 +size 14663 diff --git a/checkpoint-10500/scaler.pt b/checkpoint-10500/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..e7d37ab112d8b108640f4adea9efc18ff4f85228 --- /dev/null +++ b/checkpoint-10500/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:930aa4121a2a38f44b8af20e66d414b77fba3198866a0d743e0016592e1c420e +size 557 diff --git a/checkpoint-10500/scheduler.bin b/checkpoint-10500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..1430881e89db4bcdbcb76179c236f03dd06d39c4 --- /dev/null +++ b/checkpoint-10500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:5f880b6fdac81d333a28d35151b681f977b7d9683f1c80e80ed34339ca0814b6 +size 563 diff --git a/checkpoint-10500/unet/config.json b/checkpoint-10500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..bb1cd664fc36d4dc821a4f0fdae4b94d7d99b356 --- /dev/null +++ b/checkpoint-10500/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-10500/unet/diffusion_pytorch_model.bin b/checkpoint-10500/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..28b796593db8258c074d3d1c7f41b1211a7a1a81 --- /dev/null +++ b/checkpoint-10500/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:4833e2f77288142a970e10ca8e1fe8cce5318deb8dd44795ffe4777b09170f69 +size 36621969 diff --git a/checkpoint-10500/unet_ema/config.json b/checkpoint-10500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..f5299a68804ca514c2a12336a76acb393d2811c4 --- /dev/null +++ b/checkpoint-10500/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": 10500, + "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-10500/unet_ema/diffusion_pytorch_model.bin b/checkpoint-10500/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..c1f97a8561f50ece21e68859bdb8a36f87cb09b0 --- /dev/null +++ b/checkpoint-10500/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:210322ed229f8c67e54c3e2eb1b9170b6c7a706fd5415b1af2c0ef78247939b8 +size 36616081 diff --git a/checkpoint-11000/optimizer.bin b/checkpoint-11000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..e84d3e65de7d9a5f354d3889db285b07b672248d --- /dev/null +++ b/checkpoint-11000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:8edd4b512d28005e4bd4855a66b14c4f417c5998ac8294d2e9374cb405c624d2 +size 73251413 diff --git a/checkpoint-11000/random_states_0.pkl b/checkpoint-11000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..31644cd6307a82b0c6053557ca33f2760b0e7ddf --- /dev/null +++ b/checkpoint-11000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b74f4ceebdeb98333c28c6a41b66aa4bb14011702fc995cc1b5fa9106a2be483 +size 14663 diff --git a/checkpoint-11000/scaler.pt b/checkpoint-11000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..e3211d730f33fb63ccce37b3f6b355dd8bdf64f3 --- /dev/null +++ b/checkpoint-11000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:8df4dd185a129710516223a132f1321dd530e937b37d9a1dca8d2915b9d5a04a +size 557 diff --git a/checkpoint-11000/scheduler.bin b/checkpoint-11000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..42fd72599de13c540a3a41bf098eba5f30f76209 --- /dev/null +++ b/checkpoint-11000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:639cd51071e686afddf9c099ad4c78eed0fb800504d15c7cf337ce6a0d327ee2 +size 563 diff --git a/checkpoint-11000/unet/config.json b/checkpoint-11000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..bb1cd664fc36d4dc821a4f0fdae4b94d7d99b356 --- /dev/null +++ b/checkpoint-11000/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-11000/unet/diffusion_pytorch_model.bin b/checkpoint-11000/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..56c91b566f4265517b74280a1faacd6f019136b6 --- /dev/null +++ b/checkpoint-11000/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:a041163ff52058de2cca2226122ee64b4b0dfc0a7b562454d74819a949c566f2 +size 36621969 diff --git a/checkpoint-11000/unet_ema/config.json b/checkpoint-11000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..49c9168644e13f11423d568ba60eea5567152023 --- /dev/null +++ b/checkpoint-11000/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": 11000, + "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-11000/unet_ema/diffusion_pytorch_model.bin b/checkpoint-11000/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..61bf2f2fd958407604dfbfc9bd07fc521d489600 --- /dev/null +++ b/checkpoint-11000/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:5892a70cf5403921ea53d00744e98e9d739723b5def3298f35388f7c81788208 +size 36616081 diff --git a/checkpoint-11500/optimizer.bin b/checkpoint-11500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..b1b8427f79035f82258678306caf6da194d855de --- /dev/null +++ b/checkpoint-11500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:08dc26f52e9b86fbd60c442a490513041876a6dab18082860b1fbfdc84d60e23 +size 73251413 diff --git a/checkpoint-11500/random_states_0.pkl b/checkpoint-11500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..fed724b1a3d601ecd63617eca41e19b2a3f34fa5 --- /dev/null +++ b/checkpoint-11500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:4d54e81c10e51975e4f7b1d33fe48a7337bf29a4811a72aa978bad1e5fcfe562 +size 14663 diff --git a/checkpoint-11500/scaler.pt b/checkpoint-11500/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..0842e75026d521957a142549ee93aea39c4e6285 --- /dev/null +++ b/checkpoint-11500/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:8cab7988ce4c8b69e7970aada0017afc1c2120661c865bce9234915457ce96c2 +size 557 diff --git a/checkpoint-11500/scheduler.bin b/checkpoint-11500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..53af455eafda0931bf161fe6939947eca758cbe0 --- /dev/null +++ b/checkpoint-11500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:aadecd4a538b5d1dba9865109d974365b5fef3a16a8f15b84ad9bb69396810ab +size 563 diff --git a/checkpoint-11500/unet/config.json b/checkpoint-11500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..bb1cd664fc36d4dc821a4f0fdae4b94d7d99b356 --- /dev/null +++ b/checkpoint-11500/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-11500/unet/diffusion_pytorch_model.bin b/checkpoint-11500/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..ab46d833cf57b4a659b4e643538baa18033033ec --- /dev/null +++ b/checkpoint-11500/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:585777626add9c1eb930daee056a68a0c65fe82f4ca61a152202fc7e13dc6ff9 +size 36621969 diff --git a/checkpoint-11500/unet_ema/config.json b/checkpoint-11500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..8c06fc67ee4dcef3fab92b5acd1ff9c2bbc1446e --- /dev/null +++ b/checkpoint-11500/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": 11500, + "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-11500/unet_ema/diffusion_pytorch_model.bin b/checkpoint-11500/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..849219d0fc7602bf1f7406204c6b89a00a9e69c9 --- /dev/null +++ b/checkpoint-11500/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:2c900e3c277e02cd5dcfb1f83182fe3d6a1fedff838f1d8953b8d7250a60c3d2 +size 36616081 diff --git a/checkpoint-12000/optimizer.bin b/checkpoint-12000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..5ffaaa62a19bf637dd81108ac96656525884e24e --- /dev/null +++ b/checkpoint-12000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:ce749e708deb4834e510ee65f85ed9e541cd79227a3c9eaa2561080906f46db0 +size 73251413 diff --git a/checkpoint-12000/random_states_0.pkl b/checkpoint-12000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..2ae1f626a76dd391fe00c32c875a5de180de2d4a --- /dev/null +++ b/checkpoint-12000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:0e6be413308fdb0597b191695edf341acc944140c7f581268dae0d5cb5c64710 +size 14663 diff --git a/checkpoint-12000/scaler.pt b/checkpoint-12000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..53a6f028401ffa692d373d6dac2c2141effb5b35 --- /dev/null +++ b/checkpoint-12000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:bce458d2f0e0937df767175ef2da719ebfea2dc13abb18accc5a26fd31d6b24c +size 557 diff --git a/checkpoint-12000/scheduler.bin b/checkpoint-12000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..42f2de0a2835ddea2d46262fae43efed11adcaa0 --- /dev/null +++ b/checkpoint-12000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:1b0c5f7195a61f91e5bb9678cbb2c9866f6c8a95c81d28ec8cbff36e6b782060 +size 563 diff --git a/checkpoint-12000/unet/config.json b/checkpoint-12000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..bb1cd664fc36d4dc821a4f0fdae4b94d7d99b356 --- /dev/null +++ b/checkpoint-12000/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-12000/unet/diffusion_pytorch_model.bin b/checkpoint-12000/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..ff5f0e1a991c15b2aa31731e2ab755968864c9f0 --- /dev/null +++ b/checkpoint-12000/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:281c9e178c6cb08790db694c96498cfc9d19f20fbe90591b0e3477c3ba13c3fb +size 36621969 diff --git a/checkpoint-12000/unet_ema/config.json b/checkpoint-12000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..ccf144416ffcd8450ccc913cd47684c29d4868b2 --- /dev/null +++ b/checkpoint-12000/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": 12000, + "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-12000/unet_ema/diffusion_pytorch_model.bin b/checkpoint-12000/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..d44561838b8449b30dbd154bec88fa52128dcf66 --- /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:b2d8f56d62d75b15dae5234e3d958eccc3079bf86199f8e423064c991a7bed48 +size 36616081 diff --git a/checkpoint-12500/optimizer.bin b/checkpoint-12500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..70f4b31b00ec92ba96ea798a352bcd43e5e4508c --- /dev/null +++ b/checkpoint-12500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:f0ce0b1beca70fd5bd2240008fedeaa2102d9f4a4a24f777f966147644129e6d +size 73251413 diff --git a/checkpoint-12500/random_states_0.pkl b/checkpoint-12500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..b3a1ae8eb03dfc6f0367bbb8b5877dfb6e0ec3dc --- /dev/null +++ b/checkpoint-12500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:632ade90bd4a222b5beffbf4ec1c47bbc20e3f8c313319748e513f13bbabd23f +size 14663 diff --git a/checkpoint-12500/scaler.pt b/checkpoint-12500/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..acdb4c4fe49238ce70cc15e16af7a7a2fd4c36b0 --- /dev/null +++ b/checkpoint-12500/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:676c7ca172a8e0e2d2a49c50360fb5615bee6c75b2b21b58c40691d9ce37f540 +size 557 diff --git a/checkpoint-12500/scheduler.bin b/checkpoint-12500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..bb40921c4ebc3064e872fe1dfea1caa7a96ef5dc --- /dev/null +++ b/checkpoint-12500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:ca652ab09470c7d8ef9b2b18cd87478bea008d2b73edad3cae70fb852092376a +size 563 diff --git a/checkpoint-12500/unet/config.json b/checkpoint-12500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..bb1cd664fc36d4dc821a4f0fdae4b94d7d99b356 --- /dev/null +++ b/checkpoint-12500/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-12500/unet/diffusion_pytorch_model.bin b/checkpoint-12500/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..926664ff24c70c35fad9d217b83df49f3eb7a474 --- /dev/null +++ b/checkpoint-12500/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c9ce1765878ae36cad5674beb9fe0767842b3c333d4195819e5e3f6df76bdfbc +size 36621969 diff --git a/checkpoint-12500/unet_ema/config.json b/checkpoint-12500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..12c2cad1fde9f4620352b0553a08cec944c2f504 --- /dev/null +++ b/checkpoint-12500/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": 12500, + "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-12500/unet_ema/diffusion_pytorch_model.bin b/checkpoint-12500/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..69ad98e209fa4c9d901e890fca99029cd586ac0f --- /dev/null +++ b/checkpoint-12500/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:a400841cc3476abd9fece5d3e0bf34bd4abc08a4b8865daf3c8d4d88913840c9 +size 36616081 diff --git a/checkpoint-13000/optimizer.bin b/checkpoint-13000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..dd3d79d37203fe297045abea6bfd58e45a1a2bf4 --- /dev/null +++ b/checkpoint-13000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:0f6057e095320ebbfd1e7bfec9a15022e766c67fd01002ca99c86b14b9d31c13 +size 73251413 diff --git a/checkpoint-13000/random_states_0.pkl b/checkpoint-13000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..66aaf153d049437f24f7b6dcd16eb8b310ed983c --- /dev/null +++ b/checkpoint-13000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:74b8086748391706549155a071ad9215fee237acc56ef94e1acbe9cf7183b187 +size 14663 diff --git a/checkpoint-13000/scaler.pt b/checkpoint-13000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..736cd93987f5c507e48f52b9adc44af9129f2516 --- /dev/null +++ b/checkpoint-13000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b3bd46694cab01abb585c817c8192debd3ba2bfb2caed79a96d5c18143c7bb9a +size 557 diff --git a/checkpoint-13000/scheduler.bin b/checkpoint-13000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..c6aec2ce7026401776faa765dbaf798387a93b24 --- /dev/null +++ b/checkpoint-13000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:30d3c01f1a05ef1ae8354a8a28a3b01698b6f62fc98b5d5285d87757eccf76a4 +size 563 diff --git a/checkpoint-13000/unet/config.json b/checkpoint-13000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..bb1cd664fc36d4dc821a4f0fdae4b94d7d99b356 --- /dev/null +++ b/checkpoint-13000/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-13000/unet/diffusion_pytorch_model.bin b/checkpoint-13000/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..d0e3cde03b0a64ded0cd4235071f14475dac5ada --- /dev/null +++ b/checkpoint-13000/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:fd72031f2c1afdc33010eba7cfd0191a94cc8a09a784df5a685934c4ceedf328 +size 36621969 diff --git a/checkpoint-13000/unet_ema/config.json b/checkpoint-13000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..73b7d9ee900f648066fc7c2ccc76bf8b5f7d35a8 --- /dev/null +++ b/checkpoint-13000/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": 13000, + "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-13000/unet_ema/diffusion_pytorch_model.bin b/checkpoint-13000/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..5e7b4af315ff5ab3012d548a915e74c4a9b21a0f --- /dev/null +++ b/checkpoint-13000/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:13fd9bc6975a8592e2b95a9c4ad317b4d2043daa4b771e2d1dc705e27cc036a7 +size 36616081 diff --git a/checkpoint-13500/optimizer.bin b/checkpoint-13500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..0740f6526a3ea3d33b990648cb48dcafcd28bd6a --- /dev/null +++ b/checkpoint-13500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:ba170594436c302cd6e37c329786d79601a77c7d9f4a810194b9a356a5c81b35 +size 73251413 diff --git a/checkpoint-13500/random_states_0.pkl b/checkpoint-13500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..018acff15849e7d6d55660c7485fca95eaab9a7b --- /dev/null +++ b/checkpoint-13500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c7ce88771f3eb88369f5d3eb8100a6bdcd594ab8bcb9cd1720d146c43abe10a3 +size 14663 diff --git a/checkpoint-13500/scaler.pt b/checkpoint-13500/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..25fb04f4f0b3bc87a1efae9d449fb2bc933c65d2 --- /dev/null +++ b/checkpoint-13500/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:a17ba347132a6f99f1b8b847e2cd009bbea6fa767d1274351540ad5e2b5112ef +size 557 diff --git a/checkpoint-13500/scheduler.bin b/checkpoint-13500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..8632801c4d2b53665489971782aa47c78aa7b094 --- /dev/null +++ b/checkpoint-13500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:ca5cb59205a342441a41b2635478c07108f993e20fdafd099f7fc4a60a887e18 +size 563 diff --git a/checkpoint-13500/unet/config.json b/checkpoint-13500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..bb1cd664fc36d4dc821a4f0fdae4b94d7d99b356 --- /dev/null +++ b/checkpoint-13500/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-13500/unet/diffusion_pytorch_model.bin b/checkpoint-13500/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..0f52ac559c3feddab0d79e4d85c5429bdde9305d --- /dev/null +++ b/checkpoint-13500/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:ad6f8a494acc2d52e4fad08180924d37f4b912a78c9fcb72cd9199df6f9f8dac +size 36621969 diff --git a/checkpoint-13500/unet_ema/config.json b/checkpoint-13500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..fabe92943e54678fb61398f26a8cddcc294049a6 --- /dev/null +++ b/checkpoint-13500/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": 13500, + "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-13500/unet_ema/diffusion_pytorch_model.bin b/checkpoint-13500/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..92dea03c0bb5c79b3ff4c02927590656a594de75 --- /dev/null +++ b/checkpoint-13500/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:52ec31606fce80b3295370e871a5ab80be44f3a1e406bfa888119309082e4c1f +size 36616081 diff --git a/unet/diffusion_pytorch_model.bin b/unet/diffusion_pytorch_model.bin index b7aba9e3cf03be20b09f4f93dde59eda410cc8de..950593f107712220766774b157d1022cd15637b6 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:8e2d767168d5f6329cf93cc6180f652d5b8206fa7e69df784eac158df81c5301 +oid sha256:73001bfcf187ed88c8e290fa6820311f0121468962db2e0daf85560215aaf619 size 36621969