diff --git a/checkpoint-10000/optimizer.bin b/checkpoint-10000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..4c0fcc1cbe1a04078762c5c6f00a3fd487d47bef --- /dev/null +++ b/checkpoint-10000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:5b090d607af6016381dfcf39870c8e914ab0c521cc23f63856141be27abe1a27 +size 126008021 diff --git a/checkpoint-10000/random_states_0.pkl b/checkpoint-10000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..5604df260bc9bbb9cd54e39a407ad76149f4df8b --- /dev/null +++ b/checkpoint-10000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:dd9d5b306c500cf427061ff05d876b3f966c3ff7c0296cf75f11614c19697acd +size 14727 diff --git a/checkpoint-10000/scaler.pt b/checkpoint-10000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..5f715fea07d8187e75aecd5611f5981691c13901 --- /dev/null +++ b/checkpoint-10000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:a5861e9596f8f8ce01b577bd4dd492e27f1e29316ddca8a2e44df0d7f6501728 +size 557 diff --git a/checkpoint-10000/scheduler.bin b/checkpoint-10000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..7914b736147f3947fe7bf9be6d0924f9fc99fa1f --- /dev/null +++ b/checkpoint-10000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:716a4c1cb77e6c92b6c8d28de079a6d120ebdd3b04b64c98719aa262ed9d67fa +size 563 diff --git a/checkpoint-10000/unet/config.json b/checkpoint-10000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..06fb387aa3ed37ab3b2f455b5469059d55604f74 --- /dev/null +++ b/checkpoint-10000/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": [ + 128, + 128, + 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", + "CrossAttnDownBlock2D", + "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", + "CrossAttnUpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "use_linear_projection": false +} diff --git a/checkpoint-10000/unet/diffusion_pytorch_model.bin b/checkpoint-10000/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..bcd314cd00c0f5d078aee1c661350229300703b8 --- /dev/null +++ b/checkpoint-10000/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:0101a22b3345e80c6388de82e529415fb13b3dea20ef87ee2d119b4f2a338351 +size 63003985 diff --git a/checkpoint-10000/unet_ema/config.json b/checkpoint-10000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..37c676f4d6eac193b798f410e5ae3f850f620131 --- /dev/null +++ b/checkpoint-10000/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": [ + 128, + 128, + 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", + "CrossAttnDownBlock2D", + "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": 10000, + "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", + "CrossAttnUpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "update_after_step": 0, + "use_ema_warmup": true, + "use_linear_projection": false +} diff --git a/checkpoint-10000/unet_ema/diffusion_pytorch_model.bin b/checkpoint-10000/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..ff6c46acd7a789845a7cb6a51c7ea01fd04109cf --- /dev/null +++ b/checkpoint-10000/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:bc0ddb8ade30df6a744e78e624e80ea61a1d347ff3fa9f5885eb12f029a23342 +size 62996305 diff --git a/checkpoint-10500/optimizer.bin b/checkpoint-10500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..5aab2f10389612886bebbab46490e4f4572d331d --- /dev/null +++ b/checkpoint-10500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b36d5996c5df654edb110905edc9bae08e11d110b43cf84c5b3b70f15de81f04 +size 126008021 diff --git a/checkpoint-10500/random_states_0.pkl b/checkpoint-10500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..0a4190a73968ac5d0a20898ac9c3003f0d3779fa --- /dev/null +++ b/checkpoint-10500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:ea186ed1953c1461dfdb4b4f8cd64b90bf2f20d5c4795397be4eab2a032dedc9 +size 14727 diff --git a/checkpoint-10500/scaler.pt b/checkpoint-10500/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..5f7202434cb3ede78a9c885d02eec66e2c3832ef --- /dev/null +++ b/checkpoint-10500/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:df7dc3303bdc1a2737159ab5ce6ac6dfaf84d16b8f82e60421bc66305d6aac5e +size 557 diff --git a/checkpoint-10500/scheduler.bin b/checkpoint-10500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..1888b89521acc9dfbc9ed33b441a3fb4e688fc53 --- /dev/null +++ b/checkpoint-10500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e010ff830775f5fe481ffd405dca09282028c243d72f575341640b2632c429b9 +size 563 diff --git a/checkpoint-10500/unet/config.json b/checkpoint-10500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..06fb387aa3ed37ab3b2f455b5469059d55604f74 --- /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": [ + 128, + 128, + 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", + "CrossAttnDownBlock2D", + "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", + "CrossAttnUpBlock2D", + "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..805d607a44f22fe85db185ed6202e1ca82453b2f --- /dev/null +++ b/checkpoint-10500/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:aebe2767eddec5964362bccb08e43d52c06578dd80580e530baf667c100d5287 +size 63003985 diff --git a/checkpoint-10500/unet_ema/config.json b/checkpoint-10500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..b248beec3521ff5bbcf2ba0c73348f377622261e --- /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": [ + 128, + 128, + 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", + "CrossAttnDownBlock2D", + "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": 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": [ + "UpBlock2D", + "CrossAttnUpBlock2D", + "CrossAttnUpBlock2D", + "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..2ee3dde75eea0dab001f7151b96e2d49bcf533d2 --- /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:9a9ffebdad5f8dddae2f5153346c98ff7a75dd2b4116675eef75c9d7579992e0 +size 62996305 diff --git a/checkpoint-11000/optimizer.bin b/checkpoint-11000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..d6d5c069f00fa83aec2b029781d7b6b7d0c754d5 --- /dev/null +++ b/checkpoint-11000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:9f99a18b8660eabc917419721f0447562535c746d575566a242639883e7f661c +size 126008021 diff --git a/checkpoint-11000/random_states_0.pkl b/checkpoint-11000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..e56fc1e8f6783ad22797b31f5175441b194b208a --- /dev/null +++ b/checkpoint-11000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:8c5f940a906ca49f86eed5a962a6426c1886f5e7960c0d62c6916a9899e36816 +size 14727 diff --git a/checkpoint-11000/scaler.pt b/checkpoint-11000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..115263c9235918fc2a72413ed7e00c969b8d3527 --- /dev/null +++ b/checkpoint-11000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7fa181fa360d46feed4180ea17c8b6a4a879a9b4231c2e91aff2be20be9076cc +size 557 diff --git a/checkpoint-11000/scheduler.bin b/checkpoint-11000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..f88642ab5e5ef188e19cae139ad13cc6b710f4cf --- /dev/null +++ b/checkpoint-11000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:4c94ae12037561999a6d2a3dcc05fdbbe7cd62e455f948746fc011252f09b7ea +size 563 diff --git a/checkpoint-11000/unet/config.json b/checkpoint-11000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..06fb387aa3ed37ab3b2f455b5469059d55604f74 --- /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": [ + 128, + 128, + 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", + "CrossAttnDownBlock2D", + "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", + "CrossAttnUpBlock2D", + "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..f32e7bd385184581ee545b5aee58c14b25ec56e4 --- /dev/null +++ b/checkpoint-11000/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e1158d86c311765e5f384876362976c933cae0c14a61c62dead56bd20de8de13 +size 63003985 diff --git a/checkpoint-11000/unet_ema/config.json b/checkpoint-11000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..a937387266655f08817c5d6ac5eea7372cc899f6 --- /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": [ + 128, + 128, + 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", + "CrossAttnDownBlock2D", + "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": 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": [ + "UpBlock2D", + "CrossAttnUpBlock2D", + "CrossAttnUpBlock2D", + "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..5b9c4534c41af2b91b3ccacf5c8bf4d553d9eda2 --- /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:6004117229e7822e06dabb7ac3662dd41a744b85231d2e83d3c450402f0dabb9 +size 62996305 diff --git a/checkpoint-11500/optimizer.bin b/checkpoint-11500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..3669520f02ce7e08d5a25b75d0f5ca936f561771 --- /dev/null +++ b/checkpoint-11500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:fc607468e063f33ac98757094a1f92f9d8f3f1cc7040cadc24e975ef7cc5a9a6 +size 126008021 diff --git a/checkpoint-11500/random_states_0.pkl b/checkpoint-11500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..15e24699633c3790beb183982bf43e9c64c55574 --- /dev/null +++ b/checkpoint-11500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:f80fcf338a906fa6d8454d4f5f314e61db61dc407cec55ab8774b401f6c6b359 +size 14727 diff --git a/checkpoint-11500/scaler.pt b/checkpoint-11500/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..115263c9235918fc2a72413ed7e00c969b8d3527 --- /dev/null +++ b/checkpoint-11500/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7fa181fa360d46feed4180ea17c8b6a4a879a9b4231c2e91aff2be20be9076cc +size 557 diff --git a/checkpoint-11500/scheduler.bin b/checkpoint-11500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..39aaea5c899fc348dbcf4d0d80a25c73617de733 --- /dev/null +++ b/checkpoint-11500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:f809c4c923ded5bc448297b913b98ffd4f4e70922e5ce149c4e420a3b5988e8f +size 563 diff --git a/checkpoint-11500/unet/config.json b/checkpoint-11500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..06fb387aa3ed37ab3b2f455b5469059d55604f74 --- /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": [ + 128, + 128, + 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", + "CrossAttnDownBlock2D", + "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", + "CrossAttnUpBlock2D", + "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..f32e7bd385184581ee545b5aee58c14b25ec56e4 --- /dev/null +++ b/checkpoint-11500/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e1158d86c311765e5f384876362976c933cae0c14a61c62dead56bd20de8de13 +size 63003985 diff --git a/checkpoint-11500/unet_ema/config.json b/checkpoint-11500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..14c830dd36d1622d4eaeb9829d7cb6cd2f248581 --- /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": [ + 128, + 128, + 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", + "CrossAttnDownBlock2D", + "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": 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": [ + "UpBlock2D", + "CrossAttnUpBlock2D", + "CrossAttnUpBlock2D", + "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..0577d5dcca7496cc2df47dec9c1b77ae7e4a9823 --- /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:9eedfd6f6b326c525e67f904217a8eafadfcde751cef42d2db7c7a93408de28a +size 62996305 diff --git a/checkpoint-12000/optimizer.bin b/checkpoint-12000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..8ae99ad61466bad1656150c13275fab81a5cf5b2 --- /dev/null +++ b/checkpoint-12000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:a3380c2f30a81a02d88838efa9e8be25010083799d69e55749b0dbf355a93c93 +size 126008021 diff --git a/checkpoint-12000/random_states_0.pkl b/checkpoint-12000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..2c7a8fd81cea92244afaad3baf3f9f3473bf9c7f --- /dev/null +++ b/checkpoint-12000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:0333b197630cbf42f4daf53a7c34761da13c0b0a85b16a257dfd4b77c9293c33 +size 14727 diff --git a/checkpoint-12000/scaler.pt b/checkpoint-12000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..115263c9235918fc2a72413ed7e00c969b8d3527 --- /dev/null +++ b/checkpoint-12000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7fa181fa360d46feed4180ea17c8b6a4a879a9b4231c2e91aff2be20be9076cc +size 557 diff --git a/checkpoint-12000/scheduler.bin b/checkpoint-12000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..df25eb4d0a2ebc1dcab8cd32788cd37a6c33d34a --- /dev/null +++ b/checkpoint-12000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:cdb71a91ca2d955f0d9cf47c9afe50f8b92092d141b2aec0b57952f7f34c1cfa +size 563 diff --git a/checkpoint-12000/unet/config.json b/checkpoint-12000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..06fb387aa3ed37ab3b2f455b5469059d55604f74 --- /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": [ + 128, + 128, + 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", + "CrossAttnDownBlock2D", + "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", + "CrossAttnUpBlock2D", + "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..f32e7bd385184581ee545b5aee58c14b25ec56e4 --- /dev/null +++ b/checkpoint-12000/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e1158d86c311765e5f384876362976c933cae0c14a61c62dead56bd20de8de13 +size 63003985 diff --git a/checkpoint-12000/unet_ema/config.json b/checkpoint-12000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..56e0e1185255bda25cf71db63dd6c05bee163814 --- /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": [ + 128, + 128, + 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", + "CrossAttnDownBlock2D", + "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": 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": [ + "UpBlock2D", + "CrossAttnUpBlock2D", + "CrossAttnUpBlock2D", + "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..a04be0e3d78ce2762044bddb0fbf632d9b053729 --- /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:0157acd45e707e424ffa174c7d6606d367dc055b23271715fb0ab38ccf917b9b +size 62996305 diff --git a/checkpoint-12500/optimizer.bin b/checkpoint-12500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..103bbf271940e0f5046ba9ed19bc1ee6efc32ee8 --- /dev/null +++ b/checkpoint-12500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:74a8377a5e17b3145107e3e0e735f7abe1ca647f89c52701e79c2c7329c18047 +size 126008021 diff --git a/checkpoint-12500/random_states_0.pkl b/checkpoint-12500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..0a457efeb5bdfa485ed9b9dd95e033b8ed39900a --- /dev/null +++ b/checkpoint-12500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:72da64f6155ffba7894d19ac1c8ea61f0a4542a6e95e8875ee6af722b0e54262 +size 14727 diff --git a/checkpoint-12500/scaler.pt b/checkpoint-12500/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..115263c9235918fc2a72413ed7e00c969b8d3527 --- /dev/null +++ b/checkpoint-12500/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7fa181fa360d46feed4180ea17c8b6a4a879a9b4231c2e91aff2be20be9076cc +size 557 diff --git a/checkpoint-12500/scheduler.bin b/checkpoint-12500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..dd77965b6124b046e7d446b837d6167232cb50cd --- /dev/null +++ b/checkpoint-12500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:827cb33f0a92516b727fe6c9ba128218d2a851815d39a4f8111f682d429f1fc6 +size 563 diff --git a/checkpoint-12500/unet/config.json b/checkpoint-12500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..06fb387aa3ed37ab3b2f455b5469059d55604f74 --- /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": [ + 128, + 128, + 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", + "CrossAttnDownBlock2D", + "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", + "CrossAttnUpBlock2D", + "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..f32e7bd385184581ee545b5aee58c14b25ec56e4 --- /dev/null +++ b/checkpoint-12500/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e1158d86c311765e5f384876362976c933cae0c14a61c62dead56bd20de8de13 +size 63003985 diff --git a/checkpoint-12500/unet_ema/config.json b/checkpoint-12500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..2d802b83ca7a0f41313f468c2a4d809a231ab1c9 --- /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": [ + 128, + 128, + 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", + "CrossAttnDownBlock2D", + "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": 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": [ + "UpBlock2D", + "CrossAttnUpBlock2D", + "CrossAttnUpBlock2D", + "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..c43ae0a11c12426f8c128693d3f1b0c221b99682 --- /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:c6d0c3aec77148aad6b5e594a54b1f11ff0e467853686e30bb644d3fcce90d9f +size 62996305 diff --git a/checkpoint-13000/optimizer.bin b/checkpoint-13000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..2d8d66d75845ba0484b723f630ea569bf6e9a55e --- /dev/null +++ b/checkpoint-13000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b2e901613cb3db0eccea4d76bebf04c7c9a52a695b213878902553c4b59acd88 +size 126008021 diff --git a/checkpoint-13000/random_states_0.pkl b/checkpoint-13000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..fc93952d3b8ae6099ddb978b5d2b4805a9595143 --- /dev/null +++ b/checkpoint-13000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:1bba5aa6de53a1968c802281b3699f97ff59cbfb204b335a0fc952b819763b89 +size 14727 diff --git a/checkpoint-13000/scaler.pt b/checkpoint-13000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..115263c9235918fc2a72413ed7e00c969b8d3527 --- /dev/null +++ b/checkpoint-13000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7fa181fa360d46feed4180ea17c8b6a4a879a9b4231c2e91aff2be20be9076cc +size 557 diff --git a/checkpoint-13000/scheduler.bin b/checkpoint-13000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..ad4f17e8dc810f93fe8be8a57f43a1ffe8e069e3 --- /dev/null +++ b/checkpoint-13000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:46834e5388269885ffa7d164604c92c210c2588514bb14d02dcede4fc00b6bda +size 563 diff --git a/checkpoint-13000/unet/config.json b/checkpoint-13000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..06fb387aa3ed37ab3b2f455b5469059d55604f74 --- /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": [ + 128, + 128, + 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", + "CrossAttnDownBlock2D", + "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", + "CrossAttnUpBlock2D", + "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..f32e7bd385184581ee545b5aee58c14b25ec56e4 --- /dev/null +++ b/checkpoint-13000/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e1158d86c311765e5f384876362976c933cae0c14a61c62dead56bd20de8de13 +size 63003985 diff --git a/checkpoint-13000/unet_ema/config.json b/checkpoint-13000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..9434302749d00de1863dcbea5fa271e5b41b1cf7 --- /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": [ + 128, + 128, + 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", + "CrossAttnDownBlock2D", + "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": 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": [ + "UpBlock2D", + "CrossAttnUpBlock2D", + "CrossAttnUpBlock2D", + "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..92582af6de6729d8eff16a903ebf259c7b709e64 --- /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:f3330e4940b8ba331e752bb308fe83f67d58af119d83e6fe1fc6948af621d9f6 +size 62996305 diff --git a/checkpoint-13500/optimizer.bin b/checkpoint-13500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..52d8966e5a85a1393b946045ecf9776716b83dae --- /dev/null +++ b/checkpoint-13500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:04522e5daa25afb7517656e49e690733fb48db2db80f7b51fff2bf56bc723430 +size 126008021 diff --git a/checkpoint-13500/random_states_0.pkl b/checkpoint-13500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..47466471ee3660f028a76623ac5e039a40663961 --- /dev/null +++ b/checkpoint-13500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c92643f1986445cc256ba61644ab69456e51ff66efeb8e3c5708d90e023192eb +size 14727 diff --git a/checkpoint-13500/scaler.pt b/checkpoint-13500/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..115263c9235918fc2a72413ed7e00c969b8d3527 --- /dev/null +++ b/checkpoint-13500/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7fa181fa360d46feed4180ea17c8b6a4a879a9b4231c2e91aff2be20be9076cc +size 557 diff --git a/checkpoint-13500/scheduler.bin b/checkpoint-13500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..19b0d67ae3f54e7089702d7a62417270fba77854 --- /dev/null +++ b/checkpoint-13500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b67f5e11e6c4d04cbf866637358f9d603b19a07f64450c8aefaa88f2ffa59f8d +size 563 diff --git a/checkpoint-13500/unet/config.json b/checkpoint-13500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..06fb387aa3ed37ab3b2f455b5469059d55604f74 --- /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": [ + 128, + 128, + 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", + "CrossAttnDownBlock2D", + "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", + "CrossAttnUpBlock2D", + "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..f32e7bd385184581ee545b5aee58c14b25ec56e4 --- /dev/null +++ b/checkpoint-13500/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e1158d86c311765e5f384876362976c933cae0c14a61c62dead56bd20de8de13 +size 63003985 diff --git a/checkpoint-13500/unet_ema/config.json b/checkpoint-13500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..9d0e47f179a2b248a30375fd482ea3389f8dad28 --- /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": [ + 128, + 128, + 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", + "CrossAttnDownBlock2D", + "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": 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": [ + "UpBlock2D", + "CrossAttnUpBlock2D", + "CrossAttnUpBlock2D", + "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..bd09036d10285b467037d93c39c80e91f22d5097 --- /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:c2f1adfafe4980245c0390227fe6f81f53cae2f1976784b03a12c6e6ca23adfb +size 62996305 diff --git a/checkpoint-14000/optimizer.bin b/checkpoint-14000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..a1670f5c202bbffaebb660921dfc098ea8c57b62 --- /dev/null +++ b/checkpoint-14000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:ea2bca6b8044b88cde7dd316ee4d6bfaf20a64f545d7c7743575b4c3e6408161 +size 126008021 diff --git a/checkpoint-14000/random_states_0.pkl b/checkpoint-14000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..c1b7ccec844f7eb9b8aeb2c02acadcf2341579f6 --- /dev/null +++ b/checkpoint-14000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:8548fcb0b508d577fa85d76a2cc8269e2455d2c653752f2b5aa510d94f4ef480 +size 14727 diff --git a/checkpoint-14000/scaler.pt b/checkpoint-14000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..115263c9235918fc2a72413ed7e00c969b8d3527 --- /dev/null +++ b/checkpoint-14000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7fa181fa360d46feed4180ea17c8b6a4a879a9b4231c2e91aff2be20be9076cc +size 557 diff --git a/checkpoint-14000/scheduler.bin b/checkpoint-14000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..5f76ff065f73541be397be160c5c1821335ae446 --- /dev/null +++ b/checkpoint-14000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:5f9b2387959cd7f4735c1d305d0f331ba02fe09ecf794af2836570ecc54cdf4b +size 563 diff --git a/checkpoint-14000/unet/config.json b/checkpoint-14000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..06fb387aa3ed37ab3b2f455b5469059d55604f74 --- /dev/null +++ b/checkpoint-14000/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": [ + 128, + 128, + 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", + "CrossAttnDownBlock2D", + "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", + "CrossAttnUpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "use_linear_projection": false +} diff --git a/checkpoint-14000/unet/diffusion_pytorch_model.bin b/checkpoint-14000/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..f32e7bd385184581ee545b5aee58c14b25ec56e4 --- /dev/null +++ b/checkpoint-14000/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e1158d86c311765e5f384876362976c933cae0c14a61c62dead56bd20de8de13 +size 63003985 diff --git a/checkpoint-14000/unet_ema/config.json b/checkpoint-14000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..3c312c84946eeccfbf34bac455edbff67fa98b4d --- /dev/null +++ b/checkpoint-14000/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": [ + 128, + 128, + 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", + "CrossAttnDownBlock2D", + "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": 14000, + "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", + "CrossAttnUpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "update_after_step": 0, + "use_ema_warmup": true, + "use_linear_projection": false +} diff --git a/checkpoint-14000/unet_ema/diffusion_pytorch_model.bin b/checkpoint-14000/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..5a12b13e896b051b6a8d7f4a6176a74d90f42c94 --- /dev/null +++ b/checkpoint-14000/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:177d71edd26f7d6e28f6fb3fb912a62a76d2d0d60ae7d3a8008ba3356d98929d +size 62996305 diff --git a/checkpoint-14500/optimizer.bin b/checkpoint-14500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..2a706c9567dcaa4914141044a9c25845ecf0be5c --- /dev/null +++ b/checkpoint-14500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:ce050bf66e3bdb5b98f1731291e50b16f185a0805137a80539dac8be42c6cf05 +size 126008021 diff --git a/checkpoint-14500/random_states_0.pkl b/checkpoint-14500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..22c6d7fc002f50a3e81490b201e22b0d5ee310f8 --- /dev/null +++ b/checkpoint-14500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:fe6d745c4a8ad494d76c39f412330219d077f31681699bf55271d4e4cb38d0a4 +size 14727 diff --git a/checkpoint-14500/scaler.pt b/checkpoint-14500/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..115263c9235918fc2a72413ed7e00c969b8d3527 --- /dev/null +++ b/checkpoint-14500/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7fa181fa360d46feed4180ea17c8b6a4a879a9b4231c2e91aff2be20be9076cc +size 557 diff --git a/checkpoint-14500/scheduler.bin b/checkpoint-14500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..45405963bf03f481ea843b7b1c22d4a07a9d848a --- /dev/null +++ b/checkpoint-14500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:056f1ebfe5a753f1645345b2bcfa0c57fd9aa5d0530877481e28a52f71a0db12 +size 563 diff --git a/checkpoint-14500/unet/config.json b/checkpoint-14500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..06fb387aa3ed37ab3b2f455b5469059d55604f74 --- /dev/null +++ b/checkpoint-14500/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": [ + 128, + 128, + 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", + "CrossAttnDownBlock2D", + "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", + "CrossAttnUpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "use_linear_projection": false +} diff --git a/checkpoint-14500/unet/diffusion_pytorch_model.bin b/checkpoint-14500/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..f32e7bd385184581ee545b5aee58c14b25ec56e4 --- /dev/null +++ b/checkpoint-14500/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e1158d86c311765e5f384876362976c933cae0c14a61c62dead56bd20de8de13 +size 63003985 diff --git a/checkpoint-14500/unet_ema/config.json b/checkpoint-14500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..4f6d070370bdbb7fab7e35c4e9466fa43a22b026 --- /dev/null +++ b/checkpoint-14500/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": [ + 128, + 128, + 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", + "CrossAttnDownBlock2D", + "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": 14500, + "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", + "CrossAttnUpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "update_after_step": 0, + "use_ema_warmup": true, + "use_linear_projection": false +} diff --git a/checkpoint-14500/unet_ema/diffusion_pytorch_model.bin b/checkpoint-14500/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..fe2b19af056e4c5e4c2af50f81a5eba118e076d5 --- /dev/null +++ b/checkpoint-14500/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:f0b342f7c9ac5c906d0617dfe61b6c0d78c6ea27b512f5d602bfff1745ea5ce1 +size 62996305 diff --git a/checkpoint-15000/optimizer.bin b/checkpoint-15000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..ae5267b787ba6faccb36b47797a7ea79c8cfeba2 --- /dev/null +++ b/checkpoint-15000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:1035db5e805ec1c572b17c39af9f2c3e4b076114f85da0b148d566fa55d0ffce +size 126008021 diff --git a/checkpoint-15000/random_states_0.pkl b/checkpoint-15000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..63cfe100566d6a438b9e9856b0a3955552b4b9c8 --- /dev/null +++ b/checkpoint-15000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:9e290ac7be985d6892fdd1937fe438c252f895b064d58356444716b2cbb1149d +size 14727 diff --git a/checkpoint-15000/scaler.pt b/checkpoint-15000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..115263c9235918fc2a72413ed7e00c969b8d3527 --- /dev/null +++ b/checkpoint-15000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7fa181fa360d46feed4180ea17c8b6a4a879a9b4231c2e91aff2be20be9076cc +size 557 diff --git a/checkpoint-15000/scheduler.bin b/checkpoint-15000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..79788af349cd89e77057bfa3aa270184924e0f12 --- /dev/null +++ b/checkpoint-15000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b491ed58bd960c0a4027dba67cfa65bf9a85359083725f4c6b4a9b6707264674 +size 563 diff --git a/checkpoint-15000/unet/config.json b/checkpoint-15000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..06fb387aa3ed37ab3b2f455b5469059d55604f74 --- /dev/null +++ b/checkpoint-15000/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": [ + 128, + 128, + 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", + "CrossAttnDownBlock2D", + "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", + "CrossAttnUpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "use_linear_projection": false +} diff --git a/checkpoint-15000/unet/diffusion_pytorch_model.bin b/checkpoint-15000/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..f32e7bd385184581ee545b5aee58c14b25ec56e4 --- /dev/null +++ b/checkpoint-15000/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e1158d86c311765e5f384876362976c933cae0c14a61c62dead56bd20de8de13 +size 63003985 diff --git a/checkpoint-15000/unet_ema/config.json b/checkpoint-15000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..602d033dd506823178b96222d752a9e2d94c254f --- /dev/null +++ b/checkpoint-15000/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": [ + 128, + 128, + 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", + "CrossAttnDownBlock2D", + "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": 15000, + "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", + "CrossAttnUpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "update_after_step": 0, + "use_ema_warmup": true, + "use_linear_projection": false +} diff --git a/checkpoint-15000/unet_ema/diffusion_pytorch_model.bin b/checkpoint-15000/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..4a437c84ef602d905b209ed07170610a61d0efd5 --- /dev/null +++ b/checkpoint-15000/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:86622d00d5dd7878f506d14c75992571da341b2656760622425fb83bc9628118 +size 62996305 diff --git a/checkpoint-15500/optimizer.bin b/checkpoint-15500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..16b2439f2a87b85dacf6226acb8d94b5ef5a9684 --- /dev/null +++ b/checkpoint-15500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:8aa4323d62e00aefedd0bf455c2306a4b08f8b4fa14e58b8c272c58b2da9b581 +size 126008021 diff --git a/checkpoint-15500/random_states_0.pkl b/checkpoint-15500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..712b64e4ee312d3e117ab655973f9259e1162154 --- /dev/null +++ b/checkpoint-15500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:96a000acee6e69041325544c6c40c67acc5badf1d4fa90f0197b1792eb57778b +size 14727 diff --git a/checkpoint-15500/scaler.pt b/checkpoint-15500/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..115263c9235918fc2a72413ed7e00c969b8d3527 --- /dev/null +++ b/checkpoint-15500/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7fa181fa360d46feed4180ea17c8b6a4a879a9b4231c2e91aff2be20be9076cc +size 557 diff --git a/checkpoint-15500/scheduler.bin b/checkpoint-15500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..ad334a9df8da81a2df64f15cf2183824f3577c76 --- /dev/null +++ b/checkpoint-15500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:2a57e5c0ce8d6b7c9aa7bf52acad7530813e648ef256b26a343b9d0db23638a6 +size 563 diff --git a/checkpoint-15500/unet/config.json b/checkpoint-15500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..06fb387aa3ed37ab3b2f455b5469059d55604f74 --- /dev/null +++ b/checkpoint-15500/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": [ + 128, + 128, + 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", + "CrossAttnDownBlock2D", + "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", + "CrossAttnUpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "use_linear_projection": false +} diff --git a/checkpoint-15500/unet/diffusion_pytorch_model.bin b/checkpoint-15500/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..f32e7bd385184581ee545b5aee58c14b25ec56e4 --- /dev/null +++ b/checkpoint-15500/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e1158d86c311765e5f384876362976c933cae0c14a61c62dead56bd20de8de13 +size 63003985 diff --git a/checkpoint-15500/unet_ema/config.json b/checkpoint-15500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..2e14d1559edf83bfe1b2856b34a3f94a46755d5d --- /dev/null +++ b/checkpoint-15500/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": [ + 128, + 128, + 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", + "CrossAttnDownBlock2D", + "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": 15500, + "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", + "CrossAttnUpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "update_after_step": 0, + "use_ema_warmup": true, + "use_linear_projection": false +} diff --git a/checkpoint-15500/unet_ema/diffusion_pytorch_model.bin b/checkpoint-15500/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..a3463bda0ab051cdc97c092629e3b332f326d587 --- /dev/null +++ b/checkpoint-15500/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:4248ce2c7c5ce84db553674ad69cec89f4f82d8bea2f4c6ccbb68f4c9ec0f0a9 +size 62996305 diff --git a/checkpoint-16000/optimizer.bin b/checkpoint-16000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..06395258e04114bbaac11f9761a57198ff3892ab --- /dev/null +++ b/checkpoint-16000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:6fbf55758afe8a7c578387ff411a9164c078be0a7dc087ab3f7fb3289f1668f5 +size 126008021 diff --git a/checkpoint-16000/random_states_0.pkl b/checkpoint-16000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..cea42a466e01050a96b0daf86ca79943f16ad420 --- /dev/null +++ b/checkpoint-16000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:04995f621f70b0c0c9fc13dc2382fdc24631118f21a4de92a8a1974012dcb278 +size 14727 diff --git a/checkpoint-16000/scaler.pt b/checkpoint-16000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..115263c9235918fc2a72413ed7e00c969b8d3527 --- /dev/null +++ b/checkpoint-16000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7fa181fa360d46feed4180ea17c8b6a4a879a9b4231c2e91aff2be20be9076cc +size 557 diff --git a/checkpoint-16000/scheduler.bin b/checkpoint-16000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..630f5da3f9482aa0314b6c868bcbc5d9fbdd17b7 --- /dev/null +++ b/checkpoint-16000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:af1e84f247439c9c371b3693aed94cd53a32373a3897b3d295536c855a4e429c +size 563 diff --git a/checkpoint-16000/unet/config.json b/checkpoint-16000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..06fb387aa3ed37ab3b2f455b5469059d55604f74 --- /dev/null +++ b/checkpoint-16000/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": [ + 128, + 128, + 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", + "CrossAttnDownBlock2D", + "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", + "CrossAttnUpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "use_linear_projection": false +} diff --git a/checkpoint-16000/unet/diffusion_pytorch_model.bin b/checkpoint-16000/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..f32e7bd385184581ee545b5aee58c14b25ec56e4 --- /dev/null +++ b/checkpoint-16000/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e1158d86c311765e5f384876362976c933cae0c14a61c62dead56bd20de8de13 +size 63003985 diff --git a/checkpoint-16000/unet_ema/config.json b/checkpoint-16000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..639cbf2e4623566dfb7fa605d5f85e3420d0d838 --- /dev/null +++ b/checkpoint-16000/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": [ + 128, + 128, + 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", + "CrossAttnDownBlock2D", + "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": 16000, + "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", + "CrossAttnUpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "update_after_step": 0, + "use_ema_warmup": true, + "use_linear_projection": false +} diff --git a/checkpoint-16000/unet_ema/diffusion_pytorch_model.bin b/checkpoint-16000/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..b7747c4681b4a721687206c3196e214441a34e95 --- /dev/null +++ b/checkpoint-16000/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7a94627a23655b396d4c79e02571f2b368555d9b0f323952a66db73758a0c2bc +size 62996305 diff --git a/checkpoint-16500/optimizer.bin b/checkpoint-16500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..313d552235784146e32ec81984ced3286e06d8dc --- /dev/null +++ b/checkpoint-16500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:02c2d6c96736dce5f839aeed5df2ad1ac0c6a781b78e42888fe04493d7db47f6 +size 126008021 diff --git a/checkpoint-16500/random_states_0.pkl b/checkpoint-16500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..410d45f82d1e4b2252f8f3bdd2f4ea61388fec20 --- /dev/null +++ b/checkpoint-16500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:da5d380451254ff7738c4bd07e6e03a1f91464b1a9080fd655dc71f87f10dcc0 +size 14727 diff --git a/checkpoint-16500/scaler.pt b/checkpoint-16500/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..115263c9235918fc2a72413ed7e00c969b8d3527 --- /dev/null +++ b/checkpoint-16500/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7fa181fa360d46feed4180ea17c8b6a4a879a9b4231c2e91aff2be20be9076cc +size 557 diff --git a/checkpoint-16500/scheduler.bin b/checkpoint-16500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..54906ddfe8aafe948e7a7d83fa5468f62b1f7bf3 --- /dev/null +++ b/checkpoint-16500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b076b0fc8d9d24423b94d5b5702924560e5903ff30a3fa5fb06bd22584d3d87a +size 563 diff --git a/checkpoint-16500/unet/config.json b/checkpoint-16500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..06fb387aa3ed37ab3b2f455b5469059d55604f74 --- /dev/null +++ b/checkpoint-16500/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": [ + 128, + 128, + 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", + "CrossAttnDownBlock2D", + "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", + "CrossAttnUpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "use_linear_projection": false +} diff --git a/checkpoint-16500/unet/diffusion_pytorch_model.bin b/checkpoint-16500/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..f32e7bd385184581ee545b5aee58c14b25ec56e4 --- /dev/null +++ b/checkpoint-16500/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e1158d86c311765e5f384876362976c933cae0c14a61c62dead56bd20de8de13 +size 63003985 diff --git a/checkpoint-16500/unet_ema/config.json b/checkpoint-16500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..f26147d04e6c2735d392d1dd404d94b92ae0191a --- /dev/null +++ b/checkpoint-16500/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": [ + 128, + 128, + 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", + "CrossAttnDownBlock2D", + "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": 16500, + "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", + "CrossAttnUpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "update_after_step": 0, + "use_ema_warmup": true, + "use_linear_projection": false +} diff --git a/checkpoint-16500/unet_ema/diffusion_pytorch_model.bin b/checkpoint-16500/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..b0f8e5ee02bd64dd8814792623d6783a87a27eea --- /dev/null +++ b/checkpoint-16500/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:5b83a1c5158331f13948e2cdab0a225cacb45e651c61fb6ddfaecd9cefaf09bd +size 62996305 diff --git a/checkpoint-17000/optimizer.bin b/checkpoint-17000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..bca28114cc158d6e985a5b3701d0e5db2ab14c4e --- /dev/null +++ b/checkpoint-17000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:f37808f5414c749257da447313c814a0c1e71fa1efbc1baef55fa7b9a7068e9c +size 126008021 diff --git a/checkpoint-17000/random_states_0.pkl b/checkpoint-17000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..1bae21bd4709937a72087c4e35b5bc167f3a07d4 --- /dev/null +++ b/checkpoint-17000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:ba9f2e2306e676502edfd3df5de04817ae352a4326ff37226903f3db76da9460 +size 14727 diff --git a/checkpoint-17000/scaler.pt b/checkpoint-17000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..115263c9235918fc2a72413ed7e00c969b8d3527 --- /dev/null +++ b/checkpoint-17000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7fa181fa360d46feed4180ea17c8b6a4a879a9b4231c2e91aff2be20be9076cc +size 557 diff --git a/checkpoint-17000/scheduler.bin b/checkpoint-17000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..96cb96e0e0396895653c449aa99fa83633d790a3 --- /dev/null +++ b/checkpoint-17000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:a40ec82b5c45f4ee5c8dc230eb737c558ec99ebd64e39416666449be37526e14 +size 563 diff --git a/checkpoint-17000/unet/config.json b/checkpoint-17000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..06fb387aa3ed37ab3b2f455b5469059d55604f74 --- /dev/null +++ b/checkpoint-17000/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": [ + 128, + 128, + 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", + "CrossAttnDownBlock2D", + "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", + "CrossAttnUpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "use_linear_projection": false +} diff --git a/checkpoint-17000/unet/diffusion_pytorch_model.bin b/checkpoint-17000/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..f32e7bd385184581ee545b5aee58c14b25ec56e4 --- /dev/null +++ b/checkpoint-17000/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e1158d86c311765e5f384876362976c933cae0c14a61c62dead56bd20de8de13 +size 63003985 diff --git a/checkpoint-17000/unet_ema/config.json b/checkpoint-17000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..e5ce7b2b2fcf5d46085339bbdae561e39c583243 --- /dev/null +++ b/checkpoint-17000/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": [ + 128, + 128, + 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", + "CrossAttnDownBlock2D", + "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": 17000, + "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", + "CrossAttnUpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "update_after_step": 0, + "use_ema_warmup": true, + "use_linear_projection": false +} diff --git a/checkpoint-17000/unet_ema/diffusion_pytorch_model.bin b/checkpoint-17000/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..10cca23fcdfcdf23e8cf75a097297d4507ef5e1f --- /dev/null +++ b/checkpoint-17000/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:f89321b5d1f41f6eac521b26841c593ea870917ff7db3aa98cd5316e77a071e5 +size 62996305 diff --git a/checkpoint-9000/optimizer.bin b/checkpoint-9000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..a44f32fe0ab814aa3333a876d76f84b0beacb1ec --- /dev/null +++ b/checkpoint-9000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:44dc96d79efe0f038357c08db924c18e5364258b898b827f5060b576851b62ee +size 126008021 diff --git a/checkpoint-9000/random_states_0.pkl b/checkpoint-9000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..1e4c88a9fb3690b45ab641894a4d7324f19c8a33 --- /dev/null +++ b/checkpoint-9000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:21b4858c34af197cf44a820d152088a8182f22405dff83420af8458d61d797ba +size 14727 diff --git a/checkpoint-9000/scaler.pt b/checkpoint-9000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..d9f22b7f3aadb66d36dcc722508b1ee1bc9a19b1 --- /dev/null +++ b/checkpoint-9000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:cb5af6a8533692dccbada4e8e11fd003fac3e659f0a93507f5e8f7e606598bac +size 557 diff --git a/checkpoint-9000/scheduler.bin b/checkpoint-9000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..cbfaad0e56ff958810d96165ff272e082a85fd41 --- /dev/null +++ b/checkpoint-9000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:0acf4b62089ddc02dab9c802b4d47c4c97d57060b84e8d2a17286d98b1e6e0a3 +size 563 diff --git a/checkpoint-9000/unet/config.json b/checkpoint-9000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..06fb387aa3ed37ab3b2f455b5469059d55604f74 --- /dev/null +++ b/checkpoint-9000/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": [ + 128, + 128, + 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", + "CrossAttnDownBlock2D", + "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", + "CrossAttnUpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "use_linear_projection": false +} diff --git a/checkpoint-9000/unet/diffusion_pytorch_model.bin b/checkpoint-9000/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..18c30b61d3ba86291963637de1051a96a38bb77f --- /dev/null +++ b/checkpoint-9000/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:14ec92a06082abce843e2214f6bb011dec58452bf0172aa8181785cf35f3c1e0 +size 63003985 diff --git a/checkpoint-9000/unet_ema/config.json b/checkpoint-9000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..ff37d58f50d10efc09808b0690dc98de5c5a10dc --- /dev/null +++ b/checkpoint-9000/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": [ + 128, + 128, + 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", + "CrossAttnDownBlock2D", + "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": 9000, + "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", + "CrossAttnUpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "update_after_step": 0, + "use_ema_warmup": true, + "use_linear_projection": false +} diff --git a/checkpoint-9000/unet_ema/diffusion_pytorch_model.bin b/checkpoint-9000/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..d6a7f573729033198d712254c3fe190bcb496bfa --- /dev/null +++ b/checkpoint-9000/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:6c74cbf655222af4ff6e042422eac2cf3343904fef70e1df300742d14b586557 +size 62996305 diff --git a/checkpoint-9500/optimizer.bin b/checkpoint-9500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..03d1256e8db1e18850a8e681309dd6a2b4fa59fe --- /dev/null +++ b/checkpoint-9500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:fd1ff5e5e2aac7e35773585861bd79da34464be94678d2aa961ce41fa4f3b5ed +size 126008021 diff --git a/checkpoint-9500/random_states_0.pkl b/checkpoint-9500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..83fce315ca1b2df0b253bfe01f7194ab15fb597a --- /dev/null +++ b/checkpoint-9500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:fa1e655aea15ded5559a0e54b94bfa62a9fcf9d4b3e196d92fc1e731def65b35 +size 14727 diff --git a/checkpoint-9500/scaler.pt b/checkpoint-9500/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..d04cf5522f1cdef73e9958301f4f42635e22ca34 --- /dev/null +++ b/checkpoint-9500/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:763cb6a41f7a8c2b155859fc47318680209a651148d2fb01153339609fe49c03 +size 557 diff --git a/checkpoint-9500/scheduler.bin b/checkpoint-9500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..90c3b2988105c5c4771f098f09492e5f9dfe4722 --- /dev/null +++ b/checkpoint-9500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:59f3941b9d2918a3bf41ab7fdaee50e6598ff8a08264358d2fb92c28c48dbe59 +size 563 diff --git a/checkpoint-9500/unet/config.json b/checkpoint-9500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..06fb387aa3ed37ab3b2f455b5469059d55604f74 --- /dev/null +++ b/checkpoint-9500/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": [ + 128, + 128, + 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", + "CrossAttnDownBlock2D", + "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", + "CrossAttnUpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "use_linear_projection": false +} diff --git a/checkpoint-9500/unet/diffusion_pytorch_model.bin b/checkpoint-9500/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..4e7d5f70da2a48eb89b04d8d07fc1e2605f2d36d --- /dev/null +++ b/checkpoint-9500/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c1570889ad89a70230da41a61642331ded0d8f5169f6a783301bae8738f6bc11 +size 63003985 diff --git a/checkpoint-9500/unet_ema/config.json b/checkpoint-9500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..69b80e09d16e34f312889fe35d60369f8f889024 --- /dev/null +++ b/checkpoint-9500/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": [ + 128, + 128, + 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", + "CrossAttnDownBlock2D", + "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": 9500, + "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", + "CrossAttnUpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "update_after_step": 0, + "use_ema_warmup": true, + "use_linear_projection": false +} diff --git a/checkpoint-9500/unet_ema/diffusion_pytorch_model.bin b/checkpoint-9500/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..340fcd2727df51f62a0cf7f20614c45b662ae100 --- /dev/null +++ b/checkpoint-9500/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:a75b2d99e35c120d4750e1edf5182f40469d984eacb8878991b4111ed08358bc +size 62996305 diff --git a/unet/diffusion_pytorch_model.bin b/unet/diffusion_pytorch_model.bin index 2c36f70c662ae2e7b46abd9711ba02ab1212f288..7d327f1d5ebfcd4cd59dab3de5aaf3c33328d5f8 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:db21086d8ba98d423e52b859c23e3f4aa085540c85ca2b4ade6be5d1b9e8e83c +oid sha256:bdf15c4d535d7d266c0ea6b250c6e62f533932d59d47749d24adb2257af7b497 size 63003985