diff --git a/checkpoint-21000/optimizer.bin b/checkpoint-21000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..13e09cd3285ccd6cee786037f0c51c8418523279 --- /dev/null +++ b/checkpoint-21000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:555810784382569dc3bebca468d938a6e6036879fedcc9b1ac46ae428ca0c88a +size 73251413 diff --git a/checkpoint-21000/random_states_0.pkl b/checkpoint-21000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..b65e200f8255af31e79ea10d53efa43a6800a75d --- /dev/null +++ b/checkpoint-21000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:aa9de06b382006fc1f42fb8d96bd9953370025a8136731bcad8dc9873fbc09a6 +size 14663 diff --git a/checkpoint-21000/scaler.pt b/checkpoint-21000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..454e5ef1510361867b33989c4d4377e55b44105d --- /dev/null +++ b/checkpoint-21000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:da692e8a61b0469d66a05ee87d5569d6a98e9a4d9ac49ff837bd9b7979d7ad48 +size 557 diff --git a/checkpoint-21000/scheduler.bin b/checkpoint-21000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..c8f02d18e456f19e91d2a07ff347464c523a921e --- /dev/null +++ b/checkpoint-21000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:31f01a39c0fee8bb5e2d901a85435109f5a2d38467d324b38a24595e1b113904 +size 563 diff --git a/checkpoint-21000/unet/config.json b/checkpoint-21000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..bb1cd664fc36d4dc821a4f0fdae4b94d7d99b356 --- /dev/null +++ b/checkpoint-21000/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-21000/unet/diffusion_pytorch_model.bin b/checkpoint-21000/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..acf9da31d370d4862f53c1a2f86be5f66ee16867 --- /dev/null +++ b/checkpoint-21000/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:9a589c69d9cd714d567b2554c822661c4b269d666bf4c1c5640fa82b1ddb1d2a +size 36621969 diff --git a/checkpoint-21000/unet_ema/config.json b/checkpoint-21000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..f25611f8e2c09eccae2b2a7760123a9da6804a95 --- /dev/null +++ b/checkpoint-21000/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": 21000, + "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-21000/unet_ema/diffusion_pytorch_model.bin b/checkpoint-21000/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..9fcc9b9cc6725171fda2aff2df07d7e9be7f7930 --- /dev/null +++ b/checkpoint-21000/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:32751ed117b76dbbaeccde9acf197264aa83a06d8f3acf3ae964483479bc8014 +size 36616081 diff --git a/checkpoint-21500/optimizer.bin b/checkpoint-21500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..dfec7794367add94d5ef8954ff49d483b7b2726a --- /dev/null +++ b/checkpoint-21500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:9623e7b3bc90c2042f18189bb00e6e9fbaac97501c539e74508db7c987f0f0d3 +size 73251413 diff --git a/checkpoint-21500/random_states_0.pkl b/checkpoint-21500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..8a3ae79f9c0438566a3052967a449c38b8dd893b --- /dev/null +++ b/checkpoint-21500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:495de690b1048fdb1eaa09eebe05a629cff9080c3fe312b0fc49731e6a9f4421 +size 14663 diff --git a/checkpoint-21500/scaler.pt b/checkpoint-21500/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..b21ab0ddfe9534387b08527cf68ddefb92ff5e2d --- /dev/null +++ b/checkpoint-21500/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:45ac1a1a1f98b3964d76c33e861170daff6e880017ea91dcfeb6f1af152554e5 +size 557 diff --git a/checkpoint-21500/scheduler.bin b/checkpoint-21500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..2513a6ea23d2b0e67a38cf4b0ff8a93a7fed286b --- /dev/null +++ b/checkpoint-21500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:4181df1262670dd832b648be5211fa363f177f425ba2b40a75356d9a5c68b1af +size 563 diff --git a/checkpoint-21500/unet/config.json b/checkpoint-21500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..bb1cd664fc36d4dc821a4f0fdae4b94d7d99b356 --- /dev/null +++ b/checkpoint-21500/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-21500/unet/diffusion_pytorch_model.bin b/checkpoint-21500/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..75f8bda8abbffe8e01fa7384d1d9592dfe26cdc0 --- /dev/null +++ b/checkpoint-21500/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:8f9de1bd9fca83667286091cb220c62dd9173eff945c674840fe1afefc6f662c +size 36621969 diff --git a/checkpoint-21500/unet_ema/config.json b/checkpoint-21500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..b2ff759e527b1a81bb1fac0e47e6a333af307b76 --- /dev/null +++ b/checkpoint-21500/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": 21500, + "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-21500/unet_ema/diffusion_pytorch_model.bin b/checkpoint-21500/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..ff0443cfdb7cb78a8fad5b7201b99298d429a3f0 --- /dev/null +++ b/checkpoint-21500/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:0c04adf7b6a0d336d586cddd597b51e1a16ecc192f538aef5184d1e835a7bdfa +size 36616081 diff --git a/checkpoint-22000/optimizer.bin b/checkpoint-22000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..a72ef57b91cb4e9b7972b7749751f15653a3b907 --- /dev/null +++ b/checkpoint-22000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:286dbaea1a4107c39dd24010ac883a4d66e22d44ff0a51b09caa5793ec4d6672 +size 73251413 diff --git a/checkpoint-22000/random_states_0.pkl b/checkpoint-22000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..d7abf9c540fe3be7b13f57368a7cfe6be2f456b5 --- /dev/null +++ b/checkpoint-22000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:30b2028194f32060abea7b2829ba46ecaeb22c98c12f024a43a6a2d0e5f33604 +size 14663 diff --git a/checkpoint-22000/scaler.pt b/checkpoint-22000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..d6bfc92c3b81e7767e9cf88770a77d12b7171825 --- /dev/null +++ b/checkpoint-22000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:01e25507d940068c5fde23dde9702b0deb5c84279a3afbab729b405bce252f3e +size 557 diff --git a/checkpoint-22000/scheduler.bin b/checkpoint-22000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..65a2a7e5a0217c5ef21fd5bcb863ce0f47fb0fc4 --- /dev/null +++ b/checkpoint-22000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:13f76c143f045054c35ebd3d512c37acf70584259be8b0803d8defdd8ae55b18 +size 563 diff --git a/checkpoint-22000/unet/config.json b/checkpoint-22000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..bb1cd664fc36d4dc821a4f0fdae4b94d7d99b356 --- /dev/null +++ b/checkpoint-22000/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-22000/unet/diffusion_pytorch_model.bin b/checkpoint-22000/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..848411fa6cc78197050bb13e5dbe554a570a3cea --- /dev/null +++ b/checkpoint-22000/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:313b08fc0390ad2ca3f67d0411498075fd5e1cbc226daa024ef6ed8239058139 +size 36621969 diff --git a/checkpoint-22000/unet_ema/config.json b/checkpoint-22000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..e720e1bac9daf3eb6b990493e09f3d3a113db77b --- /dev/null +++ b/checkpoint-22000/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": 22000, + "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-22000/unet_ema/diffusion_pytorch_model.bin b/checkpoint-22000/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..c45983397debda8e51f276129b2a46c4342f530a --- /dev/null +++ b/checkpoint-22000/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:a4d6df93c97229319483ec3e9a114970ff505a7127132f0fb42b1033eb72e27f +size 36616081 diff --git a/checkpoint-22500/optimizer.bin b/checkpoint-22500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..8cdf26e11ebae6b2cfdb7fb937d70b112ac89fc1 --- /dev/null +++ b/checkpoint-22500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:3c8e851bd6517207d51dcf313f8614975685f729a3852750ead342fd2a523b31 +size 73251413 diff --git a/checkpoint-22500/random_states_0.pkl b/checkpoint-22500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..93807aff413cd6c20650d6967d7fc0a535e38660 --- /dev/null +++ b/checkpoint-22500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:6250bb69014ece110965caff6ba2f345c8fd4abac600cb1da2672ac7fd906d51 +size 14663 diff --git a/checkpoint-22500/scaler.pt b/checkpoint-22500/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..dfb8141650649b38c7e8199413107de3ab5a259c --- /dev/null +++ b/checkpoint-22500/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:3dcb3afe27cf09061cd3b8e44e41ee0c25fee5ba2c9e6cb5c332c55914728e09 +size 557 diff --git a/checkpoint-22500/scheduler.bin b/checkpoint-22500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..1386527c11e2be34e46ed6022bba5de152110b36 --- /dev/null +++ b/checkpoint-22500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:d08a3996a18f55aa96dac6abfa70e2e9d7fac834972835880b77ad8e8337c820 +size 563 diff --git a/checkpoint-22500/unet/config.json b/checkpoint-22500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..bb1cd664fc36d4dc821a4f0fdae4b94d7d99b356 --- /dev/null +++ b/checkpoint-22500/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-22500/unet/diffusion_pytorch_model.bin b/checkpoint-22500/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..ecf9f2189abdd5bf82ae0fef9c2ac78dd2ff6113 --- /dev/null +++ b/checkpoint-22500/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:1bf5e81c08963e7ab4aff8bb9098b91544bde1d098e38522dac8dcd170497eae +size 36621969 diff --git a/checkpoint-22500/unet_ema/config.json b/checkpoint-22500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..e3e06a125a91955d8746b41ff82856184c028fad --- /dev/null +++ b/checkpoint-22500/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": 22500, + "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-22500/unet_ema/diffusion_pytorch_model.bin b/checkpoint-22500/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..0fb528cddad96232b6c552017234a2b42554af72 --- /dev/null +++ b/checkpoint-22500/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:fe3e9d4a15cb4066ff3415ac1e7cc43cd1575bb4af759b6c057ba4d9b701bd1a +size 36616081 diff --git a/checkpoint-23000/optimizer.bin b/checkpoint-23000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..cc5b8faff5273e0a1fb178e5b0b0a532d538093a --- /dev/null +++ b/checkpoint-23000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:bc3f17d82025d54b40077982e8cf98825f939f8f0391de231cf4ba4cb8424dce +size 73251413 diff --git a/checkpoint-23000/random_states_0.pkl b/checkpoint-23000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..2d28810a8c1bbb83c9d35d2c25748767fc968431 --- /dev/null +++ b/checkpoint-23000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:67a2abcd92df03226d4f90be1f290fbe6af1a905699852672013222888e6d829 +size 14663 diff --git a/checkpoint-23000/scaler.pt b/checkpoint-23000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..55e415c5492848bc014571e8352ccc1709427c69 --- /dev/null +++ b/checkpoint-23000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:a16806f5d213d5174a8482b1c872f672d819d63b5279288bbc5f40bb3f8bd58b +size 557 diff --git a/checkpoint-23000/scheduler.bin b/checkpoint-23000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..92d485f9e3ca3ea437458da9acda754c7b2030cf --- /dev/null +++ b/checkpoint-23000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:2210b1e97945d57b51259cf652b1b3f8e9edca0639523f2a54d417a0d802f71a +size 563 diff --git a/checkpoint-23000/unet/config.json b/checkpoint-23000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..bb1cd664fc36d4dc821a4f0fdae4b94d7d99b356 --- /dev/null +++ b/checkpoint-23000/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-23000/unet/diffusion_pytorch_model.bin b/checkpoint-23000/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..6d1e33e5cc368c2b3985a8d7114f2012a69ad4ec --- /dev/null +++ b/checkpoint-23000/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:ca64fbf87e537a7fb73c326e66cdaa7a4ff659e86655db05bd6f19d544da8320 +size 36621969 diff --git a/checkpoint-23000/unet_ema/config.json b/checkpoint-23000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..f56dd40e127603a71125bd241326512827f293bb --- /dev/null +++ b/checkpoint-23000/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": 23000, + "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-23000/unet_ema/diffusion_pytorch_model.bin b/checkpoint-23000/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..63712bab62c1d9085a92c8ebc0b90c1c723c06db --- /dev/null +++ b/checkpoint-23000/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7260b03d49d65e188e7b5de561c4db49e8b815b38c5384fdd5c8bb906a37ece5 +size 36616081 diff --git a/checkpoint-23500/optimizer.bin b/checkpoint-23500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..a33079077a018c8a636ac1553ddccc96bcb8579c --- /dev/null +++ b/checkpoint-23500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:839c7aafbe0ff5496123654a69bbef3850fecb016e68e5ee62f27f5675569173 +size 73251413 diff --git a/checkpoint-23500/random_states_0.pkl b/checkpoint-23500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..b7dc3c5a8b2c9ff872b76ea6757805c167bfc9cd --- /dev/null +++ b/checkpoint-23500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:311503c3c60b57ff8894a866457ae58828a46b2f545952c86e261fd815685728 +size 14663 diff --git a/checkpoint-23500/scaler.pt b/checkpoint-23500/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..5c1d3de8f9ffdbf04bb29265eb92a422c737f4e0 --- /dev/null +++ b/checkpoint-23500/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e7cfc03fdc79af083558f7c1bf7400c367307566936efb1b23ca7ad8fe88645f +size 557 diff --git a/checkpoint-23500/scheduler.bin b/checkpoint-23500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..eaa809d2ddcbefda2b87b889bf6ce08c097affb2 --- /dev/null +++ b/checkpoint-23500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:10d7494e613567a6083f04637c7acd3fd8e5aa5102780943d128e19d81f3b1a2 +size 563 diff --git a/checkpoint-23500/unet/config.json b/checkpoint-23500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..bb1cd664fc36d4dc821a4f0fdae4b94d7d99b356 --- /dev/null +++ b/checkpoint-23500/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-23500/unet/diffusion_pytorch_model.bin b/checkpoint-23500/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..9d4bcbce32e71153965e2da9194e9f579518c6c1 --- /dev/null +++ b/checkpoint-23500/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c4b579e478597ff6ba5cdf5015b620481a7409c2fcd18f6def8255a5e1207466 +size 36621969 diff --git a/checkpoint-23500/unet_ema/config.json b/checkpoint-23500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..5469ee424c92d07007f968ca5938d881fadbd307 --- /dev/null +++ b/checkpoint-23500/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": 23500, + "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-23500/unet_ema/diffusion_pytorch_model.bin b/checkpoint-23500/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..03f84bdb06c5efaf3d3cf65d65b6ce37aac89320 --- /dev/null +++ b/checkpoint-23500/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:47efc9922afe4a46c49f3146dac5de73b5c5f9b45eade61eda431c4abdf51a10 +size 36616081 diff --git a/checkpoint-24000/optimizer.bin b/checkpoint-24000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..2bf8a9402a9ca5e9d264e10822ba9e83b4a21b5b --- /dev/null +++ b/checkpoint-24000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:19b306c1913af17d45459a59285cc435a02b8d8b228c91217c9495e4531fc493 +size 73251413 diff --git a/checkpoint-24000/random_states_0.pkl b/checkpoint-24000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..cb450278c9441b2b5d79b8fcaaa0c346f03aad2c --- /dev/null +++ b/checkpoint-24000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:402de6ddaa434abe18abe221dcb9cb8a6d2ad8cf70020310c9ddcdcdfc614b8e +size 14663 diff --git a/checkpoint-24000/scaler.pt b/checkpoint-24000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..27b403d72b48f76fe7b310c7cb888c94c1b71aec --- /dev/null +++ b/checkpoint-24000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:1f9fed2405de7a826c01209d4198713c7cebcc07e1118b45c1dccf76ecba501a +size 557 diff --git a/checkpoint-24000/scheduler.bin b/checkpoint-24000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..31f94d98a0bf1bfd26cf202d95dc5c653dbb6b71 --- /dev/null +++ b/checkpoint-24000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c910d0dd37dd396620d10d0e9841f92d568a7dba4f7a7af70a8c183ffbcbaf46 +size 563 diff --git a/checkpoint-24000/unet/config.json b/checkpoint-24000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..bb1cd664fc36d4dc821a4f0fdae4b94d7d99b356 --- /dev/null +++ b/checkpoint-24000/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-24000/unet/diffusion_pytorch_model.bin b/checkpoint-24000/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..8337b4d019d5869a46ba8e6f6fc57a9f8da9bc44 --- /dev/null +++ b/checkpoint-24000/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e9f16d30855442519e89a5fa96b379b0cb2ad512acb56db1554729473ce84801 +size 36621969 diff --git a/checkpoint-24000/unet_ema/config.json b/checkpoint-24000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..b4df374e30dc561808675e168ee55118f6145233 --- /dev/null +++ b/checkpoint-24000/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": 24000, + "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-24000/unet_ema/diffusion_pytorch_model.bin b/checkpoint-24000/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..a65dd10be105276d133db50aa8cee13771a9e3e7 --- /dev/null +++ b/checkpoint-24000/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:39a725ae6798afc229a8e524fa6f9aa24d819489ae0674247a1ef34cddf7da90 +size 36616081 diff --git a/unet/diffusion_pytorch_model.bin b/unet/diffusion_pytorch_model.bin index da6a89100e6cd5c1590bab6f50994a73f6febd0f..54493fbaf1423f7aec3972ab64d7c438dfd44cfb 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:8506aa8e34ca8e9d4729f58fe15a7b3b2278610e5bd4b3550568b1629ceda328 +oid sha256:b5b26a9e861b96992236c20e530678e7ae47f5e9d6157e9c0593c8cfcf0303ad size 36621969