diff --git a/checkpoint-123000/optimizer.bin b/checkpoint-123000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..6bca8d973e21837810bcf2e87969ea7427e8987a --- /dev/null +++ b/checkpoint-123000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:f9e4c0fd1ae1ca04ce7fb7abeef4000ee6476fadbaca28f23eb66417d04eb7f5 +size 73251413 diff --git a/checkpoint-123000/random_states_0.pkl b/checkpoint-123000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..01f3287ab67125e1aa9741067923e5464ccad2b0 --- /dev/null +++ b/checkpoint-123000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:58dfc5bfe41f0ad7d0db68d1a9420d81f439d4870899bab84f883b512a244017 +size 14727 diff --git a/checkpoint-123000/scaler.pt b/checkpoint-123000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..0aae7911887533d45c77503683812b18408f7f9d --- /dev/null +++ b/checkpoint-123000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c21d802995cb5f2495043786c34a8fffa300da0ba370ed6e03e94b66635ed099 +size 557 diff --git a/checkpoint-123000/scheduler.bin b/checkpoint-123000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..b2a18857a31ecf42a29de7e314a1a2cf23592843 --- /dev/null +++ b/checkpoint-123000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c41a004c0a5cf1a6c945be31e1220f1ea6bd8987368b27f8ee396c80aa0f1710 +size 563 diff --git a/checkpoint-123000/unet/config.json b/checkpoint-123000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..c2d7b8c0d490531013583af57a806d21893e494c --- /dev/null +++ b/checkpoint-123000/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", + "CrossAttnDownBlock2D", + "DownBlock2D" + ], + "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": [ + "UpBlock2D", + "CrossAttnUpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "use_linear_projection": false +} diff --git a/checkpoint-123000/unet/diffusion_pytorch_model.bin b/checkpoint-123000/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..b9e3e54414615136361b3188db07354e99c22057 --- /dev/null +++ b/checkpoint-123000/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:35196838ce9b981f2fbaa7188d119e03381cae356e95264e7d6c22f77edfa57d +size 36621969 diff --git a/checkpoint-123000/unet_ema/config.json b/checkpoint-123000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..f3178c9026f54f9aaa77b1749e2d7c8b1d8b946b --- /dev/null +++ b/checkpoint-123000/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", + "CrossAttnDownBlock2D", + "DownBlock2D" + ], + "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": 123000, + "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": [ + "UpBlock2D", + "CrossAttnUpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "update_after_step": 0, + "use_ema_warmup": true, + "use_linear_projection": false +} diff --git a/checkpoint-123000/unet_ema/diffusion_pytorch_model.bin b/checkpoint-123000/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..54c1d9d9e29e117fd57b0045e38355ab5756bad9 --- /dev/null +++ b/checkpoint-123000/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:86d36f1b471bdc10e6006a7c4e6419f9bd558854b10b43a4d51f226a9dd6262c +size 36616081 diff --git a/checkpoint-123500/optimizer.bin b/checkpoint-123500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..0d59e49ebdf433699c3907eb2f9f674b644e8696 --- /dev/null +++ b/checkpoint-123500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:d26a66429fcfdbd6b62061d954d6a87a186a26909f02fba86c7196e6327a5457 +size 73251413 diff --git a/checkpoint-123500/random_states_0.pkl b/checkpoint-123500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..218b255561d04fdc471fdcd8a73e8ba94c2cd694 --- /dev/null +++ b/checkpoint-123500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:093e002b8322250f4ddf6685db4b667093ae3fba2d2a1c498955c0780ab15de0 +size 14727 diff --git a/checkpoint-123500/scaler.pt b/checkpoint-123500/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..332b449fc45f1618ea28d3a9d47604398e6a556b --- /dev/null +++ b/checkpoint-123500/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e6110bcb1f040aaf093a80e1fba8b1020541cb90940e97529c29c753c0179c80 +size 557 diff --git a/checkpoint-123500/scheduler.bin b/checkpoint-123500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..980dea0bbbaae4af811219dace2653bfe927e68c --- /dev/null +++ b/checkpoint-123500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:97dabd11846f4d69da38500d3167683e528c216a25d81000addebad63084095c +size 563 diff --git a/checkpoint-123500/unet/config.json b/checkpoint-123500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..c2d7b8c0d490531013583af57a806d21893e494c --- /dev/null +++ b/checkpoint-123500/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", + "CrossAttnDownBlock2D", + "DownBlock2D" + ], + "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": [ + "UpBlock2D", + "CrossAttnUpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "use_linear_projection": false +} diff --git a/checkpoint-123500/unet/diffusion_pytorch_model.bin b/checkpoint-123500/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..1080e88b02ec8922f29eaf3d64740e07709fbdce --- /dev/null +++ b/checkpoint-123500/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:44cf83276aabfd86be4089fae005de1aaec97db785cf6da6e2c204bc7698a36e +size 36621969 diff --git a/checkpoint-123500/unet_ema/config.json b/checkpoint-123500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..645c8bb40dd8b840f038222c59dfcb6ba30725c3 --- /dev/null +++ b/checkpoint-123500/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", + "CrossAttnDownBlock2D", + "DownBlock2D" + ], + "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": 123500, + "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": [ + "UpBlock2D", + "CrossAttnUpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "update_after_step": 0, + "use_ema_warmup": true, + "use_linear_projection": false +} diff --git a/checkpoint-123500/unet_ema/diffusion_pytorch_model.bin b/checkpoint-123500/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..b938981e4dcb926bb4a6117e51ead444077e04ea --- /dev/null +++ b/checkpoint-123500/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:8c4f07d05a869651cdbff249d1b206e07d61ae816c75f9322a79e429c0a7ac55 +size 36616081 diff --git a/checkpoint-124000/optimizer.bin b/checkpoint-124000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..1225824b4e6e4254b97f8599545acaf3833c3a07 --- /dev/null +++ b/checkpoint-124000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e935798c9904c05eead2d8c6c4dc9d9b69126f6c97311c586a9b52aed5d44bfc +size 73251413 diff --git a/checkpoint-124000/random_states_0.pkl b/checkpoint-124000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..10ab7432f21fdfa87d8bf7045476f9592410ab8e --- /dev/null +++ b/checkpoint-124000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:0ca9f807bed9ccc12bb3581586aedc41bd73e3c840a606180411213fc49f29ba +size 14727 diff --git a/checkpoint-124000/scaler.pt b/checkpoint-124000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..69641131d6c6b4f3d187c497cd23165fd2e325c6 --- /dev/null +++ b/checkpoint-124000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:0a79813cdb55b90fe66de7b8a4c2a9c7cb7e6b2091f094dbf959f427012632e5 +size 557 diff --git a/checkpoint-124000/scheduler.bin b/checkpoint-124000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..34f67556882e2aa7a9ad13d591c32920428c43fb --- /dev/null +++ b/checkpoint-124000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c8ef80c8c82391134088586c279ce1e56881bcadadcad548e740be3563a0c0f6 +size 563 diff --git a/checkpoint-124000/unet/config.json b/checkpoint-124000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..c2d7b8c0d490531013583af57a806d21893e494c --- /dev/null +++ b/checkpoint-124000/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", + "CrossAttnDownBlock2D", + "DownBlock2D" + ], + "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": [ + "UpBlock2D", + "CrossAttnUpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "use_linear_projection": false +} diff --git a/checkpoint-124000/unet/diffusion_pytorch_model.bin b/checkpoint-124000/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..e9c58f976b533e579c431232a196312a2835c710 --- /dev/null +++ b/checkpoint-124000/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:5d2e19bdc89afc53abde4d86bae448b453015b31954aff45d287eb3fc4089a40 +size 36621969 diff --git a/checkpoint-124000/unet_ema/config.json b/checkpoint-124000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..fd6c726d1dca0373a128710e549862649a0598ca --- /dev/null +++ b/checkpoint-124000/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", + "CrossAttnDownBlock2D", + "DownBlock2D" + ], + "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": 124000, + "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": [ + "UpBlock2D", + "CrossAttnUpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "update_after_step": 0, + "use_ema_warmup": true, + "use_linear_projection": false +} diff --git a/checkpoint-124000/unet_ema/diffusion_pytorch_model.bin b/checkpoint-124000/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..34e990f4c48100f6ac48794d16a4fc9ea5dc7024 --- /dev/null +++ b/checkpoint-124000/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:6287bbd52b52bdc75c4e5b12a5db461516b3a59fa8cb261df5f550a46e05981c +size 36616081 diff --git a/checkpoint-124500/optimizer.bin b/checkpoint-124500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..b985fa3eb77c8bd2e79adfe29d901b97351516d9 --- /dev/null +++ b/checkpoint-124500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:0ea64723fcab77cd1a518e6a574d3604859c0423addd4e7a352eaca13deeb23e +size 73251413 diff --git a/checkpoint-124500/random_states_0.pkl b/checkpoint-124500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..1f8e2d9426000b44b3f4ed8bc5dca8432daffd21 --- /dev/null +++ b/checkpoint-124500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:456e6d0ac73c9e91307ae721c79c020d8ad56c3345346767af20db1ae313150a +size 14727 diff --git a/checkpoint-124500/scaler.pt b/checkpoint-124500/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..c259d89194a60240a2c1f46011eb1394879006ee --- /dev/null +++ b/checkpoint-124500/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:4fc1744c12ec914996f4635f14c15b2c2c94ddfdca982106d4a7840033a478e2 +size 557 diff --git a/checkpoint-124500/scheduler.bin b/checkpoint-124500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..30d75a947c485174e3f9eff4e570b96c050f29cd --- /dev/null +++ b/checkpoint-124500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:2d72824fc3a49f6054d16ee4d6f515c582f2efc8d9c9378daf867cdc5730e75b +size 563 diff --git a/checkpoint-124500/unet/config.json b/checkpoint-124500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..c2d7b8c0d490531013583af57a806d21893e494c --- /dev/null +++ b/checkpoint-124500/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", + "CrossAttnDownBlock2D", + "DownBlock2D" + ], + "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": [ + "UpBlock2D", + "CrossAttnUpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "use_linear_projection": false +} diff --git a/checkpoint-124500/unet/diffusion_pytorch_model.bin b/checkpoint-124500/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..cb7de810c6cb7c160f2ad0fa5fcfec666a85f638 --- /dev/null +++ b/checkpoint-124500/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:6e5df118049e721e75ae17024c0025a92bfd8c295ab13f451abf0c4cc5a6b949 +size 36621969 diff --git a/checkpoint-124500/unet_ema/config.json b/checkpoint-124500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..f2e75ca71390536ddb6927c9ebc1908ed915b471 --- /dev/null +++ b/checkpoint-124500/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", + "CrossAttnDownBlock2D", + "DownBlock2D" + ], + "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": 124500, + "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": [ + "UpBlock2D", + "CrossAttnUpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "update_after_step": 0, + "use_ema_warmup": true, + "use_linear_projection": false +} diff --git a/checkpoint-124500/unet_ema/diffusion_pytorch_model.bin b/checkpoint-124500/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..33d5f2c30b8027fe8b1887dcc8df982e5c791479 --- /dev/null +++ b/checkpoint-124500/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:09f0bd94b8c0b7705f695d1303c4e0ac0bd42f30336bd6c4cf56a68df2b0fb32 +size 36616081 diff --git a/checkpoint-125000/optimizer.bin b/checkpoint-125000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..5e25f726d14a10579455ffa54f00a417a62a8590 --- /dev/null +++ b/checkpoint-125000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:bee468b9e715f75b18e756f71e4dc52fc2ebae81bb00731fc6d9c3104c2cfe10 +size 73251413 diff --git a/checkpoint-125000/random_states_0.pkl b/checkpoint-125000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..ac9ca9eaae489aef8eb64fde8f41654d2c70a9a0 --- /dev/null +++ b/checkpoint-125000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:cab29ef0f3a0a618211f8a7638ef94d85a611c485bd8147fadf3f1a218ff897a +size 14727 diff --git a/checkpoint-125000/scaler.pt b/checkpoint-125000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..819a73e258a8a6061947f0eb27cbed4448296631 --- /dev/null +++ b/checkpoint-125000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:674f5e4c034436a15f116b72a151989157eb3831ca1a518ffaf470abf897c853 +size 557 diff --git a/checkpoint-125000/scheduler.bin b/checkpoint-125000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..fcdeb244469e9cce430bb1d98e4f536e0195aeb3 --- /dev/null +++ b/checkpoint-125000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:51f4568adccab273025634a7d4d857b3759fac7cb63279a9fb3bdfe656673281 +size 563 diff --git a/checkpoint-125000/unet/config.json b/checkpoint-125000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..c2d7b8c0d490531013583af57a806d21893e494c --- /dev/null +++ b/checkpoint-125000/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", + "CrossAttnDownBlock2D", + "DownBlock2D" + ], + "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": [ + "UpBlock2D", + "CrossAttnUpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "use_linear_projection": false +} diff --git a/checkpoint-125000/unet/diffusion_pytorch_model.bin b/checkpoint-125000/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..77806935d2ca51e673a21d8951f9bb132b42152f --- /dev/null +++ b/checkpoint-125000/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:21c7957cc20df15eb4cd902b5b71c286e8b91be7e06eaa9d4b2dceb0a16a5b17 +size 36621969 diff --git a/checkpoint-125000/unet_ema/config.json b/checkpoint-125000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..9cfc31b0c55b5dd20b8fed06d06b76f69008eea3 --- /dev/null +++ b/checkpoint-125000/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", + "CrossAttnDownBlock2D", + "DownBlock2D" + ], + "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": 125000, + "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": [ + "UpBlock2D", + "CrossAttnUpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "update_after_step": 0, + "use_ema_warmup": true, + "use_linear_projection": false +} diff --git a/checkpoint-125000/unet_ema/diffusion_pytorch_model.bin b/checkpoint-125000/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..8475a79995d9cb613bd7a0c745a046bfeb997454 --- /dev/null +++ b/checkpoint-125000/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:ce941a0e23fe0731ab64a2ed6a7593eca60c81725619358e9c54335745b0f803 +size 36616081 diff --git a/checkpoint-125500/optimizer.bin b/checkpoint-125500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..fd5732dbdb861222088d3c229ba6c77ca8d2387e --- /dev/null +++ b/checkpoint-125500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:07ed9a37b0bbe0d5c5b685e5391b99ecd1238f716af4e763e67c4fbb696005db +size 73251413 diff --git a/checkpoint-125500/random_states_0.pkl b/checkpoint-125500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..09e76b96a4ceb8dd3fbf0a4ecfe0e4d51ca4ddf2 --- /dev/null +++ b/checkpoint-125500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:8dd748c4c0f309f050b4ca4f8a86e0691c7012099bf5256885569bd1d24bc17d +size 14727 diff --git a/checkpoint-125500/scaler.pt b/checkpoint-125500/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..9ef1c135bc68372c8dc95c89dfd347afbb51deb1 --- /dev/null +++ b/checkpoint-125500/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:2afe81d64c884775a6c0f5dd0a14a2d6d453ef40c16605ba772ec92f11fc2188 +size 557 diff --git a/checkpoint-125500/scheduler.bin b/checkpoint-125500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..14a1e98abea27ff2105770fc40ca074f4a6c951a --- /dev/null +++ b/checkpoint-125500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:6aff5df8b2f7b0a1c58c3ab2ca07f127be66282ba83f193681c87f852a1e460d +size 563 diff --git a/checkpoint-125500/unet/config.json b/checkpoint-125500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..c2d7b8c0d490531013583af57a806d21893e494c --- /dev/null +++ b/checkpoint-125500/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", + "CrossAttnDownBlock2D", + "DownBlock2D" + ], + "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": [ + "UpBlock2D", + "CrossAttnUpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "use_linear_projection": false +} diff --git a/checkpoint-125500/unet/diffusion_pytorch_model.bin b/checkpoint-125500/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..0191755c67b3b52f9ac6e50733ab366fd63c0c5b --- /dev/null +++ b/checkpoint-125500/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b6814b97e426c56a17a7c4c7b8192128f5d60fb61ce3b023030fdbfb5af62d88 +size 36621969 diff --git a/checkpoint-125500/unet_ema/config.json b/checkpoint-125500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..b5d82998f208ec6ea4cd00a8c30b66f630a12213 --- /dev/null +++ b/checkpoint-125500/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", + "CrossAttnDownBlock2D", + "DownBlock2D" + ], + "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": 125500, + "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": [ + "UpBlock2D", + "CrossAttnUpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "update_after_step": 0, + "use_ema_warmup": true, + "use_linear_projection": false +} diff --git a/checkpoint-125500/unet_ema/diffusion_pytorch_model.bin b/checkpoint-125500/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..39c83db99f7c5367201ee5a810e86f541257b3df --- /dev/null +++ b/checkpoint-125500/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:8edff85b8d02f2e3056df51d55df5d7e05aa2349a5bdd9a7c9e4edb7f036e7d1 +size 36616081 diff --git a/checkpoint-126000/optimizer.bin b/checkpoint-126000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..6981447cd00d9c62e37fa91d788b3f3fbee87814 --- /dev/null +++ b/checkpoint-126000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:a9db09d08e58f422ba06f6f0e104775204f0f74a977323e9ed346704623a3b05 +size 73251413 diff --git a/checkpoint-126000/random_states_0.pkl b/checkpoint-126000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..e49bab3edd37ea90be88857dfb0f06819467c40e --- /dev/null +++ b/checkpoint-126000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:2866137d17762735d3cc216b2126d44c3c82880f070b0fd957c299daee05537b +size 14727 diff --git a/checkpoint-126000/scaler.pt b/checkpoint-126000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..592b0446bd1869891675f173eff0eb343772e426 --- /dev/null +++ b/checkpoint-126000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:85f1a1b8e9b1924144f528977a6f525327d01bbeb1353de1bb66c12f6f5cf693 +size 557 diff --git a/checkpoint-126000/scheduler.bin b/checkpoint-126000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..4be69d289f70fa5917c16b5e0a89455691ceaaf3 --- /dev/null +++ b/checkpoint-126000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:87482eb11ff0fb0a9e34ac9b94cdd5b4a69232322c57fd737ad8a4a4cf96a4c5 +size 563 diff --git a/checkpoint-126000/unet/config.json b/checkpoint-126000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..c2d7b8c0d490531013583af57a806d21893e494c --- /dev/null +++ b/checkpoint-126000/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", + "CrossAttnDownBlock2D", + "DownBlock2D" + ], + "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": [ + "UpBlock2D", + "CrossAttnUpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "use_linear_projection": false +} diff --git a/checkpoint-126000/unet/diffusion_pytorch_model.bin b/checkpoint-126000/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..c2dfdcc005c393a5aa5a934c4634184b352970b6 --- /dev/null +++ b/checkpoint-126000/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:24ef35b3a3f9f1e4c9145ff9b5759f876e37eee92bbf92c8ec899d6073efe209 +size 36621969 diff --git a/checkpoint-126000/unet_ema/config.json b/checkpoint-126000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..4aa9de4c05c36f84c0592bc8dd647bd0b488b6b6 --- /dev/null +++ b/checkpoint-126000/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", + "CrossAttnDownBlock2D", + "DownBlock2D" + ], + "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": 126000, + "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": [ + "UpBlock2D", + "CrossAttnUpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "update_after_step": 0, + "use_ema_warmup": true, + "use_linear_projection": false +} diff --git a/checkpoint-126000/unet_ema/diffusion_pytorch_model.bin b/checkpoint-126000/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..7cf6e64b437a5aa8115b1383fe757e2d1b62830b --- /dev/null +++ b/checkpoint-126000/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b85c1ff50940275adaa0cd31d0bade2b3228936889bf38fe0b192c188e0078de +size 36616081 diff --git a/unet/diffusion_pytorch_model.bin b/unet/diffusion_pytorch_model.bin index a49dbef9946d87296347a8d2a254e722b79d938e..4745bb92451e1b0d475dcf9a7c9d58c85c0b89f3 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:c0e4df710cfb22ee93a5df585e849ca05e63f2aae4b68998fb232bd0baee136b +oid sha256:aef3b603bbc9dd238a0a1d88e1ef28479093e0f1d7d98291724dd001acdcf6d8 size 36621969