diff --git a/checkpoint-10000/optimizer.bin b/checkpoint-10000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..ed816480939c8b2a507bb1917cbea2d9a17c1d70 --- /dev/null +++ b/checkpoint-10000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:3461f23d40cad5af783e587ff9f3f6260c623437cdd5b85b93162c05378951e8 +size 73251413 diff --git a/checkpoint-10000/random_states_0.pkl b/checkpoint-10000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..a39e71e80c760f32e00a2a3ea562fc4d0225b359 --- /dev/null +++ b/checkpoint-10000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:852b8476e89ba9ca5d3b25b798857f40f02fdf04681cf82395053e27e3d3ae05 +size 14663 diff --git a/checkpoint-10000/scaler.pt b/checkpoint-10000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..b345659e084dfa08f03a221d79b2e302a4748dfe --- /dev/null +++ b/checkpoint-10000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:9be2ad76dcbc923b00e6a142f6db62aad4a46c47bb83864ccb68ddc899d0ce78 +size 557 diff --git a/checkpoint-10000/scheduler.bin b/checkpoint-10000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..6355385997b8d9a199621b17588c267826cb0488 --- /dev/null +++ b/checkpoint-10000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:0f1bb79807602853a9b929a33930e620d3677c056b865ac90c06fdbcf9fe5a39 +size 563 diff --git a/checkpoint-10000/unet/config.json b/checkpoint-10000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..bb1cd664fc36d4dc821a4f0fdae4b94d7d99b356 --- /dev/null +++ b/checkpoint-10000/unet/config.json @@ -0,0 +1,60 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "attention_head_dim": 8, + "block_out_channels": [ + 64, + 64, + 128, + 128 + ], + "center_input_sample": false, + "class_embed_type": null, + "class_embeddings_concat": false, + "conv_in_kernel": 3, + "conv_out_kernel": 3, + "cross_attention_dim": 256, + "cross_attention_norm": null, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "CrossAttnDownBlock2D" + ], + "downsample_padding": 1, + "dual_cross_attention": false, + "encoder_hid_dim": null, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "layers_per_block": 2, + "mid_block_only_cross_attention": null, + "mid_block_scale_factor": 1, + "mid_block_type": "UNetMidBlock2DCrossAttn", + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "only_cross_attention": false, + "out_channels": 3, + "projection_class_embeddings_input_dim": null, + "resnet_out_scale_factor": 1.0, + "resnet_skip_time_act": false, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_cond_proj_dim": null, + "time_embedding_act_fn": null, + "time_embedding_dim": null, + "time_embedding_type": "positional", + "timestep_post_act": null, + "up_block_types": [ + "CrossAttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "use_linear_projection": false +} diff --git a/checkpoint-10000/unet/diffusion_pytorch_model.bin b/checkpoint-10000/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..e9505ccf874db778cb6a1f21cab0947b9561fba0 --- /dev/null +++ b/checkpoint-10000/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:bd5dcebaea0b32d33b85641dd3cb1452713faa00880a7e40026487baf54cb489 +size 36621969 diff --git a/checkpoint-10000/unet_ema/config.json b/checkpoint-10000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..6e8828a68f239e6460e1a9de4d1c68b8bc66f5f2 --- /dev/null +++ b/checkpoint-10000/unet_ema/config.json @@ -0,0 +1,67 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "attention_head_dim": 8, + "block_out_channels": [ + 64, + 64, + 128, + 128 + ], + "center_input_sample": false, + "class_embed_type": null, + "class_embeddings_concat": false, + "conv_in_kernel": 3, + "conv_out_kernel": 3, + "cross_attention_dim": 256, + "cross_attention_norm": null, + "decay": 0.9999, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "CrossAttnDownBlock2D" + ], + "downsample_padding": 1, + "dual_cross_attention": false, + "encoder_hid_dim": null, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "inv_gamma": 1.0, + "layers_per_block": 2, + "mid_block_only_cross_attention": null, + "mid_block_scale_factor": 1, + "mid_block_type": "UNetMidBlock2DCrossAttn", + "min_decay": 0.0, + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "only_cross_attention": false, + "optimization_step": 10000, + "out_channels": 3, + "power": 0.75, + "projection_class_embeddings_input_dim": null, + "resnet_out_scale_factor": 1.0, + "resnet_skip_time_act": false, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_cond_proj_dim": null, + "time_embedding_act_fn": null, + "time_embedding_dim": null, + "time_embedding_type": "positional", + "timestep_post_act": null, + "up_block_types": [ + "CrossAttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "update_after_step": 0, + "use_ema_warmup": true, + "use_linear_projection": false +} diff --git a/checkpoint-10000/unet_ema/diffusion_pytorch_model.bin b/checkpoint-10000/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..9e3aeb910289f04bcc79a4d3e4f5b7056858284c --- /dev/null +++ b/checkpoint-10000/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:9a7a58232fc4bca56bdccfc5651ede62692b6b9ef4c2e922b592b6e5afc488dc +size 36616081 diff --git a/checkpoint-7000/optimizer.bin b/checkpoint-7000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..dc1e16074fb68dcca2c84585e7d4244dc6cafc65 --- /dev/null +++ b/checkpoint-7000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e7f5d59b7e393078953eff769f6b788d77aa9efbff6af00e2e761479e22a74f3 +size 73251413 diff --git a/checkpoint-7000/random_states_0.pkl b/checkpoint-7000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..4138e204e3e3a927a356ee7517f7474d14519780 --- /dev/null +++ b/checkpoint-7000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:591d6d4d059dc97c7659a135895e593d93af39b1c25c432bdfca1f644cc6e698 +size 14663 diff --git a/checkpoint-7000/scaler.pt b/checkpoint-7000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..2ceac8d008df631f48c13ff48e83da859977ff02 --- /dev/null +++ b/checkpoint-7000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:d22aed3a858ee389297b2101bb2ed7ee9e37744f1a1f75273dbdf045d65b041a +size 557 diff --git a/checkpoint-7000/scheduler.bin b/checkpoint-7000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..6ec363c20b81c949003e184d90edf4b1606885a5 --- /dev/null +++ b/checkpoint-7000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:a5ffa669c96b14b85dd0b293ca82f551eb33c2fe4c81ccb71442e454826f0525 +size 563 diff --git a/checkpoint-7000/unet/config.json b/checkpoint-7000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..bb1cd664fc36d4dc821a4f0fdae4b94d7d99b356 --- /dev/null +++ b/checkpoint-7000/unet/config.json @@ -0,0 +1,60 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "attention_head_dim": 8, + "block_out_channels": [ + 64, + 64, + 128, + 128 + ], + "center_input_sample": false, + "class_embed_type": null, + "class_embeddings_concat": false, + "conv_in_kernel": 3, + "conv_out_kernel": 3, + "cross_attention_dim": 256, + "cross_attention_norm": null, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "CrossAttnDownBlock2D" + ], + "downsample_padding": 1, + "dual_cross_attention": false, + "encoder_hid_dim": null, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "layers_per_block": 2, + "mid_block_only_cross_attention": null, + "mid_block_scale_factor": 1, + "mid_block_type": "UNetMidBlock2DCrossAttn", + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "only_cross_attention": false, + "out_channels": 3, + "projection_class_embeddings_input_dim": null, + "resnet_out_scale_factor": 1.0, + "resnet_skip_time_act": false, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_cond_proj_dim": null, + "time_embedding_act_fn": null, + "time_embedding_dim": null, + "time_embedding_type": "positional", + "timestep_post_act": null, + "up_block_types": [ + "CrossAttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "use_linear_projection": false +} diff --git a/checkpoint-7000/unet/diffusion_pytorch_model.bin b/checkpoint-7000/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..7c263c9ebd53957b0499a4e9bc45ec73e737ec4e --- /dev/null +++ b/checkpoint-7000/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:62e4469e7d57c318031d778d85c27b45b281aa539fb0256099c302eda379dd56 +size 36621969 diff --git a/checkpoint-7000/unet_ema/config.json b/checkpoint-7000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..bc6d9832798ee02cd64390701c59623b220f77d6 --- /dev/null +++ b/checkpoint-7000/unet_ema/config.json @@ -0,0 +1,67 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "attention_head_dim": 8, + "block_out_channels": [ + 64, + 64, + 128, + 128 + ], + "center_input_sample": false, + "class_embed_type": null, + "class_embeddings_concat": false, + "conv_in_kernel": 3, + "conv_out_kernel": 3, + "cross_attention_dim": 256, + "cross_attention_norm": null, + "decay": 0.9999, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "CrossAttnDownBlock2D" + ], + "downsample_padding": 1, + "dual_cross_attention": false, + "encoder_hid_dim": null, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "inv_gamma": 1.0, + "layers_per_block": 2, + "mid_block_only_cross_attention": null, + "mid_block_scale_factor": 1, + "mid_block_type": "UNetMidBlock2DCrossAttn", + "min_decay": 0.0, + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "only_cross_attention": false, + "optimization_step": 7000, + "out_channels": 3, + "power": 0.75, + "projection_class_embeddings_input_dim": null, + "resnet_out_scale_factor": 1.0, + "resnet_skip_time_act": false, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_cond_proj_dim": null, + "time_embedding_act_fn": null, + "time_embedding_dim": null, + "time_embedding_type": "positional", + "timestep_post_act": null, + "up_block_types": [ + "CrossAttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "update_after_step": 0, + "use_ema_warmup": true, + "use_linear_projection": false +} diff --git a/checkpoint-7000/unet_ema/diffusion_pytorch_model.bin b/checkpoint-7000/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..e28b8cf0510f33642af949e41051a489af6778be --- /dev/null +++ b/checkpoint-7000/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:1c8ceb1273db471f16e62f0087e4a3cd414c62c7a392e3877e8e82143651a321 +size 36616081 diff --git a/checkpoint-7500/optimizer.bin b/checkpoint-7500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..a171cc29b2be4235bc1c2dab9324d70f80d1499e --- /dev/null +++ b/checkpoint-7500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e9485ad1321d89efa5b5c76eb67df347001c6a3864ed8a55ee5d7cd84656a88d +size 73251413 diff --git a/checkpoint-7500/random_states_0.pkl b/checkpoint-7500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..0b683a1433fda5878a4378d335285755fadda134 --- /dev/null +++ b/checkpoint-7500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b378e80df497993a2202f8309cf0d1ebaa089d805885b310d1187c43875462d1 +size 14663 diff --git a/checkpoint-7500/scaler.pt b/checkpoint-7500/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..60bdd119cc6377670954947196f4e3abc08dcce1 --- /dev/null +++ b/checkpoint-7500/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:fab868730e2cdc5d83f898b6c5399b352e14373e48bdbdfa9179803e96fd314f +size 557 diff --git a/checkpoint-7500/scheduler.bin b/checkpoint-7500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..9afd5d82025fe77c9a2b0804892f1af2ad764741 --- /dev/null +++ b/checkpoint-7500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:facedcdc17eb26e1728619ebe64654dc4fd1908b60d23ecb9f21dc05c8bd1136 +size 563 diff --git a/checkpoint-7500/unet/config.json b/checkpoint-7500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..bb1cd664fc36d4dc821a4f0fdae4b94d7d99b356 --- /dev/null +++ b/checkpoint-7500/unet/config.json @@ -0,0 +1,60 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "attention_head_dim": 8, + "block_out_channels": [ + 64, + 64, + 128, + 128 + ], + "center_input_sample": false, + "class_embed_type": null, + "class_embeddings_concat": false, + "conv_in_kernel": 3, + "conv_out_kernel": 3, + "cross_attention_dim": 256, + "cross_attention_norm": null, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "CrossAttnDownBlock2D" + ], + "downsample_padding": 1, + "dual_cross_attention": false, + "encoder_hid_dim": null, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "layers_per_block": 2, + "mid_block_only_cross_attention": null, + "mid_block_scale_factor": 1, + "mid_block_type": "UNetMidBlock2DCrossAttn", + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "only_cross_attention": false, + "out_channels": 3, + "projection_class_embeddings_input_dim": null, + "resnet_out_scale_factor": 1.0, + "resnet_skip_time_act": false, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_cond_proj_dim": null, + "time_embedding_act_fn": null, + "time_embedding_dim": null, + "time_embedding_type": "positional", + "timestep_post_act": null, + "up_block_types": [ + "CrossAttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "use_linear_projection": false +} diff --git a/checkpoint-7500/unet/diffusion_pytorch_model.bin b/checkpoint-7500/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..9f91a8bf63f6b367c4e1b3f18f32761cc31c4b6d --- /dev/null +++ b/checkpoint-7500/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:8f7dc2a519753a97c06a9b38d0fb1fe3eae7eb809059847ac3585c4f2c9f93b0 +size 36621969 diff --git a/checkpoint-7500/unet_ema/config.json b/checkpoint-7500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..f19fdc1b3efb39c21bf0a7259a9dd1cb52db20bb --- /dev/null +++ b/checkpoint-7500/unet_ema/config.json @@ -0,0 +1,67 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "attention_head_dim": 8, + "block_out_channels": [ + 64, + 64, + 128, + 128 + ], + "center_input_sample": false, + "class_embed_type": null, + "class_embeddings_concat": false, + "conv_in_kernel": 3, + "conv_out_kernel": 3, + "cross_attention_dim": 256, + "cross_attention_norm": null, + "decay": 0.9999, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "CrossAttnDownBlock2D" + ], + "downsample_padding": 1, + "dual_cross_attention": false, + "encoder_hid_dim": null, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "inv_gamma": 1.0, + "layers_per_block": 2, + "mid_block_only_cross_attention": null, + "mid_block_scale_factor": 1, + "mid_block_type": "UNetMidBlock2DCrossAttn", + "min_decay": 0.0, + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "only_cross_attention": false, + "optimization_step": 7500, + "out_channels": 3, + "power": 0.75, + "projection_class_embeddings_input_dim": null, + "resnet_out_scale_factor": 1.0, + "resnet_skip_time_act": false, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_cond_proj_dim": null, + "time_embedding_act_fn": null, + "time_embedding_dim": null, + "time_embedding_type": "positional", + "timestep_post_act": null, + "up_block_types": [ + "CrossAttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "update_after_step": 0, + "use_ema_warmup": true, + "use_linear_projection": false +} diff --git a/checkpoint-7500/unet_ema/diffusion_pytorch_model.bin b/checkpoint-7500/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..1f7c53e60f4f58d1b3d818044580ae6aca691b80 --- /dev/null +++ b/checkpoint-7500/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:f4fc91cbb312284924ea0c1af0b1c076168ef0d2f5843c097962cecdb91d38d8 +size 36616081 diff --git a/checkpoint-8000/optimizer.bin b/checkpoint-8000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..ce2ab5a85aa19b987f2a969c36b0a29d237fe0ec --- /dev/null +++ b/checkpoint-8000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:28511e509205687dc07a5c92676cb95aedf6a72a2bec4f7f105449f00197a9ee +size 73251413 diff --git a/checkpoint-8000/random_states_0.pkl b/checkpoint-8000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..4ccedbf6cb3e6443c937ce4c33f1bee3f23e518b --- /dev/null +++ b/checkpoint-8000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:751afdbaf6c1bfba002501fa84058ab837f7e3fc7182aa1201a93617028de380 +size 14663 diff --git a/checkpoint-8000/scaler.pt b/checkpoint-8000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..fd25dc8bd6a1a15851636b4ebafe829cb0ed1b0d --- /dev/null +++ b/checkpoint-8000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b278cd6e09360f31a3d837f80dee4c2ce4d9c9d186a939ecf157e1a0deb793f3 +size 557 diff --git a/checkpoint-8000/scheduler.bin b/checkpoint-8000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..c7d6443fb792a24c5bd7e36d2bf29e84bc85296f --- /dev/null +++ b/checkpoint-8000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:5d94c1d7c3e2abb40d6062281af4303f8cef91c9fa7a7d997f1cc9c88abcf650 +size 563 diff --git a/checkpoint-8000/unet/config.json b/checkpoint-8000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..bb1cd664fc36d4dc821a4f0fdae4b94d7d99b356 --- /dev/null +++ b/checkpoint-8000/unet/config.json @@ -0,0 +1,60 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "attention_head_dim": 8, + "block_out_channels": [ + 64, + 64, + 128, + 128 + ], + "center_input_sample": false, + "class_embed_type": null, + "class_embeddings_concat": false, + "conv_in_kernel": 3, + "conv_out_kernel": 3, + "cross_attention_dim": 256, + "cross_attention_norm": null, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "CrossAttnDownBlock2D" + ], + "downsample_padding": 1, + "dual_cross_attention": false, + "encoder_hid_dim": null, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "layers_per_block": 2, + "mid_block_only_cross_attention": null, + "mid_block_scale_factor": 1, + "mid_block_type": "UNetMidBlock2DCrossAttn", + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "only_cross_attention": false, + "out_channels": 3, + "projection_class_embeddings_input_dim": null, + "resnet_out_scale_factor": 1.0, + "resnet_skip_time_act": false, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_cond_proj_dim": null, + "time_embedding_act_fn": null, + "time_embedding_dim": null, + "time_embedding_type": "positional", + "timestep_post_act": null, + "up_block_types": [ + "CrossAttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "use_linear_projection": false +} diff --git a/checkpoint-8000/unet/diffusion_pytorch_model.bin b/checkpoint-8000/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..f95875d89185882fafb075e4b84dd4e8256e4a03 --- /dev/null +++ b/checkpoint-8000/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:0cd07fd95f5ac75626a08bd003ea15c7cacd630ecbadca9f7e979b92ce286d97 +size 36621969 diff --git a/checkpoint-8000/unet_ema/config.json b/checkpoint-8000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..0855394c994042a979b168b971b6699eee61d023 --- /dev/null +++ b/checkpoint-8000/unet_ema/config.json @@ -0,0 +1,67 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "attention_head_dim": 8, + "block_out_channels": [ + 64, + 64, + 128, + 128 + ], + "center_input_sample": false, + "class_embed_type": null, + "class_embeddings_concat": false, + "conv_in_kernel": 3, + "conv_out_kernel": 3, + "cross_attention_dim": 256, + "cross_attention_norm": null, + "decay": 0.9999, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "CrossAttnDownBlock2D" + ], + "downsample_padding": 1, + "dual_cross_attention": false, + "encoder_hid_dim": null, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "inv_gamma": 1.0, + "layers_per_block": 2, + "mid_block_only_cross_attention": null, + "mid_block_scale_factor": 1, + "mid_block_type": "UNetMidBlock2DCrossAttn", + "min_decay": 0.0, + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "only_cross_attention": false, + "optimization_step": 8000, + "out_channels": 3, + "power": 0.75, + "projection_class_embeddings_input_dim": null, + "resnet_out_scale_factor": 1.0, + "resnet_skip_time_act": false, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_cond_proj_dim": null, + "time_embedding_act_fn": null, + "time_embedding_dim": null, + "time_embedding_type": "positional", + "timestep_post_act": null, + "up_block_types": [ + "CrossAttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "update_after_step": 0, + "use_ema_warmup": true, + "use_linear_projection": false +} diff --git a/checkpoint-8000/unet_ema/diffusion_pytorch_model.bin b/checkpoint-8000/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..53fe9be82c939506769913affbf006d8c82a911f --- /dev/null +++ b/checkpoint-8000/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:9c9c030559358248221cb1e409719df455ca3deb8d2835114b157fa3572db848 +size 36616081 diff --git a/checkpoint-8500/optimizer.bin b/checkpoint-8500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..c6b0701edc25bda1e435908572746cb673f3a0f1 --- /dev/null +++ b/checkpoint-8500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:2f89ad1c2f73469bb21f1944a0b3370bfaf719ef99325248b8603b7ed275f921 +size 73251413 diff --git a/checkpoint-8500/random_states_0.pkl b/checkpoint-8500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..410aa3dae92d299ec211b3e8f69d1790d9c24098 --- /dev/null +++ b/checkpoint-8500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:8ef629902f50f8c547877a9c1b35eb6779d4489e90fef1853db3a8920168f1d9 +size 14663 diff --git a/checkpoint-8500/scaler.pt b/checkpoint-8500/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..3df7c4409be1ed9ef627cc178b2c13e5d9ed6b01 --- /dev/null +++ b/checkpoint-8500/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:5ed0eced7c80f9d236fedf48c2537ab9b4e6957d5129ef3aec426e866324bfd1 +size 557 diff --git a/checkpoint-8500/scheduler.bin b/checkpoint-8500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..48bcfd7c5d13ae4c9213aa34c91032e7d090bb2a --- /dev/null +++ b/checkpoint-8500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:a8a36ce5f73caf5b02af6dfbe5f7940870d5612887621a11651ef28fa88084ef +size 563 diff --git a/checkpoint-8500/unet/config.json b/checkpoint-8500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..bb1cd664fc36d4dc821a4f0fdae4b94d7d99b356 --- /dev/null +++ b/checkpoint-8500/unet/config.json @@ -0,0 +1,60 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "attention_head_dim": 8, + "block_out_channels": [ + 64, + 64, + 128, + 128 + ], + "center_input_sample": false, + "class_embed_type": null, + "class_embeddings_concat": false, + "conv_in_kernel": 3, + "conv_out_kernel": 3, + "cross_attention_dim": 256, + "cross_attention_norm": null, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "CrossAttnDownBlock2D" + ], + "downsample_padding": 1, + "dual_cross_attention": false, + "encoder_hid_dim": null, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "layers_per_block": 2, + "mid_block_only_cross_attention": null, + "mid_block_scale_factor": 1, + "mid_block_type": "UNetMidBlock2DCrossAttn", + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "only_cross_attention": false, + "out_channels": 3, + "projection_class_embeddings_input_dim": null, + "resnet_out_scale_factor": 1.0, + "resnet_skip_time_act": false, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_cond_proj_dim": null, + "time_embedding_act_fn": null, + "time_embedding_dim": null, + "time_embedding_type": "positional", + "timestep_post_act": null, + "up_block_types": [ + "CrossAttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "use_linear_projection": false +} diff --git a/checkpoint-8500/unet/diffusion_pytorch_model.bin b/checkpoint-8500/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..e85e807e976866e8424af2ce0b1c851f0320bae4 --- /dev/null +++ b/checkpoint-8500/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:9be248b149100ef7209d0716a1b0f02aed97d503489f23e0afea999ffc4f3487 +size 36621969 diff --git a/checkpoint-8500/unet_ema/config.json b/checkpoint-8500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..b671cd726290401586de0dddee38744e34acfa7b --- /dev/null +++ b/checkpoint-8500/unet_ema/config.json @@ -0,0 +1,67 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "attention_head_dim": 8, + "block_out_channels": [ + 64, + 64, + 128, + 128 + ], + "center_input_sample": false, + "class_embed_type": null, + "class_embeddings_concat": false, + "conv_in_kernel": 3, + "conv_out_kernel": 3, + "cross_attention_dim": 256, + "cross_attention_norm": null, + "decay": 0.9999, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "CrossAttnDownBlock2D" + ], + "downsample_padding": 1, + "dual_cross_attention": false, + "encoder_hid_dim": null, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "inv_gamma": 1.0, + "layers_per_block": 2, + "mid_block_only_cross_attention": null, + "mid_block_scale_factor": 1, + "mid_block_type": "UNetMidBlock2DCrossAttn", + "min_decay": 0.0, + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "only_cross_attention": false, + "optimization_step": 8500, + "out_channels": 3, + "power": 0.75, + "projection_class_embeddings_input_dim": null, + "resnet_out_scale_factor": 1.0, + "resnet_skip_time_act": false, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_cond_proj_dim": null, + "time_embedding_act_fn": null, + "time_embedding_dim": null, + "time_embedding_type": "positional", + "timestep_post_act": null, + "up_block_types": [ + "CrossAttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "update_after_step": 0, + "use_ema_warmup": true, + "use_linear_projection": false +} diff --git a/checkpoint-8500/unet_ema/diffusion_pytorch_model.bin b/checkpoint-8500/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..2ac988cd97a8b20f7630a6ce3cbd1d4aad0ad375 --- /dev/null +++ b/checkpoint-8500/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:1feb02d8d019532605d12e559279c5fc97fdf1b3fe700a61732f68446d5dbf4b +size 36616081 diff --git a/checkpoint-9000/optimizer.bin b/checkpoint-9000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..65391ceb91bdb4240565e981de6ab6356e423f09 --- /dev/null +++ b/checkpoint-9000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:a4e26421234c8e5db4534258d3b4d8d86c5cfce8fed7742b5e581d9021031af6 +size 73251413 diff --git a/checkpoint-9000/random_states_0.pkl b/checkpoint-9000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..0207ffbf838617b5b9a74af7a0c23d451caffb03 --- /dev/null +++ b/checkpoint-9000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:ac5be4875559e7991191515194cec5555fd9e7d029815ef533e84328e06e1d40 +size 14663 diff --git a/checkpoint-9000/scaler.pt b/checkpoint-9000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..7c7306b1b0e5b21e2fec7bb12b5a46cc7b66591f --- /dev/null +++ b/checkpoint-9000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:4be7a7b56cfc3e79d33648a4c49d4f11c6593d8d653b129207b2e38f2684a284 +size 557 diff --git a/checkpoint-9000/scheduler.bin b/checkpoint-9000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..cbccbe3437325607b2b6fcb9b95c2dfaadc1e7a7 --- /dev/null +++ b/checkpoint-9000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:bea4d834a8bceefe229ff7073ce740a413ead6db8f7c0c937e081bb6dee7f57c +size 563 diff --git a/checkpoint-9000/unet/config.json b/checkpoint-9000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..bb1cd664fc36d4dc821a4f0fdae4b94d7d99b356 --- /dev/null +++ b/checkpoint-9000/unet/config.json @@ -0,0 +1,60 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "attention_head_dim": 8, + "block_out_channels": [ + 64, + 64, + 128, + 128 + ], + "center_input_sample": false, + "class_embed_type": null, + "class_embeddings_concat": false, + "conv_in_kernel": 3, + "conv_out_kernel": 3, + "cross_attention_dim": 256, + "cross_attention_norm": null, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "CrossAttnDownBlock2D" + ], + "downsample_padding": 1, + "dual_cross_attention": false, + "encoder_hid_dim": null, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "layers_per_block": 2, + "mid_block_only_cross_attention": null, + "mid_block_scale_factor": 1, + "mid_block_type": "UNetMidBlock2DCrossAttn", + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "only_cross_attention": false, + "out_channels": 3, + "projection_class_embeddings_input_dim": null, + "resnet_out_scale_factor": 1.0, + "resnet_skip_time_act": false, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_cond_proj_dim": null, + "time_embedding_act_fn": null, + "time_embedding_dim": null, + "time_embedding_type": "positional", + "timestep_post_act": null, + "up_block_types": [ + "CrossAttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "use_linear_projection": false +} diff --git a/checkpoint-9000/unet/diffusion_pytorch_model.bin b/checkpoint-9000/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..daaf70167641cf2fd402965beb5a34f25e40e384 --- /dev/null +++ b/checkpoint-9000/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:4016a37dd139a39b303d25450933460fdad7267ef0d7578b3ca2abb3e877dbe4 +size 36621969 diff --git a/checkpoint-9000/unet_ema/config.json b/checkpoint-9000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..ad71de41c3f401f7bbe8cf7620d10daebd7b1ff3 --- /dev/null +++ b/checkpoint-9000/unet_ema/config.json @@ -0,0 +1,67 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "attention_head_dim": 8, + "block_out_channels": [ + 64, + 64, + 128, + 128 + ], + "center_input_sample": false, + "class_embed_type": null, + "class_embeddings_concat": false, + "conv_in_kernel": 3, + "conv_out_kernel": 3, + "cross_attention_dim": 256, + "cross_attention_norm": null, + "decay": 0.9999, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "CrossAttnDownBlock2D" + ], + "downsample_padding": 1, + "dual_cross_attention": false, + "encoder_hid_dim": null, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "inv_gamma": 1.0, + "layers_per_block": 2, + "mid_block_only_cross_attention": null, + "mid_block_scale_factor": 1, + "mid_block_type": "UNetMidBlock2DCrossAttn", + "min_decay": 0.0, + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "only_cross_attention": false, + "optimization_step": 9000, + "out_channels": 3, + "power": 0.75, + "projection_class_embeddings_input_dim": null, + "resnet_out_scale_factor": 1.0, + "resnet_skip_time_act": false, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_cond_proj_dim": null, + "time_embedding_act_fn": null, + "time_embedding_dim": null, + "time_embedding_type": "positional", + "timestep_post_act": null, + "up_block_types": [ + "CrossAttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "update_after_step": 0, + "use_ema_warmup": true, + "use_linear_projection": false +} diff --git a/checkpoint-9000/unet_ema/diffusion_pytorch_model.bin b/checkpoint-9000/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..243887b0c4ade0452dfdf18909d9e5b5d5b35bfe --- /dev/null +++ b/checkpoint-9000/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:3e53447dff3025dc5320871a1df84f9e4e7ce3831374133310101a7622356c31 +size 36616081 diff --git a/checkpoint-9500/optimizer.bin b/checkpoint-9500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..38b01e65bfbb642b683d27c6f1aa1623d2792bfa --- /dev/null +++ b/checkpoint-9500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:866c325db787f5fac5c272c689eb6f29ff4ef59ee4bd68780d0daaa160186e21 +size 73251413 diff --git a/checkpoint-9500/random_states_0.pkl b/checkpoint-9500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..4cfc8c7949072927d57532143b2bfd6ff60471b3 --- /dev/null +++ b/checkpoint-9500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:6c8f57baeef99ba3d0630be969218cce5a7c7606412ed67f0af2c4a395e4565b +size 14663 diff --git a/checkpoint-9500/scaler.pt b/checkpoint-9500/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..c37ed8f6ad3a1ccbab9977f6091f818ec6c46c95 --- /dev/null +++ b/checkpoint-9500/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:f0ba14834a4ee194de1acdb54bf5273a1da6bd2ffcc1c8c635b51720d72fbde8 +size 557 diff --git a/checkpoint-9500/scheduler.bin b/checkpoint-9500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..aa991ddd4ac91f169559c9642ce7bf2ebc250caa --- /dev/null +++ b/checkpoint-9500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:a0f695e284edcbd93c1ccc0fb2e49ee742600dedca27a4f141af15c4ddb44395 +size 563 diff --git a/checkpoint-9500/unet/config.json b/checkpoint-9500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..bb1cd664fc36d4dc821a4f0fdae4b94d7d99b356 --- /dev/null +++ b/checkpoint-9500/unet/config.json @@ -0,0 +1,60 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "attention_head_dim": 8, + "block_out_channels": [ + 64, + 64, + 128, + 128 + ], + "center_input_sample": false, + "class_embed_type": null, + "class_embeddings_concat": false, + "conv_in_kernel": 3, + "conv_out_kernel": 3, + "cross_attention_dim": 256, + "cross_attention_norm": null, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "CrossAttnDownBlock2D" + ], + "downsample_padding": 1, + "dual_cross_attention": false, + "encoder_hid_dim": null, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "layers_per_block": 2, + "mid_block_only_cross_attention": null, + "mid_block_scale_factor": 1, + "mid_block_type": "UNetMidBlock2DCrossAttn", + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "only_cross_attention": false, + "out_channels": 3, + "projection_class_embeddings_input_dim": null, + "resnet_out_scale_factor": 1.0, + "resnet_skip_time_act": false, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_cond_proj_dim": null, + "time_embedding_act_fn": null, + "time_embedding_dim": null, + "time_embedding_type": "positional", + "timestep_post_act": null, + "up_block_types": [ + "CrossAttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "use_linear_projection": false +} diff --git a/checkpoint-9500/unet/diffusion_pytorch_model.bin b/checkpoint-9500/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..932482d4b5c603e31dd5029c132c9f1505bc3109 --- /dev/null +++ b/checkpoint-9500/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:ca0a63ec68532c76731f9874d0f94ce5bf2c911dcbc9caceed0f1ebea8e55050 +size 36621969 diff --git a/checkpoint-9500/unet_ema/config.json b/checkpoint-9500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..57464e696854f3f12366891164dac2e77134d7fc --- /dev/null +++ b/checkpoint-9500/unet_ema/config.json @@ -0,0 +1,67 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "attention_head_dim": 8, + "block_out_channels": [ + 64, + 64, + 128, + 128 + ], + "center_input_sample": false, + "class_embed_type": null, + "class_embeddings_concat": false, + "conv_in_kernel": 3, + "conv_out_kernel": 3, + "cross_attention_dim": 256, + "cross_attention_norm": null, + "decay": 0.9999, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "CrossAttnDownBlock2D" + ], + "downsample_padding": 1, + "dual_cross_attention": false, + "encoder_hid_dim": null, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "inv_gamma": 1.0, + "layers_per_block": 2, + "mid_block_only_cross_attention": null, + "mid_block_scale_factor": 1, + "mid_block_type": "UNetMidBlock2DCrossAttn", + "min_decay": 0.0, + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "only_cross_attention": false, + "optimization_step": 9500, + "out_channels": 3, + "power": 0.75, + "projection_class_embeddings_input_dim": null, + "resnet_out_scale_factor": 1.0, + "resnet_skip_time_act": false, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_cond_proj_dim": null, + "time_embedding_act_fn": null, + "time_embedding_dim": null, + "time_embedding_type": "positional", + "timestep_post_act": null, + "up_block_types": [ + "CrossAttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "update_after_step": 0, + "use_ema_warmup": true, + "use_linear_projection": false +} diff --git a/checkpoint-9500/unet_ema/diffusion_pytorch_model.bin b/checkpoint-9500/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..3abfe2e16a79035ae1cb79f07efe276fe7d30e53 --- /dev/null +++ b/checkpoint-9500/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:17e349261c8140dc40b32093d0d81065b02dda0b5406b79de350daad88463f6d +size 36616081 diff --git a/unet/diffusion_pytorch_model.bin b/unet/diffusion_pytorch_model.bin index 185226059ab118cb6978e89b1e96fe527d4a590f..b7aba9e3cf03be20b09f4f93dde59eda410cc8de 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:d50ebe1c15f6442d848a53b2dc109fcefae157be2ed0f5dca88810dc3780b86a +oid sha256:8e2d767168d5f6329cf93cc6180f652d5b8206fa7e69df784eac158df81c5301 size 36621969