diff --git a/checkpoint-17500/optimizer.bin b/checkpoint-17500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..bd663ef6f2f69f22ead5ed24432e37c1a7871462 --- /dev/null +++ b/checkpoint-17500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:4c6159e367ea65fe273be4379ef7d709880f1da225dc18d8ac79be17809dff53 +size 126008021 diff --git a/checkpoint-17500/random_states_0.pkl b/checkpoint-17500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..3940ddf074b1d8e87eafc92cec69fab12a626cf5 --- /dev/null +++ b/checkpoint-17500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:217a116592cbd819040ebe09c9039d99e347b2bd68fda9b70fd0aa47bcdcdea6 +size 14727 diff --git a/checkpoint-17500/scaler.pt b/checkpoint-17500/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..115263c9235918fc2a72413ed7e00c969b8d3527 --- /dev/null +++ b/checkpoint-17500/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7fa181fa360d46feed4180ea17c8b6a4a879a9b4231c2e91aff2be20be9076cc +size 557 diff --git a/checkpoint-17500/scheduler.bin b/checkpoint-17500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..8cfdb4775f767f21a385de2b42c5b50ee1603710 --- /dev/null +++ b/checkpoint-17500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:56bd85fc35d6c41a48b203c61c970fdc31c26e77c655a95993bf0b0ac2668798 +size 563 diff --git a/checkpoint-17500/unet/config.json b/checkpoint-17500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..06fb387aa3ed37ab3b2f455b5469059d55604f74 --- /dev/null +++ b/checkpoint-17500/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-17500/unet/diffusion_pytorch_model.bin b/checkpoint-17500/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..f32e7bd385184581ee545b5aee58c14b25ec56e4 --- /dev/null +++ b/checkpoint-17500/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-17500/unet_ema/config.json b/checkpoint-17500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..7219820f119d7e4427fe0837a73fd92280f3c30f --- /dev/null +++ b/checkpoint-17500/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": 17500, + "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-17500/unet_ema/diffusion_pytorch_model.bin b/checkpoint-17500/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..656239859cbb369b7c0c9305a0256ee801691067 --- /dev/null +++ b/checkpoint-17500/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:661e062d61bfd4f0de88e94592a36039c09df57bd5cab9544e7c0ebd29f5c619 +size 62996305 diff --git a/checkpoint-18000/optimizer.bin b/checkpoint-18000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..4050cc90b03c099e92282212894ba0056a763f08 --- /dev/null +++ b/checkpoint-18000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:ebc62af8f79ff8bd9892ff9868000dbe56f6647dc82445491065a09dd4b3031f +size 126008021 diff --git a/checkpoint-18000/random_states_0.pkl b/checkpoint-18000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..bf3b0c30b604195a1f20e8ec3b1e681b27e12f1a --- /dev/null +++ b/checkpoint-18000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c4cbe3ec3d55d7a572306d41d13274a620979ed780a777df2ea224b4a8d33296 +size 14727 diff --git a/checkpoint-18000/scaler.pt b/checkpoint-18000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..115263c9235918fc2a72413ed7e00c969b8d3527 --- /dev/null +++ b/checkpoint-18000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7fa181fa360d46feed4180ea17c8b6a4a879a9b4231c2e91aff2be20be9076cc +size 557 diff --git a/checkpoint-18000/scheduler.bin b/checkpoint-18000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..ebb98457d531180b5ec88fd075706917e00cd847 --- /dev/null +++ b/checkpoint-18000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:6b9957baa004fc7f5206837f1c381c017cd92aa0288ebef511b6eb14a0a9eb77 +size 563 diff --git a/checkpoint-18000/unet/config.json b/checkpoint-18000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..06fb387aa3ed37ab3b2f455b5469059d55604f74 --- /dev/null +++ b/checkpoint-18000/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-18000/unet/diffusion_pytorch_model.bin b/checkpoint-18000/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..f32e7bd385184581ee545b5aee58c14b25ec56e4 --- /dev/null +++ b/checkpoint-18000/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-18000/unet_ema/config.json b/checkpoint-18000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..ac003598f6c06fabece762a32fd53bbaae27df62 --- /dev/null +++ b/checkpoint-18000/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": 18000, + "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-18000/unet_ema/diffusion_pytorch_model.bin b/checkpoint-18000/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..5edf8de299438584466017a7126e11379706e221 --- /dev/null +++ b/checkpoint-18000/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:54baf3b746e755709724f1272167c604e41373321d4d256959520898247c0f5d +size 62996305 diff --git a/checkpoint-18500/optimizer.bin b/checkpoint-18500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..13603f71cf3caf9a3b70b3382e59c57ce7c83496 --- /dev/null +++ b/checkpoint-18500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:fc88026a1b4e1afae1fed86b43d41798b597f301b3ec1e3964e977859226cb79 +size 126008021 diff --git a/checkpoint-18500/random_states_0.pkl b/checkpoint-18500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..31240b5f2086d73b8b592bc875ce6c13b7e6c33a --- /dev/null +++ b/checkpoint-18500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:d0c6921c593bfb58d2c8b022600a3f611c89834875e8b256e702fc9f76c09fdb +size 14727 diff --git a/checkpoint-18500/scaler.pt b/checkpoint-18500/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..115263c9235918fc2a72413ed7e00c969b8d3527 --- /dev/null +++ b/checkpoint-18500/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7fa181fa360d46feed4180ea17c8b6a4a879a9b4231c2e91aff2be20be9076cc +size 557 diff --git a/checkpoint-18500/scheduler.bin b/checkpoint-18500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..a10fb684f486520e47cb52476c1f53d852d7c8f8 --- /dev/null +++ b/checkpoint-18500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:8701449110ea9af39c1f8fed8852fd154eb699d5d4858770f07a99b3f2be3d96 +size 563 diff --git a/checkpoint-18500/unet/config.json b/checkpoint-18500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..06fb387aa3ed37ab3b2f455b5469059d55604f74 --- /dev/null +++ b/checkpoint-18500/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-18500/unet/diffusion_pytorch_model.bin b/checkpoint-18500/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..f32e7bd385184581ee545b5aee58c14b25ec56e4 --- /dev/null +++ b/checkpoint-18500/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-18500/unet_ema/config.json b/checkpoint-18500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..f4d1c24e8a1402b4dd59a61ca982f8e08a459cdf --- /dev/null +++ b/checkpoint-18500/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": 18500, + "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-18500/unet_ema/diffusion_pytorch_model.bin b/checkpoint-18500/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..acfefa86a153772caed973c8cc4d8e684f8236ff --- /dev/null +++ b/checkpoint-18500/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:5f0c32e0bc73d3ea7829ca14f3b31fe7a011f13398cbd5abd08129cf48ae6e30 +size 62996305 diff --git a/checkpoint-19000/optimizer.bin b/checkpoint-19000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..d3e83b4198a763bafbed15b040f0d7a64f96eafe --- /dev/null +++ b/checkpoint-19000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:206e183ca96e76bfc03dde1be6a9de4a81c9f31bc5f08043f6287487b968df86 +size 126008021 diff --git a/checkpoint-19000/random_states_0.pkl b/checkpoint-19000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..9f03d45f72986cde160be5b83a5f8b1eeb701bb9 --- /dev/null +++ b/checkpoint-19000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:236896f5ca496218ce4c04a16bad58fea6d7f3b834500e9bdb8dcfc939d5dcdd +size 14727 diff --git a/checkpoint-19000/scaler.pt b/checkpoint-19000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..115263c9235918fc2a72413ed7e00c969b8d3527 --- /dev/null +++ b/checkpoint-19000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7fa181fa360d46feed4180ea17c8b6a4a879a9b4231c2e91aff2be20be9076cc +size 557 diff --git a/checkpoint-19000/scheduler.bin b/checkpoint-19000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..e1eff774be097974e62181901540ca6bdade1824 --- /dev/null +++ b/checkpoint-19000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:0007f0636029b6d89afd70838fd442d61960d536b4a3e683b6bd6eec76443bf6 +size 563 diff --git a/checkpoint-19000/unet/config.json b/checkpoint-19000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..06fb387aa3ed37ab3b2f455b5469059d55604f74 --- /dev/null +++ b/checkpoint-19000/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-19000/unet/diffusion_pytorch_model.bin b/checkpoint-19000/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..f32e7bd385184581ee545b5aee58c14b25ec56e4 --- /dev/null +++ b/checkpoint-19000/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-19000/unet_ema/config.json b/checkpoint-19000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..df98b8a071701a9ded3dd3165aad6546ea4cba93 --- /dev/null +++ b/checkpoint-19000/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": 19000, + "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-19000/unet_ema/diffusion_pytorch_model.bin b/checkpoint-19000/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..6351ec73ff30e822911444aabfda80d96a77eb76 --- /dev/null +++ b/checkpoint-19000/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:ea7471a8fff556b04d0f49b719257b8a5058c3066d3c50809a4ca2f4ccb98316 +size 62996305 diff --git a/checkpoint-19500/optimizer.bin b/checkpoint-19500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..7c93063e9beb5ad355676d11656a6f952236a9ba --- /dev/null +++ b/checkpoint-19500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:68c5136ffd2e83c66be430c08dba66ea40abe2299d0e33c6a6236415358c76ba +size 126008021 diff --git a/checkpoint-19500/random_states_0.pkl b/checkpoint-19500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..8ca45fc50c5294b03a17e5a2fe81114d55ae9632 --- /dev/null +++ b/checkpoint-19500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:661ef46a28974d67b0b80b3a0edf1fb9cea59620daba4cb1208b4d3bf4906988 +size 14727 diff --git a/checkpoint-19500/scaler.pt b/checkpoint-19500/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..115263c9235918fc2a72413ed7e00c969b8d3527 --- /dev/null +++ b/checkpoint-19500/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7fa181fa360d46feed4180ea17c8b6a4a879a9b4231c2e91aff2be20be9076cc +size 557 diff --git a/checkpoint-19500/scheduler.bin b/checkpoint-19500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..a06e9e45f8a770252eb71b5ab15d590cbdaf0d45 --- /dev/null +++ b/checkpoint-19500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:727b52903b6abc4620b760d3c2c55b78d02626357412298bd949683b3ea4683c +size 563 diff --git a/checkpoint-19500/unet/config.json b/checkpoint-19500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..06fb387aa3ed37ab3b2f455b5469059d55604f74 --- /dev/null +++ b/checkpoint-19500/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-19500/unet/diffusion_pytorch_model.bin b/checkpoint-19500/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..f32e7bd385184581ee545b5aee58c14b25ec56e4 --- /dev/null +++ b/checkpoint-19500/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-19500/unet_ema/config.json b/checkpoint-19500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..9f3b2207d2f33bc8d759b24f49d90b54fae31b58 --- /dev/null +++ b/checkpoint-19500/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": 19500, + "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-19500/unet_ema/diffusion_pytorch_model.bin b/checkpoint-19500/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..91553d52dfc86dc319e65dd39bdc1e1399a01421 --- /dev/null +++ b/checkpoint-19500/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:ed081a12708cd41ad805a34bede412a19794b4aacf5f15cd0138bd3f0e12d955 +size 62996305 diff --git a/checkpoint-20000/optimizer.bin b/checkpoint-20000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..627b340c60775ce4f787306fd51e2b7690521a91 --- /dev/null +++ b/checkpoint-20000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:27134fb2b70776157d690d065faa1fa3bd1ef7a54d2546420db30e0c744e64a3 +size 126008021 diff --git a/checkpoint-20000/random_states_0.pkl b/checkpoint-20000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..f108588cc05ca037428d62fd24faf916aaaa210f --- /dev/null +++ b/checkpoint-20000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:9c71a4b70cbcc1bfd16e007548a0e14589f5f30bcc4ee8702067ef1c1b63cbb7 +size 14727 diff --git a/checkpoint-20000/scaler.pt b/checkpoint-20000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..115263c9235918fc2a72413ed7e00c969b8d3527 --- /dev/null +++ b/checkpoint-20000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7fa181fa360d46feed4180ea17c8b6a4a879a9b4231c2e91aff2be20be9076cc +size 557 diff --git a/checkpoint-20000/scheduler.bin b/checkpoint-20000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..9c1b9d0c1f02532aca485df2d4f5c537b08c2e4c --- /dev/null +++ b/checkpoint-20000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:635c78d1f1d3235d7b6f333fed9a14c8415cd326212ae1c901a1d3f472e4aaae +size 563 diff --git a/checkpoint-20000/unet/config.json b/checkpoint-20000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..06fb387aa3ed37ab3b2f455b5469059d55604f74 --- /dev/null +++ b/checkpoint-20000/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-20000/unet/diffusion_pytorch_model.bin b/checkpoint-20000/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..f32e7bd385184581ee545b5aee58c14b25ec56e4 --- /dev/null +++ b/checkpoint-20000/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-20000/unet_ema/config.json b/checkpoint-20000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..e2bd912b910561d1ae71b94ef5488684773713e3 --- /dev/null +++ b/checkpoint-20000/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": 20000, + "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-20000/unet_ema/diffusion_pytorch_model.bin b/checkpoint-20000/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..d8c55ae1cd5cdb5551c2af605731265299ad7f5d --- /dev/null +++ b/checkpoint-20000/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e53b711b13f393654590356177bb4b2d9c548bf0083d4c9ffbbd6928e21e835c +size 62996305 diff --git a/checkpoint-20500/optimizer.bin b/checkpoint-20500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..31b6e8b8bd543748c4aee3a21d71f13d50144627 --- /dev/null +++ b/checkpoint-20500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:588b08c216cc1eaa75a727a717b34cf7473311cdba243f88c7e028637857c60a +size 126008021 diff --git a/checkpoint-20500/random_states_0.pkl b/checkpoint-20500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..b569c97d7f19f3c7b1bb53f626b3725ecb4ad43d --- /dev/null +++ b/checkpoint-20500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:20095668e870decd919a00b88913bd00437575ad7e58fa205cf9ea07527df9bc +size 14727 diff --git a/checkpoint-20500/scaler.pt b/checkpoint-20500/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..115263c9235918fc2a72413ed7e00c969b8d3527 --- /dev/null +++ b/checkpoint-20500/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7fa181fa360d46feed4180ea17c8b6a4a879a9b4231c2e91aff2be20be9076cc +size 557 diff --git a/checkpoint-20500/scheduler.bin b/checkpoint-20500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..90aae6ca4341ac840df7faad889c130a30974731 --- /dev/null +++ b/checkpoint-20500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:34f023f9bb73d771e46eda124c4d8d38201a0b956e7d208e2ba3af423c2df7e8 +size 563 diff --git a/checkpoint-20500/unet/config.json b/checkpoint-20500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..06fb387aa3ed37ab3b2f455b5469059d55604f74 --- /dev/null +++ b/checkpoint-20500/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-20500/unet/diffusion_pytorch_model.bin b/checkpoint-20500/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..f32e7bd385184581ee545b5aee58c14b25ec56e4 --- /dev/null +++ b/checkpoint-20500/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-20500/unet_ema/config.json b/checkpoint-20500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..bd01bd82954ed19e3032e504ce9bf54bc84efad9 --- /dev/null +++ b/checkpoint-20500/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": 20500, + "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-20500/unet_ema/diffusion_pytorch_model.bin b/checkpoint-20500/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..c8f8e3e00c28a4affd1f1c9d2f1539a2e65f527d --- /dev/null +++ b/checkpoint-20500/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:295b5bc80b7c3e9f8acffea9f312eb79541afd63f3f9ccecb3cb52b8120b111d +size 62996305 diff --git a/checkpoint-21000/optimizer.bin b/checkpoint-21000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..3481d89b3a81e78cd63beacd1b23e7b18f041773 --- /dev/null +++ b/checkpoint-21000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:01b9bfe75379911306bbe58b30d28ca37eac46892b3172f50d39cf90e0d56d02 +size 126008021 diff --git a/checkpoint-21000/random_states_0.pkl b/checkpoint-21000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..8755bb497ffe3d62f3b90eb5b1064851f4f1d2cb --- /dev/null +++ b/checkpoint-21000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:36da4a4f144ff6c50528725c99f5de8b074fe6cbd7a3d853099d1075079cff16 +size 14727 diff --git a/checkpoint-21000/scaler.pt b/checkpoint-21000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..115263c9235918fc2a72413ed7e00c969b8d3527 --- /dev/null +++ b/checkpoint-21000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7fa181fa360d46feed4180ea17c8b6a4a879a9b4231c2e91aff2be20be9076cc +size 557 diff --git a/checkpoint-21000/scheduler.bin b/checkpoint-21000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..39807684d03a86de6a2531ce8777114c6bb80aca --- /dev/null +++ b/checkpoint-21000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:210fc7e639f5b3c55aeb0f0f3a3e4c269ce92e499666774b98a46ab8d1bb7706 +size 563 diff --git a/checkpoint-21000/unet/config.json b/checkpoint-21000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..06fb387aa3ed37ab3b2f455b5469059d55604f74 --- /dev/null +++ b/checkpoint-21000/unet/config.json @@ -0,0 +1,60 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "attention_head_dim": 8, + "block_out_channels": [ + 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-21000/unet/diffusion_pytorch_model.bin b/checkpoint-21000/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..f32e7bd385184581ee545b5aee58c14b25ec56e4 --- /dev/null +++ b/checkpoint-21000/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-21000/unet_ema/config.json b/checkpoint-21000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..200df7c826f4bac9c3c7b88975ed186f55840c9a --- /dev/null +++ b/checkpoint-21000/unet_ema/config.json @@ -0,0 +1,67 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "attention_head_dim": 8, + "block_out_channels": [ + 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": 21000, + "out_channels": 3, + "power": 0.75, + "projection_class_embeddings_input_dim": null, + "resnet_out_scale_factor": 1.0, + "resnet_skip_time_act": false, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_cond_proj_dim": null, + "time_embedding_act_fn": null, + "time_embedding_dim": null, + "time_embedding_type": "positional", + "timestep_post_act": null, + "up_block_types": [ + "UpBlock2D", + "CrossAttnUpBlock2D", + "CrossAttnUpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "update_after_step": 0, + "use_ema_warmup": true, + "use_linear_projection": false +} diff --git a/checkpoint-21000/unet_ema/diffusion_pytorch_model.bin b/checkpoint-21000/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..8bd29c7f39e908bf6f60b340233ec31257e33dd2 --- /dev/null +++ b/checkpoint-21000/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:3bec0f46593ea03fcfee98f6ee037676dc980c7604b0f79213c298726fdabfed +size 62996305 diff --git a/checkpoint-21500/optimizer.bin b/checkpoint-21500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..9e5f4b16525814f4d83c45d1e2d7ef9280a44ab9 --- /dev/null +++ b/checkpoint-21500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b8883a29a61f6e13bd272588334e9f81ea2cf52a1f49995b0ab58d82dd46c50e +size 126008021 diff --git a/checkpoint-21500/random_states_0.pkl b/checkpoint-21500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..a59db8573465fa125538b7c81eee43fa4b0eda6e --- /dev/null +++ b/checkpoint-21500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:cf125671b51101a73d057690c5ab405555c035d40458f0a2dd5d957cd40fe1a6 +size 14727 diff --git a/checkpoint-21500/scaler.pt b/checkpoint-21500/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..115263c9235918fc2a72413ed7e00c969b8d3527 --- /dev/null +++ b/checkpoint-21500/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7fa181fa360d46feed4180ea17c8b6a4a879a9b4231c2e91aff2be20be9076cc +size 557 diff --git a/checkpoint-21500/scheduler.bin b/checkpoint-21500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..24fd0320b2f473a56c180d98e9cb9445646aa9ce --- /dev/null +++ b/checkpoint-21500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:683c4d595189439f275ff86902463a3887854c443120795674db59ab836a0bbc +size 563 diff --git a/checkpoint-21500/unet/config.json b/checkpoint-21500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..06fb387aa3ed37ab3b2f455b5469059d55604f74 --- /dev/null +++ b/checkpoint-21500/unet/config.json @@ -0,0 +1,60 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "attention_head_dim": 8, + "block_out_channels": [ + 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-21500/unet/diffusion_pytorch_model.bin b/checkpoint-21500/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..f32e7bd385184581ee545b5aee58c14b25ec56e4 --- /dev/null +++ b/checkpoint-21500/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-21500/unet_ema/config.json b/checkpoint-21500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..27765ac77294fed60e6d33aefb29f60a7574c08c --- /dev/null +++ b/checkpoint-21500/unet_ema/config.json @@ -0,0 +1,67 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "attention_head_dim": 8, + "block_out_channels": [ + 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": 21500, + "out_channels": 3, + "power": 0.75, + "projection_class_embeddings_input_dim": null, + "resnet_out_scale_factor": 1.0, + "resnet_skip_time_act": false, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_cond_proj_dim": null, + "time_embedding_act_fn": null, + "time_embedding_dim": null, + "time_embedding_type": "positional", + "timestep_post_act": null, + "up_block_types": [ + "UpBlock2D", + "CrossAttnUpBlock2D", + "CrossAttnUpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "update_after_step": 0, + "use_ema_warmup": true, + "use_linear_projection": false +} diff --git a/checkpoint-21500/unet_ema/diffusion_pytorch_model.bin b/checkpoint-21500/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..434ed217a3adf9e72020ca3ec5abaed88e098928 --- /dev/null +++ b/checkpoint-21500/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:5ddc8f999650c0e88955bfb5f9c097eac1b1f2dd74c9ed0a4ab80ef159359c34 +size 62996305 diff --git a/checkpoint-22000/optimizer.bin b/checkpoint-22000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..a024285d4bc5f9aac3b7359a4ccfc511fda59a78 --- /dev/null +++ b/checkpoint-22000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:85ed69436dcf135a9b1c2459bec37d093e6061ce74bfb280091cbba447e38da3 +size 126008021 diff --git a/checkpoint-22000/random_states_0.pkl b/checkpoint-22000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..0dc3207a11dcadd13faafbdfa5721b3a5e728a55 --- /dev/null +++ b/checkpoint-22000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:485c67e14c9f71d4461e7bab1413f5979444b74e15f93477ef8d6f6906822161 +size 14727 diff --git a/checkpoint-22000/scaler.pt b/checkpoint-22000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..115263c9235918fc2a72413ed7e00c969b8d3527 --- /dev/null +++ b/checkpoint-22000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7fa181fa360d46feed4180ea17c8b6a4a879a9b4231c2e91aff2be20be9076cc +size 557 diff --git a/checkpoint-22000/scheduler.bin b/checkpoint-22000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..b89f5a6d1196dfb98939a04a95c760bacff676df --- /dev/null +++ b/checkpoint-22000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:496d41b4b5c599fc4239de384252147eb24ebb599f1b70147b67145c6d605052 +size 563 diff --git a/checkpoint-22000/unet/config.json b/checkpoint-22000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..06fb387aa3ed37ab3b2f455b5469059d55604f74 --- /dev/null +++ b/checkpoint-22000/unet/config.json @@ -0,0 +1,60 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "attention_head_dim": 8, + "block_out_channels": [ + 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-22000/unet/diffusion_pytorch_model.bin b/checkpoint-22000/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..f32e7bd385184581ee545b5aee58c14b25ec56e4 --- /dev/null +++ b/checkpoint-22000/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-22000/unet_ema/config.json b/checkpoint-22000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..c08fed1deeb071bba72b673b4b43f8318180f20a --- /dev/null +++ b/checkpoint-22000/unet_ema/config.json @@ -0,0 +1,67 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "attention_head_dim": 8, + "block_out_channels": [ + 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": 22000, + "out_channels": 3, + "power": 0.75, + "projection_class_embeddings_input_dim": null, + "resnet_out_scale_factor": 1.0, + "resnet_skip_time_act": false, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_cond_proj_dim": null, + "time_embedding_act_fn": null, + "time_embedding_dim": null, + "time_embedding_type": "positional", + "timestep_post_act": null, + "up_block_types": [ + "UpBlock2D", + "CrossAttnUpBlock2D", + "CrossAttnUpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "update_after_step": 0, + "use_ema_warmup": true, + "use_linear_projection": false +} diff --git a/checkpoint-22000/unet_ema/diffusion_pytorch_model.bin b/checkpoint-22000/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..cb458861cc8d40dc14223eeba3c90c40bc5cdd63 --- /dev/null +++ b/checkpoint-22000/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:84f8ebf8a401c04ca4b37073181b9520f64182d0fbd23623b3daf7460214067b +size 62996305 diff --git a/checkpoint-22500/optimizer.bin b/checkpoint-22500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..71096d06a6ceae1ed9adbbb69e549d469d101670 --- /dev/null +++ b/checkpoint-22500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:27bcdc9d97d8828709c854908779a255acf46b12687d00a705c861aa85de3414 +size 126008021 diff --git a/checkpoint-22500/random_states_0.pkl b/checkpoint-22500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..63450518f312c38f0d27e9ac23ccf29290d65f56 --- /dev/null +++ b/checkpoint-22500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7f46dd24f7771c66828639bb40c08b56193539decc390bafb89115d6a6ed3272 +size 14727 diff --git a/checkpoint-22500/scaler.pt b/checkpoint-22500/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..115263c9235918fc2a72413ed7e00c969b8d3527 --- /dev/null +++ b/checkpoint-22500/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7fa181fa360d46feed4180ea17c8b6a4a879a9b4231c2e91aff2be20be9076cc +size 557 diff --git a/checkpoint-22500/scheduler.bin b/checkpoint-22500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..bdf704a5a36b8f8627ec07d46f3864635d2d38e9 --- /dev/null +++ b/checkpoint-22500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:74e135e3c50decb4f1b49c20818a6588d2aebe6e1ac5c390422429406b45f922 +size 563 diff --git a/checkpoint-22500/unet/config.json b/checkpoint-22500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..06fb387aa3ed37ab3b2f455b5469059d55604f74 --- /dev/null +++ b/checkpoint-22500/unet/config.json @@ -0,0 +1,60 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "attention_head_dim": 8, + "block_out_channels": [ + 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-22500/unet/diffusion_pytorch_model.bin b/checkpoint-22500/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..f32e7bd385184581ee545b5aee58c14b25ec56e4 --- /dev/null +++ b/checkpoint-22500/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-22500/unet_ema/config.json b/checkpoint-22500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..acc7650848723e97d2a318c103d22069a9a8cfc6 --- /dev/null +++ b/checkpoint-22500/unet_ema/config.json @@ -0,0 +1,67 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "attention_head_dim": 8, + "block_out_channels": [ + 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": 22500, + "out_channels": 3, + "power": 0.75, + "projection_class_embeddings_input_dim": null, + "resnet_out_scale_factor": 1.0, + "resnet_skip_time_act": false, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_cond_proj_dim": null, + "time_embedding_act_fn": null, + "time_embedding_dim": null, + "time_embedding_type": "positional", + "timestep_post_act": null, + "up_block_types": [ + "UpBlock2D", + "CrossAttnUpBlock2D", + "CrossAttnUpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "update_after_step": 0, + "use_ema_warmup": true, + "use_linear_projection": false +} diff --git a/checkpoint-22500/unet_ema/diffusion_pytorch_model.bin b/checkpoint-22500/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..30e069f827f49148010792802df50a59a8940a9a --- /dev/null +++ b/checkpoint-22500/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:0a6147f208b06bc2535c2d2487dcc07f4071c9b5956ca4f8ce49d28927709c0d +size 62996305 diff --git a/checkpoint-23000/optimizer.bin b/checkpoint-23000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..e392d24ff3e05d49aff3152bc004ea97430f50a6 --- /dev/null +++ b/checkpoint-23000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:f8d4553a2889fc2b0b1a0af5ef8d3c18686188d9c8212a8587ffa658d0a2e573 +size 126008021 diff --git a/checkpoint-23000/random_states_0.pkl b/checkpoint-23000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..a776e2a7a75240ad271a86c7d2454fca21e64c56 --- /dev/null +++ b/checkpoint-23000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:282988fa1ca2fff4507bf999c51359a9328db39379c3f65ec24cab0d80aa618a +size 14727 diff --git a/checkpoint-23000/scaler.pt b/checkpoint-23000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..115263c9235918fc2a72413ed7e00c969b8d3527 --- /dev/null +++ b/checkpoint-23000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7fa181fa360d46feed4180ea17c8b6a4a879a9b4231c2e91aff2be20be9076cc +size 557 diff --git a/checkpoint-23000/scheduler.bin b/checkpoint-23000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..c104d255a8fe8ba2988714910e881fb4beda5a15 --- /dev/null +++ b/checkpoint-23000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:289bdc7c8f764b66f89c7d91c25b2544bfb7e66f32ab30476149139d8e8ad13d +size 563 diff --git a/checkpoint-23000/unet/config.json b/checkpoint-23000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..06fb387aa3ed37ab3b2f455b5469059d55604f74 --- /dev/null +++ b/checkpoint-23000/unet/config.json @@ -0,0 +1,60 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "attention_head_dim": 8, + "block_out_channels": [ + 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-23000/unet/diffusion_pytorch_model.bin b/checkpoint-23000/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..f32e7bd385184581ee545b5aee58c14b25ec56e4 --- /dev/null +++ b/checkpoint-23000/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-23000/unet_ema/config.json b/checkpoint-23000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..687f29c41c1875df0d8546fe2321e403e4ae1565 --- /dev/null +++ b/checkpoint-23000/unet_ema/config.json @@ -0,0 +1,67 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "attention_head_dim": 8, + "block_out_channels": [ + 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": 23000, + "out_channels": 3, + "power": 0.75, + "projection_class_embeddings_input_dim": null, + "resnet_out_scale_factor": 1.0, + "resnet_skip_time_act": false, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_cond_proj_dim": null, + "time_embedding_act_fn": null, + "time_embedding_dim": null, + "time_embedding_type": "positional", + "timestep_post_act": null, + "up_block_types": [ + "UpBlock2D", + "CrossAttnUpBlock2D", + "CrossAttnUpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "update_after_step": 0, + "use_ema_warmup": true, + "use_linear_projection": false +} diff --git a/checkpoint-23000/unet_ema/diffusion_pytorch_model.bin b/checkpoint-23000/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..5baac0a506339b50d09a36869c28d5fd5801625e --- /dev/null +++ b/checkpoint-23000/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:8de1bcad138e7522c725408413333527ee3daa47e878847217caba9b6568a224 +size 62996305 diff --git a/checkpoint-23500/optimizer.bin b/checkpoint-23500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..14eb37d0c4ae5b747ed9ebab45847eb8baa4b40f --- /dev/null +++ b/checkpoint-23500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:30b353d1d7d4b2cfff71c175967e8ff3bdaa51dced7ca97cafed6d7e696080c5 +size 126008021 diff --git a/checkpoint-23500/random_states_0.pkl b/checkpoint-23500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..d2bd77be8cd714a2ede81099bcdf77a8fca23ab0 --- /dev/null +++ b/checkpoint-23500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:aece4532c9677430e5afabf630900ff368b96206dfd814b8d6386ba77052a425 +size 14727 diff --git a/checkpoint-23500/scaler.pt b/checkpoint-23500/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..115263c9235918fc2a72413ed7e00c969b8d3527 --- /dev/null +++ b/checkpoint-23500/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7fa181fa360d46feed4180ea17c8b6a4a879a9b4231c2e91aff2be20be9076cc +size 557 diff --git a/checkpoint-23500/scheduler.bin b/checkpoint-23500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..21956be86a8f33a95189f483223aa0996d65545b --- /dev/null +++ b/checkpoint-23500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:30f14d106784b913c45f81e20f984b2be3f601e76b9008d3ee7bfe1707a3c978 +size 563 diff --git a/checkpoint-23500/unet/config.json b/checkpoint-23500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..06fb387aa3ed37ab3b2f455b5469059d55604f74 --- /dev/null +++ b/checkpoint-23500/unet/config.json @@ -0,0 +1,60 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "attention_head_dim": 8, + "block_out_channels": [ + 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-23500/unet/diffusion_pytorch_model.bin b/checkpoint-23500/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..f32e7bd385184581ee545b5aee58c14b25ec56e4 --- /dev/null +++ b/checkpoint-23500/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-23500/unet_ema/config.json b/checkpoint-23500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..cf0c4513f317e99782a519ddaaf8113169546e04 --- /dev/null +++ b/checkpoint-23500/unet_ema/config.json @@ -0,0 +1,67 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "attention_head_dim": 8, + "block_out_channels": [ + 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": 23500, + "out_channels": 3, + "power": 0.75, + "projection_class_embeddings_input_dim": null, + "resnet_out_scale_factor": 1.0, + "resnet_skip_time_act": false, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_cond_proj_dim": null, + "time_embedding_act_fn": null, + "time_embedding_dim": null, + "time_embedding_type": "positional", + "timestep_post_act": null, + "up_block_types": [ + "UpBlock2D", + "CrossAttnUpBlock2D", + "CrossAttnUpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "update_after_step": 0, + "use_ema_warmup": true, + "use_linear_projection": false +} diff --git a/checkpoint-23500/unet_ema/diffusion_pytorch_model.bin b/checkpoint-23500/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..1488d14fd62068d4d4bc3e73b23db1ea5c962ace --- /dev/null +++ b/checkpoint-23500/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:6bc86a3220112829a05ebfcc22736f5fc97ba6defdb6ed7f70ce6a8a1a31842c +size 62996305 diff --git a/checkpoint-24000/optimizer.bin b/checkpoint-24000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..472f9dccfe9859c74fcc265e186809b4d229a006 --- /dev/null +++ b/checkpoint-24000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b6320c08caa88615c103ce9b94bc521cca04fab3aacdb7fdefb184ef9603cb7b +size 126008021 diff --git a/checkpoint-24000/random_states_0.pkl b/checkpoint-24000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..51d0b022c87e836f7b473f49774354696a3be42c --- /dev/null +++ b/checkpoint-24000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:8725c45d4d01485dc726093348b9a82f6505f0b7c1e2ef03173940476ac1a438 +size 14727 diff --git a/checkpoint-24000/scaler.pt b/checkpoint-24000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..115263c9235918fc2a72413ed7e00c969b8d3527 --- /dev/null +++ b/checkpoint-24000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7fa181fa360d46feed4180ea17c8b6a4a879a9b4231c2e91aff2be20be9076cc +size 557 diff --git a/checkpoint-24000/scheduler.bin b/checkpoint-24000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..7db42cbcab5275dea4de67cb296cf45e49f76328 --- /dev/null +++ b/checkpoint-24000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:9b4cc59b516d86068a62ce7cd67f91a2b80bab2d9bd211af953ad200e8338092 +size 563 diff --git a/checkpoint-24000/unet/config.json b/checkpoint-24000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..06fb387aa3ed37ab3b2f455b5469059d55604f74 --- /dev/null +++ b/checkpoint-24000/unet/config.json @@ -0,0 +1,60 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "attention_head_dim": 8, + "block_out_channels": [ + 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-24000/unet/diffusion_pytorch_model.bin b/checkpoint-24000/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..f32e7bd385184581ee545b5aee58c14b25ec56e4 --- /dev/null +++ b/checkpoint-24000/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-24000/unet_ema/config.json b/checkpoint-24000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..30c151789958a7dcc5da98dd9ab08c9106e79929 --- /dev/null +++ b/checkpoint-24000/unet_ema/config.json @@ -0,0 +1,67 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "attention_head_dim": 8, + "block_out_channels": [ + 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": 24000, + "out_channels": 3, + "power": 0.75, + "projection_class_embeddings_input_dim": null, + "resnet_out_scale_factor": 1.0, + "resnet_skip_time_act": false, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_cond_proj_dim": null, + "time_embedding_act_fn": null, + "time_embedding_dim": null, + "time_embedding_type": "positional", + "timestep_post_act": null, + "up_block_types": [ + "UpBlock2D", + "CrossAttnUpBlock2D", + "CrossAttnUpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "update_after_step": 0, + "use_ema_warmup": true, + "use_linear_projection": false +} diff --git a/checkpoint-24000/unet_ema/diffusion_pytorch_model.bin b/checkpoint-24000/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..421eaca0f3355490d11d27d5875fc3f75dfadeaf --- /dev/null +++ b/checkpoint-24000/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:6163c08cae6e4fbcbbc4dfaa8d499a389fb0504015d065f964453bf6023ae9ce +size 62996305 diff --git a/checkpoint-24500/optimizer.bin b/checkpoint-24500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..05c934cfa9ef0d5cf7ca0ef5a6d68b8386c2510f --- /dev/null +++ b/checkpoint-24500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:170d890b5c12a4053742208cfd9332460a40712e1af85a526220bafa44aeca35 +size 126008021 diff --git a/checkpoint-24500/random_states_0.pkl b/checkpoint-24500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..e783adeff23c9c789c9d73fa9cd90cc85a40a334 --- /dev/null +++ b/checkpoint-24500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:170008d78e9ca585ee6551fe1c8629c1c61f21ad3792f80c50678e3b3c52321a +size 14727 diff --git a/checkpoint-24500/scaler.pt b/checkpoint-24500/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..115263c9235918fc2a72413ed7e00c969b8d3527 --- /dev/null +++ b/checkpoint-24500/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7fa181fa360d46feed4180ea17c8b6a4a879a9b4231c2e91aff2be20be9076cc +size 557 diff --git a/checkpoint-24500/scheduler.bin b/checkpoint-24500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..9d14176ba0f530c8d14f3dac65e02623902fc302 --- /dev/null +++ b/checkpoint-24500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:d05022af3cc418656daef6cb1c92472676da2e832b274f8e5f7412543a95f5ff +size 563 diff --git a/checkpoint-24500/unet/config.json b/checkpoint-24500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..06fb387aa3ed37ab3b2f455b5469059d55604f74 --- /dev/null +++ b/checkpoint-24500/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-24500/unet/diffusion_pytorch_model.bin b/checkpoint-24500/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..f32e7bd385184581ee545b5aee58c14b25ec56e4 --- /dev/null +++ b/checkpoint-24500/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-24500/unet_ema/config.json b/checkpoint-24500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..a39e1dd4ac8c0578977ca52972d680ec8c36a371 --- /dev/null +++ b/checkpoint-24500/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": 24500, + "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-24500/unet_ema/diffusion_pytorch_model.bin b/checkpoint-24500/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..076569b82937af9857911b8f1f6a1cf09e8631f1 --- /dev/null +++ b/checkpoint-24500/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e28e6fb177ee2a225897b30fa3fa03b1801188c4cb3df3462cb14a2aabf6cd44 +size 62996305 diff --git a/checkpoint-25000/optimizer.bin b/checkpoint-25000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..aa96825be24e1310cc8b3f925dd84b7acfbb749a --- /dev/null +++ b/checkpoint-25000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:fcea1ed947cce2b7d9a8c6a498e227c9fc7ed6b535f08b4f3f1dc4e578ef3619 +size 126008021 diff --git a/checkpoint-25000/random_states_0.pkl b/checkpoint-25000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..490d3b89231a9c00b7cd47ba6f9f12383c1dda25 --- /dev/null +++ b/checkpoint-25000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:8da41d4b12ea30d6759d8b55dd0db0c3cf52de5ac646bc21ba1f47307277f855 +size 14727 diff --git a/checkpoint-25000/scaler.pt b/checkpoint-25000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..115263c9235918fc2a72413ed7e00c969b8d3527 --- /dev/null +++ b/checkpoint-25000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7fa181fa360d46feed4180ea17c8b6a4a879a9b4231c2e91aff2be20be9076cc +size 557 diff --git a/checkpoint-25000/scheduler.bin b/checkpoint-25000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..70034e58cb46062e1618084f6e40ac56b55e5597 --- /dev/null +++ b/checkpoint-25000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:162d7823446aff3f5bb62fb82ada5feaa62165718ffee0b6df4459221ee7d706 +size 563 diff --git a/checkpoint-25000/unet/config.json b/checkpoint-25000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..06fb387aa3ed37ab3b2f455b5469059d55604f74 --- /dev/null +++ b/checkpoint-25000/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-25000/unet/diffusion_pytorch_model.bin b/checkpoint-25000/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..f32e7bd385184581ee545b5aee58c14b25ec56e4 --- /dev/null +++ b/checkpoint-25000/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-25000/unet_ema/config.json b/checkpoint-25000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..2f1cd9bbad7496d6147fe9d27a6ac23c3eeb457f --- /dev/null +++ b/checkpoint-25000/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": 25000, + "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-25000/unet_ema/diffusion_pytorch_model.bin b/checkpoint-25000/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..cd99ad4716f769afdf601847bfbea9299960b677 --- /dev/null +++ b/checkpoint-25000/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:34f728401d41f7fd0e655af9642ef7e6a8b60d0ff89a22f6eb8d18786b4b2f55 +size 62996305 diff --git a/checkpoint-25500/optimizer.bin b/checkpoint-25500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..b094f9ba502dc865e25447a5f52cc84110fdd626 --- /dev/null +++ b/checkpoint-25500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:aff9e29a3a30b7236b29c4bdbea69d00e10764d621b552c69b4e4c547228c7d9 +size 126008021 diff --git a/checkpoint-25500/random_states_0.pkl b/checkpoint-25500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..e96ae7052259ede1e58060e86dc60e1f41fbc536 --- /dev/null +++ b/checkpoint-25500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:8ef28bce66ee19eaac49fd72f8825b63f9b7de50fe69b367b2915b9ba34f7f77 +size 14727 diff --git a/checkpoint-25500/scaler.pt b/checkpoint-25500/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..115263c9235918fc2a72413ed7e00c969b8d3527 --- /dev/null +++ b/checkpoint-25500/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7fa181fa360d46feed4180ea17c8b6a4a879a9b4231c2e91aff2be20be9076cc +size 557 diff --git a/checkpoint-25500/scheduler.bin b/checkpoint-25500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..635213ecdb34595e2f99d5ee360ef043cdac534f --- /dev/null +++ b/checkpoint-25500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:62451b1e4327d7d54f96a5bc9d911a2f0d5258971c8eeaf377cbb77ed0eb375f +size 563 diff --git a/checkpoint-25500/unet/config.json b/checkpoint-25500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..06fb387aa3ed37ab3b2f455b5469059d55604f74 --- /dev/null +++ b/checkpoint-25500/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-25500/unet/diffusion_pytorch_model.bin b/checkpoint-25500/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..f32e7bd385184581ee545b5aee58c14b25ec56e4 --- /dev/null +++ b/checkpoint-25500/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-25500/unet_ema/config.json b/checkpoint-25500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..a9680b593eba5f7caf53cbf5b081baa47562f02d --- /dev/null +++ b/checkpoint-25500/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": 25500, + "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-25500/unet_ema/diffusion_pytorch_model.bin b/checkpoint-25500/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..6cf742eb866dd40e75e5963c0b6b6f32b528defd --- /dev/null +++ b/checkpoint-25500/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:66419fb8db25a1becccb1f4b854c66bbf320ff1821a7b0311db22a16227c37bf +size 62996305 diff --git a/unet/diffusion_pytorch_model.bin b/unet/diffusion_pytorch_model.bin index 7d327f1d5ebfcd4cd59dab3de5aaf3c33328d5f8..f6b04a14f4c2a5f522db6be0343cb2f860e6cc08 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:bdf15c4d535d7d266c0ea6b250c6e62f533932d59d47749d24adb2257af7b497 +oid sha256:fbe74914f5b21d7ffdccc97296c4afb089e658de6a444e8dd21a912b3c7ee718 size 63003985