diff --git a/checkpoint-1000/optimizer.bin b/checkpoint-1000/optimizer.bin index b3831ed48f25f3155ab0db477f5e0ecb316c3655..82918764737f6e28b1eba7ef429955c09850f2c6 100644 --- a/checkpoint-1000/optimizer.bin +++ b/checkpoint-1000/optimizer.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:5d4307d25c023a3bea6f3641ced6141d446847069a805ff1fde5e0cb315bdda3 -size 73251413 +oid sha256:8fe7c14a3fc7237b5514943d6d0379ef044e623f3b270674f0e4ea97bb80a705 +size 126008021 diff --git a/checkpoint-1000/random_states_0.pkl b/checkpoint-1000/random_states_0.pkl index ec0964495b1d0d95c50c4a7693fb311b05a7e2ff..051a9abb33c274fe3fcac414cad74972072d9585 100644 --- a/checkpoint-1000/random_states_0.pkl +++ b/checkpoint-1000/random_states_0.pkl @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:3eafd210728a28804a293f9081bc9bcf8f045f697b01b9b3a263ab4857866659 -size 14727 +oid sha256:890957394de8e6d5753a669da0f972ecb43e49b4de1e0a3394fbbfa18a54d36d +size 14663 diff --git a/checkpoint-1000/scheduler.bin b/checkpoint-1000/scheduler.bin index eadc8d207d12f94331973d1646fc97a94ac02a17..840698a8c2b12f7a0965cc0dd4d22c4b70814ee9 100644 --- a/checkpoint-1000/scheduler.bin +++ b/checkpoint-1000/scheduler.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:cdda063ad7b43dbc234b8c984f2f13df097879f0af1a5c065c4c8027fb3ef402 +oid sha256:01bdad9bf6a832cd95023c21764987eda8b509dba48cf1fbec5a5685ddf24d85 size 563 diff --git a/checkpoint-1000/unet/config.json b/checkpoint-1000/unet/config.json index c2d7b8c0d490531013583af57a806d21893e494c..06fb387aa3ed37ab3b2f455b5469059d55604f74 100644 --- a/checkpoint-1000/unet/config.json +++ b/checkpoint-1000/unet/config.json @@ -6,8 +6,8 @@ "addition_embed_type_num_heads": 64, "attention_head_dim": 8, "block_out_channels": [ - 64, - 64, + 128, + 128, 128, 128 ], @@ -19,9 +19,9 @@ "cross_attention_dim": 256, "cross_attention_norm": null, "down_block_types": [ - "DownBlock2D", "DownBlock2D", "CrossAttnDownBlock2D", + "CrossAttnDownBlock2D", "DownBlock2D" ], "downsample_padding": 1, @@ -52,7 +52,7 @@ "up_block_types": [ "UpBlock2D", "CrossAttnUpBlock2D", - "UpBlock2D", + "CrossAttnUpBlock2D", "UpBlock2D" ], "upcast_attention": false, diff --git a/checkpoint-1000/unet/diffusion_pytorch_model.bin b/checkpoint-1000/unet/diffusion_pytorch_model.bin index 4f5c4e6f954613f2c09c64853bba5f293e7bc316..8b18a5df6a82abc76449917e35eceb76c0bfdfa8 100644 --- a/checkpoint-1000/unet/diffusion_pytorch_model.bin +++ b/checkpoint-1000/unet/diffusion_pytorch_model.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:3bd8e6950d72cb21481e60e9ab866d260eb3ad53ec55ea9554847f2ae8a7b4d7 -size 36621969 +oid sha256:d57a42e21dc5377e56662497ee1255c741baebdcc4792cc013d53959ea393223 +size 63003985 diff --git a/checkpoint-1000/unet_ema/config.json b/checkpoint-1000/unet_ema/config.json index 50f94f5f3704076f4dcae9c887af7cccce268286..bd4c847ac725ce565773e7d4794a2f012faf0b26 100644 --- a/checkpoint-1000/unet_ema/config.json +++ b/checkpoint-1000/unet_ema/config.json @@ -6,8 +6,8 @@ "addition_embed_type_num_heads": 64, "attention_head_dim": 8, "block_out_channels": [ - 64, - 64, + 128, + 128, 128, 128 ], @@ -20,9 +20,9 @@ "cross_attention_norm": null, "decay": 0.9999, "down_block_types": [ - "DownBlock2D", "DownBlock2D", "CrossAttnDownBlock2D", + "CrossAttnDownBlock2D", "DownBlock2D" ], "downsample_padding": 1, @@ -57,7 +57,7 @@ "up_block_types": [ "UpBlock2D", "CrossAttnUpBlock2D", - "UpBlock2D", + "CrossAttnUpBlock2D", "UpBlock2D" ], "upcast_attention": false, diff --git a/checkpoint-1000/unet_ema/diffusion_pytorch_model.bin b/checkpoint-1000/unet_ema/diffusion_pytorch_model.bin index 0f821a231dc3d3165dbf736b8d351119b3853e22..85e7ee9751fc42e65eb1230498fddecea6bc8678 100644 --- a/checkpoint-1000/unet_ema/diffusion_pytorch_model.bin +++ b/checkpoint-1000/unet_ema/diffusion_pytorch_model.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:4606e9ae0d80dcc6f71737251b4ec6d2f6227c59a2f323de4251ef963c8f6da2 -size 36616081 +oid sha256:cbb8d48c9e57b7448804a7a5d7038f7240236cf4f20b41ac012f7defb4a4b4eb +size 62996305 diff --git a/checkpoint-130500/optimizer.bin b/checkpoint-130500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..eb8111859077920fa4f0518521a0d1a04acfafd9 --- /dev/null +++ b/checkpoint-130500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:31fa96551fe66c69d3d0dfcb76d96b2ac002bf4d08e29f327000fa1d9f26899e +size 73251413 diff --git a/checkpoint-130500/random_states_0.pkl b/checkpoint-130500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..8cd25e90c690e10389b259d00936a07c997aa2a3 --- /dev/null +++ b/checkpoint-130500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:0cebab60f9d55455aaaca590ea88a0b7c31b1af86d582b9a65db69f2524784aa +size 14727 diff --git a/checkpoint-130500/scaler.pt b/checkpoint-130500/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..a3e198653f4aee45e75ee1ace4db63f15cab94c0 --- /dev/null +++ b/checkpoint-130500/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7f9f749742d662432a42b45d794266f7dd7e412a892ecec5195e9fdfddd254da +size 557 diff --git a/checkpoint-130500/scheduler.bin b/checkpoint-130500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..4e9d161d27b80835b2412b73d35488bbdad158be --- /dev/null +++ b/checkpoint-130500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e6abb11e4951ffdcd40192dde9860cfd99c16ce80fbd45aa466dce04f18d2ba7 +size 563 diff --git a/checkpoint-130500/unet/config.json b/checkpoint-130500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..c2d7b8c0d490531013583af57a806d21893e494c --- /dev/null +++ b/checkpoint-130500/unet/config.json @@ -0,0 +1,60 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "attention_head_dim": 8, + "block_out_channels": [ + 64, + 64, + 128, + 128 + ], + "center_input_sample": false, + "class_embed_type": null, + "class_embeddings_concat": false, + "conv_in_kernel": 3, + "conv_out_kernel": 3, + "cross_attention_dim": 256, + "cross_attention_norm": null, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "CrossAttnDownBlock2D", + "DownBlock2D" + ], + "downsample_padding": 1, + "dual_cross_attention": false, + "encoder_hid_dim": null, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "layers_per_block": 2, + "mid_block_only_cross_attention": null, + "mid_block_scale_factor": 1, + "mid_block_type": "UNetMidBlock2DCrossAttn", + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "only_cross_attention": false, + "out_channels": 3, + "projection_class_embeddings_input_dim": null, + "resnet_out_scale_factor": 1.0, + "resnet_skip_time_act": false, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_cond_proj_dim": null, + "time_embedding_act_fn": null, + "time_embedding_dim": null, + "time_embedding_type": "positional", + "timestep_post_act": null, + "up_block_types": [ + "UpBlock2D", + "CrossAttnUpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "use_linear_projection": false +} diff --git a/checkpoint-130500/unet/diffusion_pytorch_model.bin b/checkpoint-130500/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..a7b8974a5a3a1706c3ce4e603bea727250c65d8b --- /dev/null +++ b/checkpoint-130500/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:889e5bc83ade517f3e4c0f88beed1a070d98f8beb282af191c13203cc7d3f3a5 +size 36621969 diff --git a/checkpoint-130500/unet_ema/config.json b/checkpoint-130500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..e126a418aa544035caf4a6faf026f45a8ccb7f22 --- /dev/null +++ b/checkpoint-130500/unet_ema/config.json @@ -0,0 +1,67 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "attention_head_dim": 8, + "block_out_channels": [ + 64, + 64, + 128, + 128 + ], + "center_input_sample": false, + "class_embed_type": null, + "class_embeddings_concat": false, + "conv_in_kernel": 3, + "conv_out_kernel": 3, + "cross_attention_dim": 256, + "cross_attention_norm": null, + "decay": 0.9999, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "CrossAttnDownBlock2D", + "DownBlock2D" + ], + "downsample_padding": 1, + "dual_cross_attention": false, + "encoder_hid_dim": null, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "inv_gamma": 1.0, + "layers_per_block": 2, + "mid_block_only_cross_attention": null, + "mid_block_scale_factor": 1, + "mid_block_type": "UNetMidBlock2DCrossAttn", + "min_decay": 0.0, + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "only_cross_attention": false, + "optimization_step": 130500, + "out_channels": 3, + "power": 0.75, + "projection_class_embeddings_input_dim": null, + "resnet_out_scale_factor": 1.0, + "resnet_skip_time_act": false, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_cond_proj_dim": null, + "time_embedding_act_fn": null, + "time_embedding_dim": null, + "time_embedding_type": "positional", + "timestep_post_act": null, + "up_block_types": [ + "UpBlock2D", + "CrossAttnUpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "update_after_step": 0, + "use_ema_warmup": true, + "use_linear_projection": false +} diff --git a/checkpoint-130500/unet_ema/diffusion_pytorch_model.bin b/checkpoint-130500/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..58a84ae8f350dd35d17b1c5e7205fa92a2d8d6ae --- /dev/null +++ b/checkpoint-130500/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:27fcc1c22895af12ddfc5c2c949370aa5d4a7880b8253e03a79479adb3e8f99b +size 36616081 diff --git a/checkpoint-131000/optimizer.bin b/checkpoint-131000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..b4534e1f4b607653030636feb49c101334f6e164 --- /dev/null +++ b/checkpoint-131000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:bcef8dc71af10ffd4c6942d41525ddf2d159e3b819b73b919092f82db81e11ce +size 73251413 diff --git a/checkpoint-131000/random_states_0.pkl b/checkpoint-131000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..65c1602bfabceaf029f3779231ca253783ba58aa --- /dev/null +++ b/checkpoint-131000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e4f52dddb12dd29e7e7262391f45f10bb16eed84937f1bc1d4e7f55d19dcb3cf +size 14727 diff --git a/checkpoint-131000/scaler.pt b/checkpoint-131000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..c94ce1b8e79dfc417142b2a05269b5eb15d9637e --- /dev/null +++ b/checkpoint-131000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c32970399230723622bf00c07af291c7305060776e4946dcb61c1269d75a765b +size 557 diff --git a/checkpoint-131000/scheduler.bin b/checkpoint-131000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..bd52ef1d2b76a1602cb024d5d5845b4bd29bf585 --- /dev/null +++ b/checkpoint-131000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:373a81fe2bfcc060cc4cf443d525d334a49034553fd2ff17b05c661769b0c2fd +size 563 diff --git a/checkpoint-131000/unet/config.json b/checkpoint-131000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..c2d7b8c0d490531013583af57a806d21893e494c --- /dev/null +++ b/checkpoint-131000/unet/config.json @@ -0,0 +1,60 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "attention_head_dim": 8, + "block_out_channels": [ + 64, + 64, + 128, + 128 + ], + "center_input_sample": false, + "class_embed_type": null, + "class_embeddings_concat": false, + "conv_in_kernel": 3, + "conv_out_kernel": 3, + "cross_attention_dim": 256, + "cross_attention_norm": null, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "CrossAttnDownBlock2D", + "DownBlock2D" + ], + "downsample_padding": 1, + "dual_cross_attention": false, + "encoder_hid_dim": null, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "layers_per_block": 2, + "mid_block_only_cross_attention": null, + "mid_block_scale_factor": 1, + "mid_block_type": "UNetMidBlock2DCrossAttn", + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "only_cross_attention": false, + "out_channels": 3, + "projection_class_embeddings_input_dim": null, + "resnet_out_scale_factor": 1.0, + "resnet_skip_time_act": false, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_cond_proj_dim": null, + "time_embedding_act_fn": null, + "time_embedding_dim": null, + "time_embedding_type": "positional", + "timestep_post_act": null, + "up_block_types": [ + "UpBlock2D", + "CrossAttnUpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "use_linear_projection": false +} diff --git a/checkpoint-131000/unet/diffusion_pytorch_model.bin b/checkpoint-131000/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..8b3230ecd65a1f3e0a4cdc725bcf2a51c8858a9f --- /dev/null +++ b/checkpoint-131000/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:69143fd319b5072cc3e55c91a8966b13ba3b3eb113469f4876a967965c16a66b +size 36621969 diff --git a/checkpoint-131000/unet_ema/config.json b/checkpoint-131000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..3aa5fea39aa8199466eb921e30311246b0a836de --- /dev/null +++ b/checkpoint-131000/unet_ema/config.json @@ -0,0 +1,67 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "attention_head_dim": 8, + "block_out_channels": [ + 64, + 64, + 128, + 128 + ], + "center_input_sample": false, + "class_embed_type": null, + "class_embeddings_concat": false, + "conv_in_kernel": 3, + "conv_out_kernel": 3, + "cross_attention_dim": 256, + "cross_attention_norm": null, + "decay": 0.9999, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "CrossAttnDownBlock2D", + "DownBlock2D" + ], + "downsample_padding": 1, + "dual_cross_attention": false, + "encoder_hid_dim": null, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "inv_gamma": 1.0, + "layers_per_block": 2, + "mid_block_only_cross_attention": null, + "mid_block_scale_factor": 1, + "mid_block_type": "UNetMidBlock2DCrossAttn", + "min_decay": 0.0, + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "only_cross_attention": false, + "optimization_step": 131000, + "out_channels": 3, + "power": 0.75, + "projection_class_embeddings_input_dim": null, + "resnet_out_scale_factor": 1.0, + "resnet_skip_time_act": false, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_cond_proj_dim": null, + "time_embedding_act_fn": null, + "time_embedding_dim": null, + "time_embedding_type": "positional", + "timestep_post_act": null, + "up_block_types": [ + "UpBlock2D", + "CrossAttnUpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "update_after_step": 0, + "use_ema_warmup": true, + "use_linear_projection": false +} diff --git a/checkpoint-131000/unet_ema/diffusion_pytorch_model.bin b/checkpoint-131000/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..e3a84125a59095935fbfe500bc168689a1eb4595 --- /dev/null +++ b/checkpoint-131000/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:8b6abe60ac42996b2f7e8bcf7cc151925f638f2afff503b28e346173b1f7e3a1 +size 36616081 diff --git a/checkpoint-131500/optimizer.bin b/checkpoint-131500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..ba7d25612f6f49eed478917fe6f37288248de0cb --- /dev/null +++ b/checkpoint-131500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:0c769c5c0a18143e94cb61d74b3f9fd996c131c055886e9689b8d4af0d49345f +size 73251413 diff --git a/checkpoint-131500/random_states_0.pkl b/checkpoint-131500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..316118499883ad0b3092aa62d89ab0b7a2b07623 --- /dev/null +++ b/checkpoint-131500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:752b8bca1c2441607778a36b1b9ffeeae8dcd7c3b38e705c0ef173bf89346ed9 +size 14727 diff --git a/checkpoint-131500/scaler.pt b/checkpoint-131500/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..65e40ff2a10c016991521ad40f9525750eaab635 --- /dev/null +++ b/checkpoint-131500/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:5372bd0819ae0a2b5a4497b7ed1b967dd8bcdb8f26557967c3f123cf89e4b713 +size 557 diff --git a/checkpoint-131500/scheduler.bin b/checkpoint-131500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..dc1869796808e14ec9a2f3c114dc59c70889a4f8 --- /dev/null +++ b/checkpoint-131500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:716d76137a36c6dc78a08c9a6c5c9b91d6d26960b84d669dd63549882510300e +size 563 diff --git a/checkpoint-131500/unet/config.json b/checkpoint-131500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..c2d7b8c0d490531013583af57a806d21893e494c --- /dev/null +++ b/checkpoint-131500/unet/config.json @@ -0,0 +1,60 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "attention_head_dim": 8, + "block_out_channels": [ + 64, + 64, + 128, + 128 + ], + "center_input_sample": false, + "class_embed_type": null, + "class_embeddings_concat": false, + "conv_in_kernel": 3, + "conv_out_kernel": 3, + "cross_attention_dim": 256, + "cross_attention_norm": null, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "CrossAttnDownBlock2D", + "DownBlock2D" + ], + "downsample_padding": 1, + "dual_cross_attention": false, + "encoder_hid_dim": null, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "layers_per_block": 2, + "mid_block_only_cross_attention": null, + "mid_block_scale_factor": 1, + "mid_block_type": "UNetMidBlock2DCrossAttn", + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "only_cross_attention": false, + "out_channels": 3, + "projection_class_embeddings_input_dim": null, + "resnet_out_scale_factor": 1.0, + "resnet_skip_time_act": false, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_cond_proj_dim": null, + "time_embedding_act_fn": null, + "time_embedding_dim": null, + "time_embedding_type": "positional", + "timestep_post_act": null, + "up_block_types": [ + "UpBlock2D", + "CrossAttnUpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "use_linear_projection": false +} diff --git a/checkpoint-131500/unet/diffusion_pytorch_model.bin b/checkpoint-131500/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..b71098f0e3b38de22d4cbffc1f44b71b083fcb74 --- /dev/null +++ b/checkpoint-131500/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:488612c1b95fb833be1b37ffa0f453e9731292834fb8813ef314a4ba16ee28c2 +size 36621969 diff --git a/checkpoint-131500/unet_ema/config.json b/checkpoint-131500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..89226adb91ccb85abe4c474966cda5da8262291b --- /dev/null +++ b/checkpoint-131500/unet_ema/config.json @@ -0,0 +1,67 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "attention_head_dim": 8, + "block_out_channels": [ + 64, + 64, + 128, + 128 + ], + "center_input_sample": false, + "class_embed_type": null, + "class_embeddings_concat": false, + "conv_in_kernel": 3, + "conv_out_kernel": 3, + "cross_attention_dim": 256, + "cross_attention_norm": null, + "decay": 0.9999, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "CrossAttnDownBlock2D", + "DownBlock2D" + ], + "downsample_padding": 1, + "dual_cross_attention": false, + "encoder_hid_dim": null, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "inv_gamma": 1.0, + "layers_per_block": 2, + "mid_block_only_cross_attention": null, + "mid_block_scale_factor": 1, + "mid_block_type": "UNetMidBlock2DCrossAttn", + "min_decay": 0.0, + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "only_cross_attention": false, + "optimization_step": 131500, + "out_channels": 3, + "power": 0.75, + "projection_class_embeddings_input_dim": null, + "resnet_out_scale_factor": 1.0, + "resnet_skip_time_act": false, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_cond_proj_dim": null, + "time_embedding_act_fn": null, + "time_embedding_dim": null, + "time_embedding_type": "positional", + "timestep_post_act": null, + "up_block_types": [ + "UpBlock2D", + "CrossAttnUpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "update_after_step": 0, + "use_ema_warmup": true, + "use_linear_projection": false +} diff --git a/checkpoint-131500/unet_ema/diffusion_pytorch_model.bin b/checkpoint-131500/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..c22e89ef694f4c1412c22f821b43feb6b577add9 --- /dev/null +++ b/checkpoint-131500/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:6912034fd19ff078ed0f978714b453c30d70cfbdd1457b6becffe1810f1d8595 +size 36616081 diff --git a/checkpoint-132000/optimizer.bin b/checkpoint-132000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..b868e76760b83cf9be1fbaf972a0d369178d07f4 --- /dev/null +++ b/checkpoint-132000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:0870cb21144eab8bb4ac74be561d8f6d9eb526c38a6c6406c771c5cc03a3b765 +size 73251413 diff --git a/checkpoint-132000/random_states_0.pkl b/checkpoint-132000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..759fb0dccdb60fce93c3ac8cc3ca00f1f6717406 --- /dev/null +++ b/checkpoint-132000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:bfbfd3bafc2c068bbfe5b8ae653c0fbdf0f9f941cbefa859461f38a6fc263edd +size 14727 diff --git a/checkpoint-132000/scaler.pt b/checkpoint-132000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..57303abd92ffe4e7b6278eaed82080ba8ea985fc --- /dev/null +++ b/checkpoint-132000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:de5dbe841727a40b3625ce083fd2228fd53e866092180e8a8f2f1e4cf92e3656 +size 557 diff --git a/checkpoint-132000/scheduler.bin b/checkpoint-132000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..530a983d0ec9a39d473924d11fd51faa3a20df88 --- /dev/null +++ b/checkpoint-132000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:97f99b962652d1ad0d2c65cce2c96bf72397968c5a175d22539f8c244440ae5d +size 563 diff --git a/checkpoint-132000/unet/config.json b/checkpoint-132000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..c2d7b8c0d490531013583af57a806d21893e494c --- /dev/null +++ b/checkpoint-132000/unet/config.json @@ -0,0 +1,60 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "attention_head_dim": 8, + "block_out_channels": [ + 64, + 64, + 128, + 128 + ], + "center_input_sample": false, + "class_embed_type": null, + "class_embeddings_concat": false, + "conv_in_kernel": 3, + "conv_out_kernel": 3, + "cross_attention_dim": 256, + "cross_attention_norm": null, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "CrossAttnDownBlock2D", + "DownBlock2D" + ], + "downsample_padding": 1, + "dual_cross_attention": false, + "encoder_hid_dim": null, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "layers_per_block": 2, + "mid_block_only_cross_attention": null, + "mid_block_scale_factor": 1, + "mid_block_type": "UNetMidBlock2DCrossAttn", + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "only_cross_attention": false, + "out_channels": 3, + "projection_class_embeddings_input_dim": null, + "resnet_out_scale_factor": 1.0, + "resnet_skip_time_act": false, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_cond_proj_dim": null, + "time_embedding_act_fn": null, + "time_embedding_dim": null, + "time_embedding_type": "positional", + "timestep_post_act": null, + "up_block_types": [ + "UpBlock2D", + "CrossAttnUpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "use_linear_projection": false +} diff --git a/checkpoint-132000/unet/diffusion_pytorch_model.bin b/checkpoint-132000/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..19ec3944f806b0249fc3877908765c81decaf1df --- /dev/null +++ b/checkpoint-132000/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:25016fe883a7dcc533d7ea8e6be560be9b40ae002251b304373ec69e1c47f2fc +size 36621969 diff --git a/checkpoint-132000/unet_ema/config.json b/checkpoint-132000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..2edccaacb9d06d043a6bfbe35f9ec5809e1046b3 --- /dev/null +++ b/checkpoint-132000/unet_ema/config.json @@ -0,0 +1,67 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "attention_head_dim": 8, + "block_out_channels": [ + 64, + 64, + 128, + 128 + ], + "center_input_sample": false, + "class_embed_type": null, + "class_embeddings_concat": false, + "conv_in_kernel": 3, + "conv_out_kernel": 3, + "cross_attention_dim": 256, + "cross_attention_norm": null, + "decay": 0.9999, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "CrossAttnDownBlock2D", + "DownBlock2D" + ], + "downsample_padding": 1, + "dual_cross_attention": false, + "encoder_hid_dim": null, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "inv_gamma": 1.0, + "layers_per_block": 2, + "mid_block_only_cross_attention": null, + "mid_block_scale_factor": 1, + "mid_block_type": "UNetMidBlock2DCrossAttn", + "min_decay": 0.0, + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "only_cross_attention": false, + "optimization_step": 132000, + "out_channels": 3, + "power": 0.75, + "projection_class_embeddings_input_dim": null, + "resnet_out_scale_factor": 1.0, + "resnet_skip_time_act": false, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_cond_proj_dim": null, + "time_embedding_act_fn": null, + "time_embedding_dim": null, + "time_embedding_type": "positional", + "timestep_post_act": null, + "up_block_types": [ + "UpBlock2D", + "CrossAttnUpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "update_after_step": 0, + "use_ema_warmup": true, + "use_linear_projection": false +} diff --git a/checkpoint-132000/unet_ema/diffusion_pytorch_model.bin b/checkpoint-132000/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..ad90ba5b753dcc99855ef727a70e86bfd0335b46 --- /dev/null +++ b/checkpoint-132000/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:ce2887263584f7b5e82e2ce827a984196f46baa01678f9f2f96a97add1eea9b1 +size 36616081 diff --git a/checkpoint-132500/optimizer.bin b/checkpoint-132500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..6f0ccd3f446791ed19d546b8478e54d9db53e81f --- /dev/null +++ b/checkpoint-132500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c89559f2131a61e139bc7ccdf3b5f79f383b385564aa307a95cbeeaa457476dd +size 73251413 diff --git a/checkpoint-132500/random_states_0.pkl b/checkpoint-132500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..f919a6b91d60b432184b13e35182488675ee624d --- /dev/null +++ b/checkpoint-132500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:385f0819b5a850b0b804e9ebdde2c115ee527a61d2146e14bdb58c839b2355f8 +size 14727 diff --git a/checkpoint-132500/scaler.pt b/checkpoint-132500/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..a4310065ea9c93680e1542868e63fb3b70223aac --- /dev/null +++ b/checkpoint-132500/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:bc90da5e0ffe9d477d01dd263420400914273acd530da5602f05166a1b6583ad +size 557 diff --git a/checkpoint-132500/scheduler.bin b/checkpoint-132500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..a588156cfdd4bc2156ebdc6951f23efb85082e82 --- /dev/null +++ b/checkpoint-132500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:55d6e9291bb491b68a58170af2546919dad42a3f43a973da573206015cb9942a +size 563 diff --git a/checkpoint-132500/unet/config.json b/checkpoint-132500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..c2d7b8c0d490531013583af57a806d21893e494c --- /dev/null +++ b/checkpoint-132500/unet/config.json @@ -0,0 +1,60 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "attention_head_dim": 8, + "block_out_channels": [ + 64, + 64, + 128, + 128 + ], + "center_input_sample": false, + "class_embed_type": null, + "class_embeddings_concat": false, + "conv_in_kernel": 3, + "conv_out_kernel": 3, + "cross_attention_dim": 256, + "cross_attention_norm": null, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "CrossAttnDownBlock2D", + "DownBlock2D" + ], + "downsample_padding": 1, + "dual_cross_attention": false, + "encoder_hid_dim": null, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "layers_per_block": 2, + "mid_block_only_cross_attention": null, + "mid_block_scale_factor": 1, + "mid_block_type": "UNetMidBlock2DCrossAttn", + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "only_cross_attention": false, + "out_channels": 3, + "projection_class_embeddings_input_dim": null, + "resnet_out_scale_factor": 1.0, + "resnet_skip_time_act": false, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_cond_proj_dim": null, + "time_embedding_act_fn": null, + "time_embedding_dim": null, + "time_embedding_type": "positional", + "timestep_post_act": null, + "up_block_types": [ + "UpBlock2D", + "CrossAttnUpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "use_linear_projection": false +} diff --git a/checkpoint-132500/unet/diffusion_pytorch_model.bin b/checkpoint-132500/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..f0dc14a6634523edafc076e20ddbeb100f7a48bb --- /dev/null +++ b/checkpoint-132500/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:f7001735996abef68c8e77293f9e5cecb207a5dd57629b02ba04a311a4600b14 +size 36621969 diff --git a/checkpoint-132500/unet_ema/config.json b/checkpoint-132500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..a6c58733219ea964aab77783f5f979d72852ed81 --- /dev/null +++ b/checkpoint-132500/unet_ema/config.json @@ -0,0 +1,67 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "attention_head_dim": 8, + "block_out_channels": [ + 64, + 64, + 128, + 128 + ], + "center_input_sample": false, + "class_embed_type": null, + "class_embeddings_concat": false, + "conv_in_kernel": 3, + "conv_out_kernel": 3, + "cross_attention_dim": 256, + "cross_attention_norm": null, + "decay": 0.9999, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "CrossAttnDownBlock2D", + "DownBlock2D" + ], + "downsample_padding": 1, + "dual_cross_attention": false, + "encoder_hid_dim": null, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "inv_gamma": 1.0, + "layers_per_block": 2, + "mid_block_only_cross_attention": null, + "mid_block_scale_factor": 1, + "mid_block_type": "UNetMidBlock2DCrossAttn", + "min_decay": 0.0, + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "only_cross_attention": false, + "optimization_step": 132500, + "out_channels": 3, + "power": 0.75, + "projection_class_embeddings_input_dim": null, + "resnet_out_scale_factor": 1.0, + "resnet_skip_time_act": false, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_cond_proj_dim": null, + "time_embedding_act_fn": null, + "time_embedding_dim": null, + "time_embedding_type": "positional", + "timestep_post_act": null, + "up_block_types": [ + "UpBlock2D", + "CrossAttnUpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "update_after_step": 0, + "use_ema_warmup": true, + "use_linear_projection": false +} diff --git a/checkpoint-132500/unet_ema/diffusion_pytorch_model.bin b/checkpoint-132500/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..2caee4b2d8254f711b6718bee732c95b678627ea --- /dev/null +++ b/checkpoint-132500/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:52683a906a09197f5658a8d27750b7b943143f64f75beed3eb19043a4d20605c +size 36616081 diff --git a/checkpoint-133000/optimizer.bin b/checkpoint-133000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..5dae6d7aa07244f63c49acb9d85e0e7202ad1392 --- /dev/null +++ b/checkpoint-133000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7db9bfc3babac207624c8eb5638f2e05bee0894810e9cdbebc792a9a243d4875 +size 73251413 diff --git a/checkpoint-133000/random_states_0.pkl b/checkpoint-133000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..4453880773ecdec790be9a93819dbfb6e17e303e --- /dev/null +++ b/checkpoint-133000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:8689c33b4ee59735478bdb695110e5c65e625eb75ce9440793ce99282ae2acc2 +size 14727 diff --git a/checkpoint-133000/scaler.pt b/checkpoint-133000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..438439cb8882b25102a6a4e13651030b61a3751c --- /dev/null +++ b/checkpoint-133000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:bdd22e5adafb1516fca80ef23b56c7d3862c341d569448f79dd37c9eb7f9bef3 +size 557 diff --git a/checkpoint-133000/scheduler.bin b/checkpoint-133000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..9b6ea9b17010ba570b07dbe29b2128571e18354a --- /dev/null +++ b/checkpoint-133000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:aa0b2f3a23a70944c9e8f1de4fae4e2f373cd280ba139838d03f22aaa2db99ed +size 563 diff --git a/checkpoint-133000/unet/config.json b/checkpoint-133000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..c2d7b8c0d490531013583af57a806d21893e494c --- /dev/null +++ b/checkpoint-133000/unet/config.json @@ -0,0 +1,60 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "attention_head_dim": 8, + "block_out_channels": [ + 64, + 64, + 128, + 128 + ], + "center_input_sample": false, + "class_embed_type": null, + "class_embeddings_concat": false, + "conv_in_kernel": 3, + "conv_out_kernel": 3, + "cross_attention_dim": 256, + "cross_attention_norm": null, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "CrossAttnDownBlock2D", + "DownBlock2D" + ], + "downsample_padding": 1, + "dual_cross_attention": false, + "encoder_hid_dim": null, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "layers_per_block": 2, + "mid_block_only_cross_attention": null, + "mid_block_scale_factor": 1, + "mid_block_type": "UNetMidBlock2DCrossAttn", + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "only_cross_attention": false, + "out_channels": 3, + "projection_class_embeddings_input_dim": null, + "resnet_out_scale_factor": 1.0, + "resnet_skip_time_act": false, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_cond_proj_dim": null, + "time_embedding_act_fn": null, + "time_embedding_dim": null, + "time_embedding_type": "positional", + "timestep_post_act": null, + "up_block_types": [ + "UpBlock2D", + "CrossAttnUpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "use_linear_projection": false +} diff --git a/checkpoint-133000/unet/diffusion_pytorch_model.bin b/checkpoint-133000/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..2ec4a1ca8b86c9d841fcbd1892b565717982c1a9 --- /dev/null +++ b/checkpoint-133000/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:9e6cc2cf0972cb94051e20ba1850eeb91eae77cde09a4e40ce45cc9bcc93e0ed +size 36621969 diff --git a/checkpoint-133000/unet_ema/config.json b/checkpoint-133000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..cdefc0090deebb70af70f9562473031d5a5a64ba --- /dev/null +++ b/checkpoint-133000/unet_ema/config.json @@ -0,0 +1,67 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "attention_head_dim": 8, + "block_out_channels": [ + 64, + 64, + 128, + 128 + ], + "center_input_sample": false, + "class_embed_type": null, + "class_embeddings_concat": false, + "conv_in_kernel": 3, + "conv_out_kernel": 3, + "cross_attention_dim": 256, + "cross_attention_norm": null, + "decay": 0.9999, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "CrossAttnDownBlock2D", + "DownBlock2D" + ], + "downsample_padding": 1, + "dual_cross_attention": false, + "encoder_hid_dim": null, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "inv_gamma": 1.0, + "layers_per_block": 2, + "mid_block_only_cross_attention": null, + "mid_block_scale_factor": 1, + "mid_block_type": "UNetMidBlock2DCrossAttn", + "min_decay": 0.0, + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "only_cross_attention": false, + "optimization_step": 133000, + "out_channels": 3, + "power": 0.75, + "projection_class_embeddings_input_dim": null, + "resnet_out_scale_factor": 1.0, + "resnet_skip_time_act": false, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_cond_proj_dim": null, + "time_embedding_act_fn": null, + "time_embedding_dim": null, + "time_embedding_type": "positional", + "timestep_post_act": null, + "up_block_types": [ + "UpBlock2D", + "CrossAttnUpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "update_after_step": 0, + "use_ema_warmup": true, + "use_linear_projection": false +} diff --git a/checkpoint-133000/unet_ema/diffusion_pytorch_model.bin b/checkpoint-133000/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..5c089e12f18c5cc9103b299203b939e0435de12e --- /dev/null +++ b/checkpoint-133000/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:a482033301f1440ec95fb92011984fc3c1207bafa3a5b486c5979a5108395071 +size 36616081 diff --git a/checkpoint-1500/optimizer.bin b/checkpoint-1500/optimizer.bin index 77c2cfd42f586296fc831438bfe99d2c4493f264..0f8e4a22218991e283de0b3fe3947a08c64be3f6 100644 --- a/checkpoint-1500/optimizer.bin +++ b/checkpoint-1500/optimizer.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:78af43f7e9a80c67091c2a19d461cc36ddf16e494fb3f4487297daf8edb1c9c5 -size 73251413 +oid sha256:f560ef510629780c2dcdce09225fd318c107aacfe93f1bd0956ee9126846392c +size 126008021 diff --git a/checkpoint-1500/random_states_0.pkl b/checkpoint-1500/random_states_0.pkl index af2b9b51dc1d1deb27fefe1a52e750b4ff02292b..3c12dfdf66ddc6c772823425c4bd35e33d524150 100644 --- a/checkpoint-1500/random_states_0.pkl +++ b/checkpoint-1500/random_states_0.pkl @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:71b85d110ecebb34006c6477bd5ba16bae6f5e0fef070687a0a545a02ae117aa -size 14727 +oid sha256:2b0c5e069cacc8cb9231371c916c9d28df1dee887cad0eb80a931e27de452a8a +size 14663 diff --git a/checkpoint-1500/scheduler.bin b/checkpoint-1500/scheduler.bin index 24d68868adaf2f229b1ddb3030ff6f5b1960ec7d..cfc5284116b1b5623cafdbc7de1f6f9b502ddb71 100644 --- a/checkpoint-1500/scheduler.bin +++ b/checkpoint-1500/scheduler.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:1f8d9e742b2369c72be9812e9941b1dfb14b93ec64cf59100eae509b08367944 +oid sha256:06ecdeeadd11cce6878f3642d62737a72674866d6e35d37cf331845c50f9d6f8 size 563 diff --git a/checkpoint-1500/unet/config.json b/checkpoint-1500/unet/config.json index c2d7b8c0d490531013583af57a806d21893e494c..06fb387aa3ed37ab3b2f455b5469059d55604f74 100644 --- a/checkpoint-1500/unet/config.json +++ b/checkpoint-1500/unet/config.json @@ -6,8 +6,8 @@ "addition_embed_type_num_heads": 64, "attention_head_dim": 8, "block_out_channels": [ - 64, - 64, + 128, + 128, 128, 128 ], @@ -19,9 +19,9 @@ "cross_attention_dim": 256, "cross_attention_norm": null, "down_block_types": [ - "DownBlock2D", "DownBlock2D", "CrossAttnDownBlock2D", + "CrossAttnDownBlock2D", "DownBlock2D" ], "downsample_padding": 1, @@ -52,7 +52,7 @@ "up_block_types": [ "UpBlock2D", "CrossAttnUpBlock2D", - "UpBlock2D", + "CrossAttnUpBlock2D", "UpBlock2D" ], "upcast_attention": false, diff --git a/checkpoint-1500/unet/diffusion_pytorch_model.bin b/checkpoint-1500/unet/diffusion_pytorch_model.bin index 54772604bb0a5462b174c486044568d26e544dc0..7db57a7a2e92920be7d6a2865651a8a7cc47efb6 100644 --- a/checkpoint-1500/unet/diffusion_pytorch_model.bin +++ b/checkpoint-1500/unet/diffusion_pytorch_model.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:4482fd8d418528814d76200bd2d4f1917f202ea28f1591a4895df86742ba2143 -size 36621969 +oid sha256:2d8cf7d7fc0a2f532e4e0139ed65418e34b9164f19f2ddf17bd6ab2842370a8d +size 63003985 diff --git a/checkpoint-1500/unet_ema/config.json b/checkpoint-1500/unet_ema/config.json index 8c13e4f3333faa2edd25ac460e6461a8156b394f..fc45df8ea9a2147746845917b2b36ee3083c1418 100644 --- a/checkpoint-1500/unet_ema/config.json +++ b/checkpoint-1500/unet_ema/config.json @@ -6,8 +6,8 @@ "addition_embed_type_num_heads": 64, "attention_head_dim": 8, "block_out_channels": [ - 64, - 64, + 128, + 128, 128, 128 ], @@ -20,9 +20,9 @@ "cross_attention_norm": null, "decay": 0.9999, "down_block_types": [ - "DownBlock2D", "DownBlock2D", "CrossAttnDownBlock2D", + "CrossAttnDownBlock2D", "DownBlock2D" ], "downsample_padding": 1, @@ -57,7 +57,7 @@ "up_block_types": [ "UpBlock2D", "CrossAttnUpBlock2D", - "UpBlock2D", + "CrossAttnUpBlock2D", "UpBlock2D" ], "upcast_attention": false, diff --git a/checkpoint-1500/unet_ema/diffusion_pytorch_model.bin b/checkpoint-1500/unet_ema/diffusion_pytorch_model.bin index ddbd9e74fe33c5edd6da43776e4c60afacbf0558..b52e94c085eec4a75b0b1bbed3abe4d8ff30ca2d 100644 --- a/checkpoint-1500/unet_ema/diffusion_pytorch_model.bin +++ b/checkpoint-1500/unet_ema/diffusion_pytorch_model.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:052155aa218ad8a4d3e6cd9fe272872e77beebaeceb8242459a631e35fe35f7d -size 36616081 +oid sha256:de7af0891025e01d98fdde103a4a5427256f6aa54ba0e4b9c618e29e086e15fe +size 62996305 diff --git a/checkpoint-2000/optimizer.bin b/checkpoint-2000/optimizer.bin index e33184e2f6333a75a803c8298c13fe0f0cb951cb..b5b1fb765b4e3503ccdbb26600594098e25a1d85 100644 --- a/checkpoint-2000/optimizer.bin +++ b/checkpoint-2000/optimizer.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:c37fd7abf6109d2efad550ced22bfce6abba3e849515ecb452cc35e27875353b -size 73251413 +oid sha256:513e1106b73943fbc7f7c51eb376f0ef51394415e25c68389135232947a0cd35 +size 126008021 diff --git a/checkpoint-2000/random_states_0.pkl b/checkpoint-2000/random_states_0.pkl index 10b81428a8b1250d91301099c1d5c70d4e8d3b76..bf4efafc65d5ae50c55676f3bcc54e7166b4d1b1 100644 --- a/checkpoint-2000/random_states_0.pkl +++ b/checkpoint-2000/random_states_0.pkl @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:22d1de8a8ec14828b46576a3ec62cb8a43988d937df15edf275ac63c5895c8c5 -size 14727 +oid sha256:3ef393fe21486a202fe3aa1ec46bf3a2737cd1f721bf9593957fd7cbf3278004 +size 14663 diff --git a/checkpoint-2000/scheduler.bin b/checkpoint-2000/scheduler.bin index bc8da3e4d6b1057a3d0f3044878f957c3e0f5561..78874e651a0317f2cc51687312e215355dc0650f 100644 --- a/checkpoint-2000/scheduler.bin +++ b/checkpoint-2000/scheduler.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:331970d63bb3fbb5cb88f00b7021230535edf1c63e1881d0ea078f4df652afc0 +oid sha256:0dc5152ca8f051f2afe3ab72f0ea85c90133dde5b0e603d389d8006b9832e141 size 563 diff --git a/checkpoint-2000/unet/config.json b/checkpoint-2000/unet/config.json index c2d7b8c0d490531013583af57a806d21893e494c..06fb387aa3ed37ab3b2f455b5469059d55604f74 100644 --- a/checkpoint-2000/unet/config.json +++ b/checkpoint-2000/unet/config.json @@ -6,8 +6,8 @@ "addition_embed_type_num_heads": 64, "attention_head_dim": 8, "block_out_channels": [ - 64, - 64, + 128, + 128, 128, 128 ], @@ -19,9 +19,9 @@ "cross_attention_dim": 256, "cross_attention_norm": null, "down_block_types": [ - "DownBlock2D", "DownBlock2D", "CrossAttnDownBlock2D", + "CrossAttnDownBlock2D", "DownBlock2D" ], "downsample_padding": 1, @@ -52,7 +52,7 @@ "up_block_types": [ "UpBlock2D", "CrossAttnUpBlock2D", - "UpBlock2D", + "CrossAttnUpBlock2D", "UpBlock2D" ], "upcast_attention": false, diff --git a/checkpoint-2000/unet/diffusion_pytorch_model.bin b/checkpoint-2000/unet/diffusion_pytorch_model.bin index ce6c5a4c9f45621fda2fc7d6ed99fcd4cc429bfe..9dc311bc3cbb4209e62ba15c41a0e6866b4bb55c 100644 --- a/checkpoint-2000/unet/diffusion_pytorch_model.bin +++ b/checkpoint-2000/unet/diffusion_pytorch_model.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:955f117ef9c585415c5d84be263bc1a6cdc9cfa98a9cc604e62b642ee8c04763 -size 36621969 +oid sha256:6f5681194fc39bfaf1061f2a761e68816ffa2301864b64777f2384511020a538 +size 63003985 diff --git a/checkpoint-2000/unet_ema/config.json b/checkpoint-2000/unet_ema/config.json index 896e1557de3eda22484c5592b6e518b02f48dff8..a554a9e6ab76e9c27177d79b0ff0bd983be33555 100644 --- a/checkpoint-2000/unet_ema/config.json +++ b/checkpoint-2000/unet_ema/config.json @@ -6,8 +6,8 @@ "addition_embed_type_num_heads": 64, "attention_head_dim": 8, "block_out_channels": [ - 64, - 64, + 128, + 128, 128, 128 ], @@ -20,9 +20,9 @@ "cross_attention_norm": null, "decay": 0.9999, "down_block_types": [ - "DownBlock2D", "DownBlock2D", "CrossAttnDownBlock2D", + "CrossAttnDownBlock2D", "DownBlock2D" ], "downsample_padding": 1, @@ -57,7 +57,7 @@ "up_block_types": [ "UpBlock2D", "CrossAttnUpBlock2D", - "UpBlock2D", + "CrossAttnUpBlock2D", "UpBlock2D" ], "upcast_attention": false, diff --git a/checkpoint-2000/unet_ema/diffusion_pytorch_model.bin b/checkpoint-2000/unet_ema/diffusion_pytorch_model.bin index 379510ba80c06246268152d12edd8576326788c8..97b134722865c4a7d26977fb83df55ca6c40c1ae 100644 --- a/checkpoint-2000/unet_ema/diffusion_pytorch_model.bin +++ b/checkpoint-2000/unet_ema/diffusion_pytorch_model.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:dc6d2ec6022c625c675b6bfb318bf484fc57105fa507b4046a87f77475a51750 -size 36616081 +oid sha256:d4e10c1f58a05785516c7c5943a4053b7ff9088230b0c58ea232697c66a6b1c5 +size 62996305 diff --git a/checkpoint-2500/optimizer.bin b/checkpoint-2500/optimizer.bin index 686df5d203a914f97c9636c0e28cffa23112508b..ddd3014fbd61802cd84e31cb22fa0adefdcd9f62 100644 --- a/checkpoint-2500/optimizer.bin +++ b/checkpoint-2500/optimizer.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:d3f728e5f3601a90121b68aef319e9d3ce1065f5ad7d3bd5c4825a65a6729cf5 -size 73251413 +oid sha256:582233fed70e53e3eade3be7cc06223256d491719df57dab2570de17d643c3ec +size 126008021 diff --git a/checkpoint-2500/random_states_0.pkl b/checkpoint-2500/random_states_0.pkl index 17ace97ccae508585366953516d97cd7dca16ff0..998dd7f6e6d8d13072213b23a0b01c735bebf4ee 100644 --- a/checkpoint-2500/random_states_0.pkl +++ b/checkpoint-2500/random_states_0.pkl @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:eaa67a53e258157e5377546e2e8403b6d6153a8b8b6c489cdfbfe3ff50fa85e0 -size 14727 +oid sha256:43bc9e6c841bfa2583fe5e6c2cf6eb3e5c53e1de3edf000036d89f59af6b1812 +size 14663 diff --git a/checkpoint-2500/scheduler.bin b/checkpoint-2500/scheduler.bin index b16b828405c381a20d3f4f6e3c0f5728967dec00..f5fd71ac749fa5ef18ff16db030e292c82ee6ca8 100644 --- a/checkpoint-2500/scheduler.bin +++ b/checkpoint-2500/scheduler.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:d0a322c87567d0df423cd81f0153864f5557e9f5870c2dac859a4c34dcca9d02 +oid sha256:00d01d5e469cd0e6f4ee6fe64e003202743208df0d0e839a792f3dc7be165281 size 563 diff --git a/checkpoint-2500/unet/config.json b/checkpoint-2500/unet/config.json index c2d7b8c0d490531013583af57a806d21893e494c..06fb387aa3ed37ab3b2f455b5469059d55604f74 100644 --- a/checkpoint-2500/unet/config.json +++ b/checkpoint-2500/unet/config.json @@ -6,8 +6,8 @@ "addition_embed_type_num_heads": 64, "attention_head_dim": 8, "block_out_channels": [ - 64, - 64, + 128, + 128, 128, 128 ], @@ -19,9 +19,9 @@ "cross_attention_dim": 256, "cross_attention_norm": null, "down_block_types": [ - "DownBlock2D", "DownBlock2D", "CrossAttnDownBlock2D", + "CrossAttnDownBlock2D", "DownBlock2D" ], "downsample_padding": 1, @@ -52,7 +52,7 @@ "up_block_types": [ "UpBlock2D", "CrossAttnUpBlock2D", - "UpBlock2D", + "CrossAttnUpBlock2D", "UpBlock2D" ], "upcast_attention": false, diff --git a/checkpoint-2500/unet/diffusion_pytorch_model.bin b/checkpoint-2500/unet/diffusion_pytorch_model.bin index 6fa6608563ee06030381133926f658aadc580229..1aec4bcc4d84e22afa8c54f01ad7332de781ea62 100644 --- a/checkpoint-2500/unet/diffusion_pytorch_model.bin +++ b/checkpoint-2500/unet/diffusion_pytorch_model.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:22187f011302b3274b18955c0ed82ae4572b614afba45407be5627f8517b52bf -size 36621969 +oid sha256:35a41254c60de4f332e181be9fe0f3fd0713879103d7bdc67f1e7fe124db8f76 +size 63003985 diff --git a/checkpoint-2500/unet_ema/config.json b/checkpoint-2500/unet_ema/config.json index 89580b730f49d569b290563dcb0946c0d4072db3..be2dee609175cc75fad491d5b0b0af3b6bfab28d 100644 --- a/checkpoint-2500/unet_ema/config.json +++ b/checkpoint-2500/unet_ema/config.json @@ -6,8 +6,8 @@ "addition_embed_type_num_heads": 64, "attention_head_dim": 8, "block_out_channels": [ - 64, - 64, + 128, + 128, 128, 128 ], @@ -20,9 +20,9 @@ "cross_attention_norm": null, "decay": 0.9999, "down_block_types": [ - "DownBlock2D", "DownBlock2D", "CrossAttnDownBlock2D", + "CrossAttnDownBlock2D", "DownBlock2D" ], "downsample_padding": 1, @@ -57,7 +57,7 @@ "up_block_types": [ "UpBlock2D", "CrossAttnUpBlock2D", - "UpBlock2D", + "CrossAttnUpBlock2D", "UpBlock2D" ], "upcast_attention": false, diff --git a/checkpoint-2500/unet_ema/diffusion_pytorch_model.bin b/checkpoint-2500/unet_ema/diffusion_pytorch_model.bin index af9b2bf1e39dcd0884bf3482f01d121551c80b21..b0a90ead6e197ee4e86772315fec46d769befcd0 100644 --- a/checkpoint-2500/unet_ema/diffusion_pytorch_model.bin +++ b/checkpoint-2500/unet_ema/diffusion_pytorch_model.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:3f860bfccd883a7894d8e10e03df1d8f8ce10b1008f3ff82428d64132ea82ec5 -size 36616081 +oid sha256:7930378e5e7f015a41c87a964492394ab8d75b4b8edd70a1f883096425b5bcdb +size 62996305 diff --git a/checkpoint-3000/optimizer.bin b/checkpoint-3000/optimizer.bin index 9120c8b666de5f58fb35390556f376f789603e0e..7bd97f1f29c2b34334f1f1734086679cfeb3c26f 100644 --- a/checkpoint-3000/optimizer.bin +++ b/checkpoint-3000/optimizer.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:7967ad457073d2188337fd5f5789522ce80236537bdb3275e5d72289b61e1664 -size 73251413 +oid sha256:cc0d2c745527b4d4c4e8d7dd88197f657324d4cddf727488e75714171cdab524 +size 126008021 diff --git a/checkpoint-3000/random_states_0.pkl b/checkpoint-3000/random_states_0.pkl index 8ef4add7bd603f8508059c4638e4e6d5ec96c5f5..e5c298d654608b9fa859d30c9060b51412852b2e 100644 --- a/checkpoint-3000/random_states_0.pkl +++ b/checkpoint-3000/random_states_0.pkl @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:be504a9bda7f68387d5556d92ef0a68f59bfed0fb96aac43ee52c70ccc17f589 -size 14727 +oid sha256:1835a8b7fb0812c9df5b8316f90643e9b9cdad75d94da99e1e45d5ad15935b7d +size 14663 diff --git a/checkpoint-3000/scheduler.bin b/checkpoint-3000/scheduler.bin index 232e78e92ad81cf6a616cda8bfa85feff7b7de83..272dbc924dfb0b62758a37c311e2a05024de04b8 100644 --- a/checkpoint-3000/scheduler.bin +++ b/checkpoint-3000/scheduler.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:08e49e2597d43758a2a73217953fc47ebb33745db0e58de998b3984fc63c026c +oid sha256:081be089c5756ea89ac4cdd97385790fa3f7f03c874cc4a260cbd08bce2c8bf0 size 563 diff --git a/checkpoint-3000/unet/config.json b/checkpoint-3000/unet/config.json index c2d7b8c0d490531013583af57a806d21893e494c..06fb387aa3ed37ab3b2f455b5469059d55604f74 100644 --- a/checkpoint-3000/unet/config.json +++ b/checkpoint-3000/unet/config.json @@ -6,8 +6,8 @@ "addition_embed_type_num_heads": 64, "attention_head_dim": 8, "block_out_channels": [ - 64, - 64, + 128, + 128, 128, 128 ], @@ -19,9 +19,9 @@ "cross_attention_dim": 256, "cross_attention_norm": null, "down_block_types": [ - "DownBlock2D", "DownBlock2D", "CrossAttnDownBlock2D", + "CrossAttnDownBlock2D", "DownBlock2D" ], "downsample_padding": 1, @@ -52,7 +52,7 @@ "up_block_types": [ "UpBlock2D", "CrossAttnUpBlock2D", - "UpBlock2D", + "CrossAttnUpBlock2D", "UpBlock2D" ], "upcast_attention": false, diff --git a/checkpoint-3000/unet/diffusion_pytorch_model.bin b/checkpoint-3000/unet/diffusion_pytorch_model.bin index 1aa2c0fe2308b1d0f77ad984973590b0911bd226..12ef281b5310d2a33035fbc56aae79a6a41cb0bf 100644 --- a/checkpoint-3000/unet/diffusion_pytorch_model.bin +++ b/checkpoint-3000/unet/diffusion_pytorch_model.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:76fd1d99ea33b94a270a163595c1bf3827fbe265ad7b840ff6beda420ebdb19a -size 36621969 +oid sha256:b3565844d7d282231cc7b9261884983dee08cbdfb830000be8a1e0e6c97cad1d +size 63003985 diff --git a/checkpoint-3000/unet_ema/config.json b/checkpoint-3000/unet_ema/config.json index c85928f50f1c665362c34d41eb94f7aba4e58997..2e3616833e61ec6afdc3f5b47317609ed0849ad9 100644 --- a/checkpoint-3000/unet_ema/config.json +++ b/checkpoint-3000/unet_ema/config.json @@ -6,8 +6,8 @@ "addition_embed_type_num_heads": 64, "attention_head_dim": 8, "block_out_channels": [ - 64, - 64, + 128, + 128, 128, 128 ], @@ -20,9 +20,9 @@ "cross_attention_norm": null, "decay": 0.9999, "down_block_types": [ - "DownBlock2D", "DownBlock2D", "CrossAttnDownBlock2D", + "CrossAttnDownBlock2D", "DownBlock2D" ], "downsample_padding": 1, @@ -57,7 +57,7 @@ "up_block_types": [ "UpBlock2D", "CrossAttnUpBlock2D", - "UpBlock2D", + "CrossAttnUpBlock2D", "UpBlock2D" ], "upcast_attention": false, diff --git a/checkpoint-3000/unet_ema/diffusion_pytorch_model.bin b/checkpoint-3000/unet_ema/diffusion_pytorch_model.bin index eb54db08ee4bb85e7ae73a30ebc679a72100046e..b65a19836558d9f05d4adc5513c57df9563ed896 100644 --- a/checkpoint-3000/unet_ema/diffusion_pytorch_model.bin +++ b/checkpoint-3000/unet_ema/diffusion_pytorch_model.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:d8ab5e51c21e372eaa3b56d698efdf84f7a3d53b506a00ca7df4f4b79058fa4b -size 36616081 +oid sha256:5b4e4430bc72643009e6667a7a0d048417cf759265f6fcb3d2fdf2f9fc5b267d +size 62996305 diff --git a/checkpoint-3500/optimizer.bin b/checkpoint-3500/optimizer.bin index c3f292479f8ef28f8a01678ce519c28a96f099ae..32dd3a2426cecbe73f2afa998e5e510a0f69d701 100644 --- a/checkpoint-3500/optimizer.bin +++ b/checkpoint-3500/optimizer.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:b79e26d695afddf1ac0f4d5a74a0d43d695fc626fb2c037a74f33e1bd1b8bb34 -size 73251413 +oid sha256:fa052fb33da22baf2178faed4348b77c09920e560abaa2c6d7985e5c6b4f93c7 +size 126008021 diff --git a/checkpoint-3500/random_states_0.pkl b/checkpoint-3500/random_states_0.pkl index b2123a9b60c20690a37cfe1122bd6aa5bae66096..f914347d558748ec441f0ac890458eb9c6fe1996 100644 --- a/checkpoint-3500/random_states_0.pkl +++ b/checkpoint-3500/random_states_0.pkl @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:29c3acb9050536c76e2da3d094e196d7152ba0a8e90a296b872209d3538f79df -size 14727 +oid sha256:eb1d014e174663dcf1968fbcbc31a0fea4d573f2d091d72f58e65bbd74240ca0 +size 14663 diff --git a/checkpoint-3500/scheduler.bin b/checkpoint-3500/scheduler.bin index 4c1266c37e090229987fc84c4d9ebe753481fd61..384f3901d2ef3c4a56e74c218336725323778b67 100644 --- a/checkpoint-3500/scheduler.bin +++ b/checkpoint-3500/scheduler.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:9093e18efca55fa5b8c446a5ccf8579309bcfe1793ee32b99ea135f26be19906 +oid sha256:307ffe68e01fc663601188447a91da1bfe0285e3513ce94df066351bf6c1a0c9 size 563 diff --git a/checkpoint-3500/unet/config.json b/checkpoint-3500/unet/config.json index c2d7b8c0d490531013583af57a806d21893e494c..06fb387aa3ed37ab3b2f455b5469059d55604f74 100644 --- a/checkpoint-3500/unet/config.json +++ b/checkpoint-3500/unet/config.json @@ -6,8 +6,8 @@ "addition_embed_type_num_heads": 64, "attention_head_dim": 8, "block_out_channels": [ - 64, - 64, + 128, + 128, 128, 128 ], @@ -19,9 +19,9 @@ "cross_attention_dim": 256, "cross_attention_norm": null, "down_block_types": [ - "DownBlock2D", "DownBlock2D", "CrossAttnDownBlock2D", + "CrossAttnDownBlock2D", "DownBlock2D" ], "downsample_padding": 1, @@ -52,7 +52,7 @@ "up_block_types": [ "UpBlock2D", "CrossAttnUpBlock2D", - "UpBlock2D", + "CrossAttnUpBlock2D", "UpBlock2D" ], "upcast_attention": false, diff --git a/checkpoint-3500/unet/diffusion_pytorch_model.bin b/checkpoint-3500/unet/diffusion_pytorch_model.bin index 08336989081f7b475d398f520684558a3c2444b4..d0a4ed27392f359675b2f5b57357f2843b97e62d 100644 --- a/checkpoint-3500/unet/diffusion_pytorch_model.bin +++ b/checkpoint-3500/unet/diffusion_pytorch_model.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:bc32070562556d6682d0d0c5e7a39ce3ad79d9e9d6601bffde4837efdbdb41aa -size 36621969 +oid sha256:159b065e769dd84a6fc1497e479d840632f9c9243462b3488015b91dae355a4d +size 63003985 diff --git a/checkpoint-3500/unet_ema/config.json b/checkpoint-3500/unet_ema/config.json index d7008745ec6ba7294506202c588b98c82a3994a7..076f6493b91a022df6b8451fa1a11836bd8382ab 100644 --- a/checkpoint-3500/unet_ema/config.json +++ b/checkpoint-3500/unet_ema/config.json @@ -6,8 +6,8 @@ "addition_embed_type_num_heads": 64, "attention_head_dim": 8, "block_out_channels": [ - 64, - 64, + 128, + 128, 128, 128 ], @@ -20,9 +20,9 @@ "cross_attention_norm": null, "decay": 0.9999, "down_block_types": [ - "DownBlock2D", "DownBlock2D", "CrossAttnDownBlock2D", + "CrossAttnDownBlock2D", "DownBlock2D" ], "downsample_padding": 1, @@ -57,7 +57,7 @@ "up_block_types": [ "UpBlock2D", "CrossAttnUpBlock2D", - "UpBlock2D", + "CrossAttnUpBlock2D", "UpBlock2D" ], "upcast_attention": false, diff --git a/checkpoint-3500/unet_ema/diffusion_pytorch_model.bin b/checkpoint-3500/unet_ema/diffusion_pytorch_model.bin index 00b40a5bfdb9687b2ba105c38618ed28faf39668..3bf8a29d0067bfb37140f83b9d4511e8f9d53909 100644 --- a/checkpoint-3500/unet_ema/diffusion_pytorch_model.bin +++ b/checkpoint-3500/unet_ema/diffusion_pytorch_model.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:d560ae76fd1e16b85f42be5f4a1ff4de2a04a3e424005851fd5792ba2317b6d1 -size 36616081 +oid sha256:4026276d5341dc5d6d5ab9689c5ce75cd82cea9014afcc002f4e2f5a198ec87c +size 62996305 diff --git a/checkpoint-4000/optimizer.bin b/checkpoint-4000/optimizer.bin index f3181dc84c363ec4d05b20639262a2f87fc9fbb7..928f961d06aa485f433a8914177991967e3eeda6 100644 --- a/checkpoint-4000/optimizer.bin +++ b/checkpoint-4000/optimizer.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:b36a7b0291be393e378c0c72e5aef4afb16c2bf36d3e5110aa6274563e0d8e8a -size 73251413 +oid sha256:4143d0b78d7a1d51a4131e2a292b210d5d690db5e8f565949cb12d808df29439 +size 126008021 diff --git a/checkpoint-4000/random_states_0.pkl b/checkpoint-4000/random_states_0.pkl index 83d4ef5ee2f7738490271f927f718fc83ee60fa1..921c8e55684e0cc0f26e58ad7757acacedf8317d 100644 --- a/checkpoint-4000/random_states_0.pkl +++ b/checkpoint-4000/random_states_0.pkl @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:fbb41d5bdbe32246f67c87dcb5b00284b3045eb60149deba939fc1966cd7553c -size 14727 +oid sha256:8e9ab7fa76ac92ac52b2d670d932b7323a23e6f34cd6df8984d2e150704105d6 +size 14663 diff --git a/checkpoint-4000/scheduler.bin b/checkpoint-4000/scheduler.bin index 18678f5bde86f2f1c360c923d3f66657d6985ea3..d2f47966aeefbbf667e0580b9fec5459425f435d 100644 --- a/checkpoint-4000/scheduler.bin +++ b/checkpoint-4000/scheduler.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:9a8c2cf905aadd885e3e32e4a24a4ef0d5248c62354b5d14da4aa63d6caf0331 +oid sha256:1b75b687afe3b478ecbd691ccca02c20f1efef9103007fdd65fe81009853f7de size 563 diff --git a/checkpoint-4000/unet/config.json b/checkpoint-4000/unet/config.json index c2d7b8c0d490531013583af57a806d21893e494c..06fb387aa3ed37ab3b2f455b5469059d55604f74 100644 --- a/checkpoint-4000/unet/config.json +++ b/checkpoint-4000/unet/config.json @@ -6,8 +6,8 @@ "addition_embed_type_num_heads": 64, "attention_head_dim": 8, "block_out_channels": [ - 64, - 64, + 128, + 128, 128, 128 ], @@ -19,9 +19,9 @@ "cross_attention_dim": 256, "cross_attention_norm": null, "down_block_types": [ - "DownBlock2D", "DownBlock2D", "CrossAttnDownBlock2D", + "CrossAttnDownBlock2D", "DownBlock2D" ], "downsample_padding": 1, @@ -52,7 +52,7 @@ "up_block_types": [ "UpBlock2D", "CrossAttnUpBlock2D", - "UpBlock2D", + "CrossAttnUpBlock2D", "UpBlock2D" ], "upcast_attention": false, diff --git a/checkpoint-4000/unet/diffusion_pytorch_model.bin b/checkpoint-4000/unet/diffusion_pytorch_model.bin index 01cf4ea3fcc36fd0d5dd554d93f8c5d3642a6177..3c20757749d3feb67e6e6efbcaadd7987f7cf6da 100644 --- a/checkpoint-4000/unet/diffusion_pytorch_model.bin +++ b/checkpoint-4000/unet/diffusion_pytorch_model.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:c90c99333d4a312611ae1995a3e6e85b53b8f9d79b77cf0ed401a14e9380dc5d -size 36621969 +oid sha256:e91814d7b860c63124baf6e7c6a20c97232a4722785d8869848f187e6b4355fe +size 63003985 diff --git a/checkpoint-4000/unet_ema/config.json b/checkpoint-4000/unet_ema/config.json index 6d6b48549faf1d927e663601071ad54de88cb231..eec2757dd7352bb4f312ff346022e9296231e718 100644 --- a/checkpoint-4000/unet_ema/config.json +++ b/checkpoint-4000/unet_ema/config.json @@ -6,8 +6,8 @@ "addition_embed_type_num_heads": 64, "attention_head_dim": 8, "block_out_channels": [ - 64, - 64, + 128, + 128, 128, 128 ], @@ -20,9 +20,9 @@ "cross_attention_norm": null, "decay": 0.9999, "down_block_types": [ - "DownBlock2D", "DownBlock2D", "CrossAttnDownBlock2D", + "CrossAttnDownBlock2D", "DownBlock2D" ], "downsample_padding": 1, @@ -57,7 +57,7 @@ "up_block_types": [ "UpBlock2D", "CrossAttnUpBlock2D", - "UpBlock2D", + "CrossAttnUpBlock2D", "UpBlock2D" ], "upcast_attention": false, diff --git a/checkpoint-4000/unet_ema/diffusion_pytorch_model.bin b/checkpoint-4000/unet_ema/diffusion_pytorch_model.bin index e7148fbd624e9f6d9752c8fe8b46007bcc6fc902..49bd667e0f7a01fae462a0fcbbca9a7268891537 100644 --- a/checkpoint-4000/unet_ema/diffusion_pytorch_model.bin +++ b/checkpoint-4000/unet_ema/diffusion_pytorch_model.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:df2854aacb470f9e57bf9f185f8e54a82ba64b86198819f584737869c8a2c9a9 -size 36616081 +oid sha256:9e695dd4f1ddf2f1ec9b2209a1021688d28780476060ef0a8c70b7499cf1191e +size 62996305 diff --git a/checkpoint-4500/optimizer.bin b/checkpoint-4500/optimizer.bin index cf192cb049a8138e6e9a9f9de8e1bdea03e1f7fe..30bcc33c28f38d4c3ca0f6303ad3088972980a79 100644 --- a/checkpoint-4500/optimizer.bin +++ b/checkpoint-4500/optimizer.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:616e2d478fb70a6a07fe37c82657236bc09184410b4a86899b3e1f0872f23d2f -size 73251413 +oid sha256:7737fd8be6be57290dc085ddfd7fb5728a3605b75507b882c073e7abc12bc3d7 +size 126008021 diff --git a/checkpoint-4500/random_states_0.pkl b/checkpoint-4500/random_states_0.pkl index b854d9416cf000131e899df2bc96b4da67d65929..ec2b3675d107c263b7508e3c395995162bc11865 100644 --- a/checkpoint-4500/random_states_0.pkl +++ b/checkpoint-4500/random_states_0.pkl @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:115544717e8801c1d0307efeab8151bfdbf4e45e57b0934a3a71591d413ac7df -size 14727 +oid sha256:fe07f85abacbdb48eaea9fd12ea4049301c8461ba86c7c70d6c8e894176f42c6 +size 14663 diff --git a/checkpoint-4500/scheduler.bin b/checkpoint-4500/scheduler.bin index 281aded4ddf3608ee6e6f8b785c5ef0ebe0f6cc2..f71a8c9ecce639117d1fb03ed5385fe38e4fa409 100644 --- a/checkpoint-4500/scheduler.bin +++ b/checkpoint-4500/scheduler.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:eaf3951b9f7c1a4c0276a3c1b89579e4c16228803496bc4def9b92e76fb07794 +oid sha256:16e2956f5e49f7e5453a5735778415fd8ebd10164f10c9120c90289df3dc9985 size 563 diff --git a/checkpoint-4500/unet/config.json b/checkpoint-4500/unet/config.json index c2d7b8c0d490531013583af57a806d21893e494c..06fb387aa3ed37ab3b2f455b5469059d55604f74 100644 --- a/checkpoint-4500/unet/config.json +++ b/checkpoint-4500/unet/config.json @@ -6,8 +6,8 @@ "addition_embed_type_num_heads": 64, "attention_head_dim": 8, "block_out_channels": [ - 64, - 64, + 128, + 128, 128, 128 ], @@ -19,9 +19,9 @@ "cross_attention_dim": 256, "cross_attention_norm": null, "down_block_types": [ - "DownBlock2D", "DownBlock2D", "CrossAttnDownBlock2D", + "CrossAttnDownBlock2D", "DownBlock2D" ], "downsample_padding": 1, @@ -52,7 +52,7 @@ "up_block_types": [ "UpBlock2D", "CrossAttnUpBlock2D", - "UpBlock2D", + "CrossAttnUpBlock2D", "UpBlock2D" ], "upcast_attention": false, diff --git a/checkpoint-4500/unet/diffusion_pytorch_model.bin b/checkpoint-4500/unet/diffusion_pytorch_model.bin index 8384102f3657d615d7cecfe29ffb675e9b2f3bbd..ff78ccceb210199a981b8e77153e4bc5eaed541e 100644 --- a/checkpoint-4500/unet/diffusion_pytorch_model.bin +++ b/checkpoint-4500/unet/diffusion_pytorch_model.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:4475a55a2d1d7b4fb3415387b6a35830f8d3a92a12d475fb28b821f2de14312a -size 36621969 +oid sha256:8799e34a025932bc7c99d6182231088939083e6e847db3d36ac503901eb4005d +size 63003985 diff --git a/checkpoint-4500/unet_ema/config.json b/checkpoint-4500/unet_ema/config.json index 44745db21d99a7a0a1baf1c2e5b238acaa3eee1a..c98afdf87faf50aa11f8195c019bf1a66092d2e2 100644 --- a/checkpoint-4500/unet_ema/config.json +++ b/checkpoint-4500/unet_ema/config.json @@ -6,8 +6,8 @@ "addition_embed_type_num_heads": 64, "attention_head_dim": 8, "block_out_channels": [ - 64, - 64, + 128, + 128, 128, 128 ], @@ -20,9 +20,9 @@ "cross_attention_norm": null, "decay": 0.9999, "down_block_types": [ - "DownBlock2D", "DownBlock2D", "CrossAttnDownBlock2D", + "CrossAttnDownBlock2D", "DownBlock2D" ], "downsample_padding": 1, @@ -57,7 +57,7 @@ "up_block_types": [ "UpBlock2D", "CrossAttnUpBlock2D", - "UpBlock2D", + "CrossAttnUpBlock2D", "UpBlock2D" ], "upcast_attention": false, diff --git a/checkpoint-4500/unet_ema/diffusion_pytorch_model.bin b/checkpoint-4500/unet_ema/diffusion_pytorch_model.bin index 83da272d85cdbc862394d6514c1ee093dc99ce59..c2f87049444318d4690ce9cfb18ff0a673fa5c53 100644 --- a/checkpoint-4500/unet_ema/diffusion_pytorch_model.bin +++ b/checkpoint-4500/unet_ema/diffusion_pytorch_model.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:213f80e7e7aaccf92da994ec8e62727f863513a72e81d859bf53f35b6785ac2e -size 36616081 +oid sha256:29dcabb25d6304db3165c47e34303c6180588bf5640fb4ce090d8a13d2a097fe +size 62996305 diff --git a/checkpoint-500/optimizer.bin b/checkpoint-500/optimizer.bin index 629f33f2b98d0f4c7c4f60e6c346a39d1eb7226b..01625894c9a6bd6b1fdc2397d24c112410507065 100644 --- a/checkpoint-500/optimizer.bin +++ b/checkpoint-500/optimizer.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:c5c3ccea6fdddff86832147a5f02fa9fb1550c79ca986e49bbb0cde3b5754932 -size 73251413 +oid sha256:5bc803ffd925c1152ea0c2231de660083a2f131cbac3487770fe5089a8aa841d +size 126008021 diff --git a/checkpoint-500/random_states_0.pkl b/checkpoint-500/random_states_0.pkl index 604aa8ac93800fbb2814d8d863935908a24a0d78..437a36688830c8521bffbcb6a04f98e937fd65df 100644 --- a/checkpoint-500/random_states_0.pkl +++ b/checkpoint-500/random_states_0.pkl @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:cf527a284a26be54a37a68abe6fb7944532cd66003eef5ba8e83e72861ed9cdc -size 14727 +oid sha256:279385b635675e78afeb749451b248ac32e31a3fb82d6a92511f75c4ad9cdc58 +size 14663 diff --git a/checkpoint-500/scheduler.bin b/checkpoint-500/scheduler.bin index 4b543b2f9480e9e7e05c8212347b4b5324d41dc2..ca69e08769743e3bee2e7876b2b72ded8d4c2acf 100644 --- a/checkpoint-500/scheduler.bin +++ b/checkpoint-500/scheduler.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:891ed7c0af227817d0ad1614c90e9d6c94613f9dd864e80c888f3d9320e13677 +oid sha256:cbdf252ee5dc55ecc31a0fb8797006b115989babd4fde1cc270bf46ec7a45786 size 563 diff --git a/checkpoint-500/unet/config.json b/checkpoint-500/unet/config.json index c2d7b8c0d490531013583af57a806d21893e494c..06fb387aa3ed37ab3b2f455b5469059d55604f74 100644 --- a/checkpoint-500/unet/config.json +++ b/checkpoint-500/unet/config.json @@ -6,8 +6,8 @@ "addition_embed_type_num_heads": 64, "attention_head_dim": 8, "block_out_channels": [ - 64, - 64, + 128, + 128, 128, 128 ], @@ -19,9 +19,9 @@ "cross_attention_dim": 256, "cross_attention_norm": null, "down_block_types": [ - "DownBlock2D", "DownBlock2D", "CrossAttnDownBlock2D", + "CrossAttnDownBlock2D", "DownBlock2D" ], "downsample_padding": 1, @@ -52,7 +52,7 @@ "up_block_types": [ "UpBlock2D", "CrossAttnUpBlock2D", - "UpBlock2D", + "CrossAttnUpBlock2D", "UpBlock2D" ], "upcast_attention": false, diff --git a/checkpoint-500/unet/diffusion_pytorch_model.bin b/checkpoint-500/unet/diffusion_pytorch_model.bin index ef3f486561428b7fe272c41425ed7227c5273a8c..6e4a0284009cabd49994380ebb1b3630a3f78241 100644 --- a/checkpoint-500/unet/diffusion_pytorch_model.bin +++ b/checkpoint-500/unet/diffusion_pytorch_model.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:dcd54693463747ac907e53f5755855e764517279b6683165ab3b5586040215f3 -size 36621969 +oid sha256:0b4f29dc106c2d6f7ca2c2fc5ae523c981f579ac09c325456d7cdd182dd8fbf0 +size 63003985 diff --git a/checkpoint-500/unet_ema/config.json b/checkpoint-500/unet_ema/config.json index 7e15f9b1a5938f7b18e3897842ae33373c99303d..90f3bb80c2d2b1f8b0da7ec9fdcf49f5301531db 100644 --- a/checkpoint-500/unet_ema/config.json +++ b/checkpoint-500/unet_ema/config.json @@ -6,8 +6,8 @@ "addition_embed_type_num_heads": 64, "attention_head_dim": 8, "block_out_channels": [ - 64, - 64, + 128, + 128, 128, 128 ], @@ -20,9 +20,9 @@ "cross_attention_norm": null, "decay": 0.9999, "down_block_types": [ - "DownBlock2D", "DownBlock2D", "CrossAttnDownBlock2D", + "CrossAttnDownBlock2D", "DownBlock2D" ], "downsample_padding": 1, @@ -57,7 +57,7 @@ "up_block_types": [ "UpBlock2D", "CrossAttnUpBlock2D", - "UpBlock2D", + "CrossAttnUpBlock2D", "UpBlock2D" ], "upcast_attention": false, diff --git a/checkpoint-500/unet_ema/diffusion_pytorch_model.bin b/checkpoint-500/unet_ema/diffusion_pytorch_model.bin index 09c9773c99de9e5c4442c99464d6e3f451926b47..e740c622658fd5071d99261de5c9027e764d7c90 100644 --- a/checkpoint-500/unet_ema/diffusion_pytorch_model.bin +++ b/checkpoint-500/unet_ema/diffusion_pytorch_model.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:9af8a4601a02cfae61b9d6e9dbd784619f26b70e8ea0e1986dc3e0fd4a8dfb20 -size 36616081 +oid sha256:1cb3e56d2137d1bba46ce7348554c1ceeeb1ee19e8af28aeeb89918cf7e10a29 +size 62996305 diff --git a/checkpoint-5000/optimizer.bin b/checkpoint-5000/optimizer.bin index 8411e02e19acff8542698cb4a79f5acaf0988bd5..2be572a2c09cbe1a80123de63dcb4c984425d9e7 100644 --- a/checkpoint-5000/optimizer.bin +++ b/checkpoint-5000/optimizer.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:ac03dbda1174f64faab0b3ebbdccc1a9a3a02b03fc702daa8a1e2d05fc5cd39d -size 73251413 +oid sha256:90ec862c5efa786e97471a114e76126f5788e5e5ca20f6796d6dfb21a1416332 +size 126008021 diff --git a/checkpoint-5000/random_states_0.pkl b/checkpoint-5000/random_states_0.pkl index 4ac00c00c815d84516fe6179e24165a95fb61b42..968d000aa3c4d370701a49bf4a44c2c5cf37ef79 100644 --- a/checkpoint-5000/random_states_0.pkl +++ b/checkpoint-5000/random_states_0.pkl @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:5cf8d7e653c8f12b99213c7b396789d96c964c40002adf5e5cbe65209730d638 -size 14727 +oid sha256:24f23d96dade3c5ac46ac46403269613aa111e16574a43844eabeab892c28059 +size 14663 diff --git a/checkpoint-5000/scheduler.bin b/checkpoint-5000/scheduler.bin index fd6d6162e8fbb8330a684812566ad67762f72d02..4dc5991bc2d1f6b627243e5c965707bdf6d2d32a 100644 --- a/checkpoint-5000/scheduler.bin +++ b/checkpoint-5000/scheduler.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:70c7d63024c89008d8db0b4f33560df320a58f68b308ada30300997408086de3 +oid sha256:a3f2957c8261280358b0e22d6c4a4b6ba0bea4da6e570d3482706e0abf4735c8 size 563 diff --git a/checkpoint-5000/unet/config.json b/checkpoint-5000/unet/config.json index c2d7b8c0d490531013583af57a806d21893e494c..06fb387aa3ed37ab3b2f455b5469059d55604f74 100644 --- a/checkpoint-5000/unet/config.json +++ b/checkpoint-5000/unet/config.json @@ -6,8 +6,8 @@ "addition_embed_type_num_heads": 64, "attention_head_dim": 8, "block_out_channels": [ - 64, - 64, + 128, + 128, 128, 128 ], @@ -19,9 +19,9 @@ "cross_attention_dim": 256, "cross_attention_norm": null, "down_block_types": [ - "DownBlock2D", "DownBlock2D", "CrossAttnDownBlock2D", + "CrossAttnDownBlock2D", "DownBlock2D" ], "downsample_padding": 1, @@ -52,7 +52,7 @@ "up_block_types": [ "UpBlock2D", "CrossAttnUpBlock2D", - "UpBlock2D", + "CrossAttnUpBlock2D", "UpBlock2D" ], "upcast_attention": false, diff --git a/checkpoint-5000/unet/diffusion_pytorch_model.bin b/checkpoint-5000/unet/diffusion_pytorch_model.bin index 7f2395328f094d13d1a5a8d166a6dc5cf268cae6..83c3acf7bb602ecd8296c061cd8864671d3ee4b0 100644 --- a/checkpoint-5000/unet/diffusion_pytorch_model.bin +++ b/checkpoint-5000/unet/diffusion_pytorch_model.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:1d9aa39d832fdefd6436bd0eaa340fb55f8544cb23fda373dcb0fbcc4b24543c -size 36621969 +oid sha256:8fc58596685621b281d1a582c7727230406bc153139aa5bba0b686dc5094acf3 +size 63003985 diff --git a/checkpoint-5000/unet_ema/config.json b/checkpoint-5000/unet_ema/config.json index d5ca0b8849f271b5c2948d484ae629d88ee85733..71b7b7c41a9d06d15c085a8191d46d84c76020e8 100644 --- a/checkpoint-5000/unet_ema/config.json +++ b/checkpoint-5000/unet_ema/config.json @@ -6,8 +6,8 @@ "addition_embed_type_num_heads": 64, "attention_head_dim": 8, "block_out_channels": [ - 64, - 64, + 128, + 128, 128, 128 ], @@ -20,9 +20,9 @@ "cross_attention_norm": null, "decay": 0.9999, "down_block_types": [ - "DownBlock2D", "DownBlock2D", "CrossAttnDownBlock2D", + "CrossAttnDownBlock2D", "DownBlock2D" ], "downsample_padding": 1, @@ -57,7 +57,7 @@ "up_block_types": [ "UpBlock2D", "CrossAttnUpBlock2D", - "UpBlock2D", + "CrossAttnUpBlock2D", "UpBlock2D" ], "upcast_attention": false, diff --git a/checkpoint-5000/unet_ema/diffusion_pytorch_model.bin b/checkpoint-5000/unet_ema/diffusion_pytorch_model.bin index e327d2f637330c3227046babc35481ff0407c9d4..0a0285e0256160c2d3d4ad341e9b630a50d43f11 100644 --- a/checkpoint-5000/unet_ema/diffusion_pytorch_model.bin +++ b/checkpoint-5000/unet_ema/diffusion_pytorch_model.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:bed912b78ad1d1ee111b2938d6643542d45cf37fc1f9b75029088f4b2a0c3dc0 -size 36616081 +oid sha256:72a1e958e36d9b34b9ccdfb35a404a7526e7c929900d30d4f1f2c4078ac59f26 +size 62996305 diff --git a/checkpoint-5500/optimizer.bin b/checkpoint-5500/optimizer.bin index bd163ada540905b1799e7a3243723369ce86990d..886c44a5c3a2b195fee12c5db6b2d663c0160dec 100644 --- a/checkpoint-5500/optimizer.bin +++ b/checkpoint-5500/optimizer.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:606b2035c4f804b6e776cc1fa8954856debd5edafe30caf50379a72e2cc54d56 -size 73251413 +oid sha256:83847745cac5c692dfc62cf2a7bffcf2830e4d34ce9d4433d21a4b086786a84d +size 126008021 diff --git a/checkpoint-5500/random_states_0.pkl b/checkpoint-5500/random_states_0.pkl index 90eda44523bf431161efe349d01dd96a7bb010d0..58e3f6f5393e416bd61edef1c471d153cdbe9809 100644 --- a/checkpoint-5500/random_states_0.pkl +++ b/checkpoint-5500/random_states_0.pkl @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:01c7c0dabb696c5409e280dbc2a45a61a2932ce68a7030c653e02d53068113d3 -size 14727 +oid sha256:3574d43e764e889c8624c8f660e48d11e746876c70d321a98fe4757f93b70d1a +size 14663 diff --git a/checkpoint-5500/scheduler.bin b/checkpoint-5500/scheduler.bin index 483897d76acd8465587599e1d6ef31766d9e7809..d7f3b1984cbcf15e8d1487becaea6b4fb4d85497 100644 --- a/checkpoint-5500/scheduler.bin +++ b/checkpoint-5500/scheduler.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:77d351906df33024a8aa8b83b7d307dbd8eee6f2f56012ddbe9867c86dc52a25 +oid sha256:9ae4b17de7907330da2042873133a8721cc2b185a350a597051948aee17f6b8a size 563 diff --git a/checkpoint-5500/unet/config.json b/checkpoint-5500/unet/config.json index c2d7b8c0d490531013583af57a806d21893e494c..06fb387aa3ed37ab3b2f455b5469059d55604f74 100644 --- a/checkpoint-5500/unet/config.json +++ b/checkpoint-5500/unet/config.json @@ -6,8 +6,8 @@ "addition_embed_type_num_heads": 64, "attention_head_dim": 8, "block_out_channels": [ - 64, - 64, + 128, + 128, 128, 128 ], @@ -19,9 +19,9 @@ "cross_attention_dim": 256, "cross_attention_norm": null, "down_block_types": [ - "DownBlock2D", "DownBlock2D", "CrossAttnDownBlock2D", + "CrossAttnDownBlock2D", "DownBlock2D" ], "downsample_padding": 1, @@ -52,7 +52,7 @@ "up_block_types": [ "UpBlock2D", "CrossAttnUpBlock2D", - "UpBlock2D", + "CrossAttnUpBlock2D", "UpBlock2D" ], "upcast_attention": false, diff --git a/checkpoint-5500/unet/diffusion_pytorch_model.bin b/checkpoint-5500/unet/diffusion_pytorch_model.bin index 980d9a4119a7a78563465b124ba9f614f1340a95..10d68d1279e0e0a9de810e978f6cafd03d6a8e93 100644 --- a/checkpoint-5500/unet/diffusion_pytorch_model.bin +++ b/checkpoint-5500/unet/diffusion_pytorch_model.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:b7f0f3cfe58c832c6ffb665b9af518db73e4a645b2d896afe02654a14de872dd -size 36621969 +oid sha256:59962caf0ee4086c30d51c2c5eb1a685354dc7620bfd0e8687ca27e8ecd673f0 +size 63003985 diff --git a/checkpoint-5500/unet_ema/config.json b/checkpoint-5500/unet_ema/config.json index c1653186fc0ea4c04f5ee62b449b85e2f72ab60a..15479739ecd7bd67e2dfb783b956d46047f2b111 100644 --- a/checkpoint-5500/unet_ema/config.json +++ b/checkpoint-5500/unet_ema/config.json @@ -6,8 +6,8 @@ "addition_embed_type_num_heads": 64, "attention_head_dim": 8, "block_out_channels": [ - 64, - 64, + 128, + 128, 128, 128 ], @@ -20,9 +20,9 @@ "cross_attention_norm": null, "decay": 0.9999, "down_block_types": [ - "DownBlock2D", "DownBlock2D", "CrossAttnDownBlock2D", + "CrossAttnDownBlock2D", "DownBlock2D" ], "downsample_padding": 1, @@ -57,7 +57,7 @@ "up_block_types": [ "UpBlock2D", "CrossAttnUpBlock2D", - "UpBlock2D", + "CrossAttnUpBlock2D", "UpBlock2D" ], "upcast_attention": false, diff --git a/checkpoint-5500/unet_ema/diffusion_pytorch_model.bin b/checkpoint-5500/unet_ema/diffusion_pytorch_model.bin index 78a1bec3ecdcc8bd1230e28102d44ffaf856f4ad..d6055a77d70074a34c1c4046046e811701849379 100644 --- a/checkpoint-5500/unet_ema/diffusion_pytorch_model.bin +++ b/checkpoint-5500/unet_ema/diffusion_pytorch_model.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:d555a93d289a68fb52366fdeb5766ab33daa9f00389233053cb3bf103fc9e7c5 -size 36616081 +oid sha256:28c09cc7618a907cd3ba86d8e03b99eaf4b281a4891bd43cdbae5357236155a6 +size 62996305 diff --git a/checkpoint-6000/optimizer.bin b/checkpoint-6000/optimizer.bin index 30c663a56879d8b5b261ba23c0a84805a57c48d8..0884108de507f596e8bbe3ed368efc634c6a6123 100644 --- a/checkpoint-6000/optimizer.bin +++ b/checkpoint-6000/optimizer.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:d7fbff33077bdb0780186542f4dbdb4882b069f86f183fed1be86e603061acc5 -size 73251413 +oid sha256:54b75d363fd564d2ac06c91be97046b3585c1c662051a51648be9a7a0789ec92 +size 126008021 diff --git a/checkpoint-6000/random_states_0.pkl b/checkpoint-6000/random_states_0.pkl index 300241b274cb6d2d655decccb5f8c8c90176655f..547530763568939866ca48eba563f01c05f833ab 100644 --- a/checkpoint-6000/random_states_0.pkl +++ b/checkpoint-6000/random_states_0.pkl @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:143b7e5020ff18898caa1f225d178cb96da73c5ae5d3b892fdf49d58d3fe73af -size 14727 +oid sha256:e8e370556a348877c83bff42a03de0f1504cef24fe874ce051200aab0ffc4737 +size 14663 diff --git a/checkpoint-6000/scheduler.bin b/checkpoint-6000/scheduler.bin index d93e026870bf5e1c223e32c927494446c63b4bf4..144b3ee01c00ba8bc7facd796459a11293a71d8a 100644 --- a/checkpoint-6000/scheduler.bin +++ b/checkpoint-6000/scheduler.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:389678ea47b9e4f4251d30b2f9a3351a1e0659b1066842fa6f47d6142a10a571 +oid sha256:f112a163f59a4821b3326dcdcff0043a3808c0a39fea1059ad6df6559a4e6b68 size 563 diff --git a/checkpoint-6000/unet/config.json b/checkpoint-6000/unet/config.json index c2d7b8c0d490531013583af57a806d21893e494c..06fb387aa3ed37ab3b2f455b5469059d55604f74 100644 --- a/checkpoint-6000/unet/config.json +++ b/checkpoint-6000/unet/config.json @@ -6,8 +6,8 @@ "addition_embed_type_num_heads": 64, "attention_head_dim": 8, "block_out_channels": [ - 64, - 64, + 128, + 128, 128, 128 ], @@ -19,9 +19,9 @@ "cross_attention_dim": 256, "cross_attention_norm": null, "down_block_types": [ - "DownBlock2D", "DownBlock2D", "CrossAttnDownBlock2D", + "CrossAttnDownBlock2D", "DownBlock2D" ], "downsample_padding": 1, @@ -52,7 +52,7 @@ "up_block_types": [ "UpBlock2D", "CrossAttnUpBlock2D", - "UpBlock2D", + "CrossAttnUpBlock2D", "UpBlock2D" ], "upcast_attention": false, diff --git a/checkpoint-6000/unet/diffusion_pytorch_model.bin b/checkpoint-6000/unet/diffusion_pytorch_model.bin index 9865abf89eab0a601f106dd11568d5c0a629049f..72fad32cfeba6ede7ec025cdb836253b86e3dba6 100644 --- a/checkpoint-6000/unet/diffusion_pytorch_model.bin +++ b/checkpoint-6000/unet/diffusion_pytorch_model.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:bd886d8d36a4ebe61a183764f77c36e83e674e8018a483604249163c8bc67c4b -size 36621969 +oid sha256:043717197d4865825227da725a2cfab8787f612e078aa6384c705052fec05f79 +size 63003985 diff --git a/checkpoint-6000/unet_ema/config.json b/checkpoint-6000/unet_ema/config.json index 1e93e24e6f3823631529ea7736146bf5276ed102..f6c8f55de0438d115e9a979cf0179352ee9b5c66 100644 --- a/checkpoint-6000/unet_ema/config.json +++ b/checkpoint-6000/unet_ema/config.json @@ -6,8 +6,8 @@ "addition_embed_type_num_heads": 64, "attention_head_dim": 8, "block_out_channels": [ - 64, - 64, + 128, + 128, 128, 128 ], @@ -20,9 +20,9 @@ "cross_attention_norm": null, "decay": 0.9999, "down_block_types": [ - "DownBlock2D", "DownBlock2D", "CrossAttnDownBlock2D", + "CrossAttnDownBlock2D", "DownBlock2D" ], "downsample_padding": 1, @@ -57,7 +57,7 @@ "up_block_types": [ "UpBlock2D", "CrossAttnUpBlock2D", - "UpBlock2D", + "CrossAttnUpBlock2D", "UpBlock2D" ], "upcast_attention": false, diff --git a/checkpoint-6000/unet_ema/diffusion_pytorch_model.bin b/checkpoint-6000/unet_ema/diffusion_pytorch_model.bin index c8a944dcb2397bdc4b6f623f5d5991739b36a6de..62ec6bc719a8c95ab73f28b7d550039c28d15d96 100644 --- a/checkpoint-6000/unet_ema/diffusion_pytorch_model.bin +++ b/checkpoint-6000/unet_ema/diffusion_pytorch_model.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:0843dace922a31ca45a6db98f1ba19fb2a1bf5e811a17e6f9bc3d039bc2653f6 -size 36616081 +oid sha256:1f2cfe5ef585a58abc6af52d6fa0bc5fb1eee678aeccb94a1a871b72839c6ced +size 62996305 diff --git a/checkpoint-6500/optimizer.bin b/checkpoint-6500/optimizer.bin index 416dc68da23f9c99265a6a4593a45ce2ff58cabc..16c25832952905357a07056c5e39f0980b5aa466 100644 --- a/checkpoint-6500/optimizer.bin +++ b/checkpoint-6500/optimizer.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:b48790f8fc8d1a944da92514b209ea7e8dd3691501b3eb68abb47d1f934f6c99 -size 73251413 +oid sha256:746f84d55d43165f57cf8eeb5d571145918d03a7426826dac61c3c4a6bdaa695 +size 126008021 diff --git a/checkpoint-6500/random_states_0.pkl b/checkpoint-6500/random_states_0.pkl index 8cb53520abbb1988a6469c468f8d77d691931d95..bec3b1252703f4b25f07ca9a2140417f9c856cc6 100644 --- a/checkpoint-6500/random_states_0.pkl +++ b/checkpoint-6500/random_states_0.pkl @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:08385557fd8e613173e419fefddedc3232e00e22680de844d2b3c1a63dd9b674 -size 14727 +oid sha256:28e3ac293dd82af107a54d2b044878cf51e32e91b1139eda5b4515a30c7c57d5 +size 14663 diff --git a/checkpoint-6500/scheduler.bin b/checkpoint-6500/scheduler.bin index 44481f649e22a1a760455e2ccf689073b471fba8..e2309690f232cea9f5e9e60b3e6766500472994c 100644 --- a/checkpoint-6500/scheduler.bin +++ b/checkpoint-6500/scheduler.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:b7bfdf8c3e71b3d963ff6dd33c57ebc98359d0a2b33cf41679efa34cd0d72e33 +oid sha256:b6daf4f02e6dc9625ff0a0311b1715f29bcc5749642a7ddd95df6da07e867ae3 size 563 diff --git a/checkpoint-6500/unet/config.json b/checkpoint-6500/unet/config.json index c2d7b8c0d490531013583af57a806d21893e494c..06fb387aa3ed37ab3b2f455b5469059d55604f74 100644 --- a/checkpoint-6500/unet/config.json +++ b/checkpoint-6500/unet/config.json @@ -6,8 +6,8 @@ "addition_embed_type_num_heads": 64, "attention_head_dim": 8, "block_out_channels": [ - 64, - 64, + 128, + 128, 128, 128 ], @@ -19,9 +19,9 @@ "cross_attention_dim": 256, "cross_attention_norm": null, "down_block_types": [ - "DownBlock2D", "DownBlock2D", "CrossAttnDownBlock2D", + "CrossAttnDownBlock2D", "DownBlock2D" ], "downsample_padding": 1, @@ -52,7 +52,7 @@ "up_block_types": [ "UpBlock2D", "CrossAttnUpBlock2D", - "UpBlock2D", + "CrossAttnUpBlock2D", "UpBlock2D" ], "upcast_attention": false, diff --git a/checkpoint-6500/unet/diffusion_pytorch_model.bin b/checkpoint-6500/unet/diffusion_pytorch_model.bin index f1a4a72da7e0948193112b0f52720f521346a270..c1277be9c179174d83bceb6ef325adca400d544d 100644 --- a/checkpoint-6500/unet/diffusion_pytorch_model.bin +++ b/checkpoint-6500/unet/diffusion_pytorch_model.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:c0ebd3eb963b0e0f1863568f70160fafc19b0d3a2083f3458164549749679086 -size 36621969 +oid sha256:f5fbf8214fb573ef9f5fc0ee4708257d097512c0dcd4e71a19ce5ca408bbae6d +size 63003985 diff --git a/checkpoint-6500/unet_ema/config.json b/checkpoint-6500/unet_ema/config.json index 6106cb1ea2ff1db78b885857904d2de0be7f5362..32048701484ee06cffc75141224271367a25de28 100644 --- a/checkpoint-6500/unet_ema/config.json +++ b/checkpoint-6500/unet_ema/config.json @@ -6,8 +6,8 @@ "addition_embed_type_num_heads": 64, "attention_head_dim": 8, "block_out_channels": [ - 64, - 64, + 128, + 128, 128, 128 ], @@ -20,9 +20,9 @@ "cross_attention_norm": null, "decay": 0.9999, "down_block_types": [ - "DownBlock2D", "DownBlock2D", "CrossAttnDownBlock2D", + "CrossAttnDownBlock2D", "DownBlock2D" ], "downsample_padding": 1, @@ -57,7 +57,7 @@ "up_block_types": [ "UpBlock2D", "CrossAttnUpBlock2D", - "UpBlock2D", + "CrossAttnUpBlock2D", "UpBlock2D" ], "upcast_attention": false, diff --git a/checkpoint-6500/unet_ema/diffusion_pytorch_model.bin b/checkpoint-6500/unet_ema/diffusion_pytorch_model.bin index 65e80f95684455647e2328ba54d32c7e903b344d..2fbd5d1cc82d27a0c25dd92003813d0a153e5581 100644 --- a/checkpoint-6500/unet_ema/diffusion_pytorch_model.bin +++ b/checkpoint-6500/unet_ema/diffusion_pytorch_model.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:f151e9050699ab07a129507c9b78e55bbb35b12684a65d8eb8abfeeba38610e3 -size 36616081 +oid sha256:904c00206b98b97004aa0210f7869c55fd43b7cb2721712da3e1ca074d34cddb +size 62996305 diff --git a/checkpoint-7000/optimizer.bin b/checkpoint-7000/optimizer.bin index 739a6fbc5fc9bc8b02af86359655d1d15ff73715..7f6f79426b3ae2d8d09deda8662b393188dd290b 100644 --- a/checkpoint-7000/optimizer.bin +++ b/checkpoint-7000/optimizer.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:2a0c128ca57492bd9f07200914f768f31e953ae44bb56406b88d076100fe8f0a -size 73251413 +oid sha256:5ea6e72c44d4fd4e5292685a508dfe15c0c91db93d2a5dd8084822507a9abe28 +size 126008021 diff --git a/checkpoint-7000/random_states_0.pkl b/checkpoint-7000/random_states_0.pkl index 63e604cbcb3db28676042e0c7eb54f49dd5515d4..2e3a6bd020c3aa2eacafae78b5f00abe1c6cbf04 100644 --- a/checkpoint-7000/random_states_0.pkl +++ b/checkpoint-7000/random_states_0.pkl @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:b85cb0fbafebbbc87c5e0c12d3d589043a6046b747e7f84d573ce623f27ac93c -size 14727 +oid sha256:a9b9a1705a7956e3cf4ed4c00e9f493bf0ec534d8312d9f3bda92bfa15e39f50 +size 14663 diff --git a/checkpoint-7000/scheduler.bin b/checkpoint-7000/scheduler.bin index 352fe7f6722d899a6f8e8134b9cea06c893c8f61..5095101fd6bfd6aba5d1112750f5b9874a7e4f0b 100644 --- a/checkpoint-7000/scheduler.bin +++ b/checkpoint-7000/scheduler.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:ed298a0975060242c9e54957567d3c304e0b3a9df7fd219bc0365f87f985d82b +oid sha256:9de799bcfe66cc6b0b98bf6cf9337eea60932c337c2fcc8b629e5dacc3da22a3 size 563 diff --git a/checkpoint-7000/unet/config.json b/checkpoint-7000/unet/config.json index c2d7b8c0d490531013583af57a806d21893e494c..06fb387aa3ed37ab3b2f455b5469059d55604f74 100644 --- a/checkpoint-7000/unet/config.json +++ b/checkpoint-7000/unet/config.json @@ -6,8 +6,8 @@ "addition_embed_type_num_heads": 64, "attention_head_dim": 8, "block_out_channels": [ - 64, - 64, + 128, + 128, 128, 128 ], @@ -19,9 +19,9 @@ "cross_attention_dim": 256, "cross_attention_norm": null, "down_block_types": [ - "DownBlock2D", "DownBlock2D", "CrossAttnDownBlock2D", + "CrossAttnDownBlock2D", "DownBlock2D" ], "downsample_padding": 1, @@ -52,7 +52,7 @@ "up_block_types": [ "UpBlock2D", "CrossAttnUpBlock2D", - "UpBlock2D", + "CrossAttnUpBlock2D", "UpBlock2D" ], "upcast_attention": false, diff --git a/checkpoint-7000/unet/diffusion_pytorch_model.bin b/checkpoint-7000/unet/diffusion_pytorch_model.bin index 9f5e5a1ffb7a16dbd837ba1b3de103db15bf4590..ca8108e3b22ab416cf144d8a18e2ccfa8129ed47 100644 --- a/checkpoint-7000/unet/diffusion_pytorch_model.bin +++ b/checkpoint-7000/unet/diffusion_pytorch_model.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:86c642028ccaa094ed55874fb63782cbec2f0d78bc91ab848f3296bcae23b4c7 -size 36621969 +oid sha256:b9515190724c3b75dcb70b2d04ca224bd03d3525b87acdd2680d669c5969f95c +size 63003985 diff --git a/checkpoint-7000/unet_ema/config.json b/checkpoint-7000/unet_ema/config.json index 2dcffae6b16ae612a59763efa0c01bfda7ee2b60..8cb662a3341fd364e383d10640a94721e78ca009 100644 --- a/checkpoint-7000/unet_ema/config.json +++ b/checkpoint-7000/unet_ema/config.json @@ -6,8 +6,8 @@ "addition_embed_type_num_heads": 64, "attention_head_dim": 8, "block_out_channels": [ - 64, - 64, + 128, + 128, 128, 128 ], @@ -20,9 +20,9 @@ "cross_attention_norm": null, "decay": 0.9999, "down_block_types": [ - "DownBlock2D", "DownBlock2D", "CrossAttnDownBlock2D", + "CrossAttnDownBlock2D", "DownBlock2D" ], "downsample_padding": 1, @@ -57,7 +57,7 @@ "up_block_types": [ "UpBlock2D", "CrossAttnUpBlock2D", - "UpBlock2D", + "CrossAttnUpBlock2D", "UpBlock2D" ], "upcast_attention": false, diff --git a/checkpoint-7000/unet_ema/diffusion_pytorch_model.bin b/checkpoint-7000/unet_ema/diffusion_pytorch_model.bin index d22f760f5afcf3a5fea384d0ce730bd008f63654..e36e0f0b16d4db7e631137e4ea631dc188688a62 100644 --- a/checkpoint-7000/unet_ema/diffusion_pytorch_model.bin +++ b/checkpoint-7000/unet_ema/diffusion_pytorch_model.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:0f8f0f00cf2c9686294481a97478f1bbc3f236e8ac06f3d5d72eff34401913f9 -size 36616081 +oid sha256:8084d3d8ab9d7af8f691e1991a8648989007baf03ff2d68529f8b80c6ef8eab1 +size 62996305 diff --git a/checkpoint-7500/optimizer.bin b/checkpoint-7500/optimizer.bin index 22954879623973456197f1a19c437d5780562a1e..ea03150e5a4a674d00308bf2dd544a5417895d12 100644 --- a/checkpoint-7500/optimizer.bin +++ b/checkpoint-7500/optimizer.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:41cac709dc9c3923e46c40eaf2dea3a0076e4a32a973ffb7573c232368de6cbf -size 73251413 +oid sha256:8c9f4930241e768952aacb743460d8824db23bd814f063b7f5c16f12340ae6e1 +size 126008021 diff --git a/checkpoint-7500/random_states_0.pkl b/checkpoint-7500/random_states_0.pkl index a23ca6ef67d155255d53ec8c807167be29a1cec8..739a5df7b6e316d153f692f42450ac7939ed08b1 100644 --- a/checkpoint-7500/random_states_0.pkl +++ b/checkpoint-7500/random_states_0.pkl @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:b6eaa806c39c3662c07ad15d9b26e6dcb2a212a765b8b653f0e35320732df322 -size 14727 +oid sha256:7c3258f3b67164497ce0e39dc13e36bb71ae797403a517d1946ea45cb9eb09e8 +size 14663 diff --git a/checkpoint-7500/scheduler.bin b/checkpoint-7500/scheduler.bin index 5951bc2d038c2df3910547dc0f3bf8799bf7cc52..d81f706b12e5443a026400bcb235d249ca47dd49 100644 --- a/checkpoint-7500/scheduler.bin +++ b/checkpoint-7500/scheduler.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:1903e404dbf47fb7c51279a3af6dc1bbb892594d8abbc425be6b33fef456b57b +oid sha256:7bcf62a8b393bfb5fb23005eda15b1e56f6e7b1acd51fb1408ee4899f1a280c2 size 563 diff --git a/checkpoint-7500/unet/config.json b/checkpoint-7500/unet/config.json index c2d7b8c0d490531013583af57a806d21893e494c..06fb387aa3ed37ab3b2f455b5469059d55604f74 100644 --- a/checkpoint-7500/unet/config.json +++ b/checkpoint-7500/unet/config.json @@ -6,8 +6,8 @@ "addition_embed_type_num_heads": 64, "attention_head_dim": 8, "block_out_channels": [ - 64, - 64, + 128, + 128, 128, 128 ], @@ -19,9 +19,9 @@ "cross_attention_dim": 256, "cross_attention_norm": null, "down_block_types": [ - "DownBlock2D", "DownBlock2D", "CrossAttnDownBlock2D", + "CrossAttnDownBlock2D", "DownBlock2D" ], "downsample_padding": 1, @@ -52,7 +52,7 @@ "up_block_types": [ "UpBlock2D", "CrossAttnUpBlock2D", - "UpBlock2D", + "CrossAttnUpBlock2D", "UpBlock2D" ], "upcast_attention": false, diff --git a/checkpoint-7500/unet/diffusion_pytorch_model.bin b/checkpoint-7500/unet/diffusion_pytorch_model.bin index 88d297fe2af3648ad4e13cab1f5cccaec87268c5..22172f157840bd745a9f4ebe24cf5c09673a15b0 100644 --- a/checkpoint-7500/unet/diffusion_pytorch_model.bin +++ b/checkpoint-7500/unet/diffusion_pytorch_model.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:50df84cffc5720edd5cc094c345ce5cf1b3f69403d76bfb92001e33d7e4a09d5 -size 36621969 +oid sha256:95d8624dc3350ca2afbd3df928fe69b5ea13270156dc55958b134c62829ef6c5 +size 63003985 diff --git a/checkpoint-7500/unet_ema/config.json b/checkpoint-7500/unet_ema/config.json index 3e78303b1abfcacbdc1c9abf77538c86365f2542..3c787c949b53f2359f95a2cc8fbba000c473c4b3 100644 --- a/checkpoint-7500/unet_ema/config.json +++ b/checkpoint-7500/unet_ema/config.json @@ -6,8 +6,8 @@ "addition_embed_type_num_heads": 64, "attention_head_dim": 8, "block_out_channels": [ - 64, - 64, + 128, + 128, 128, 128 ], @@ -20,9 +20,9 @@ "cross_attention_norm": null, "decay": 0.9999, "down_block_types": [ - "DownBlock2D", "DownBlock2D", "CrossAttnDownBlock2D", + "CrossAttnDownBlock2D", "DownBlock2D" ], "downsample_padding": 1, @@ -57,7 +57,7 @@ "up_block_types": [ "UpBlock2D", "CrossAttnUpBlock2D", - "UpBlock2D", + "CrossAttnUpBlock2D", "UpBlock2D" ], "upcast_attention": false, diff --git a/checkpoint-7500/unet_ema/diffusion_pytorch_model.bin b/checkpoint-7500/unet_ema/diffusion_pytorch_model.bin index 74a85c73fa6cabd3765c433ccd4df9c2d05ea031..67915ce9148d7b95408b3ccee3288b0a97695a54 100644 --- a/checkpoint-7500/unet_ema/diffusion_pytorch_model.bin +++ b/checkpoint-7500/unet_ema/diffusion_pytorch_model.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:351274fb917bccda78b0edfa47e566ce01d373eb0b10744bda33e9ec9b6af38b -size 36616081 +oid sha256:e19452d559447e70947e6e2180bc07855d6e4b5c8c3f50aaa53b1899eb564808 +size 62996305 diff --git a/checkpoint-8000/optimizer.bin b/checkpoint-8000/optimizer.bin index dbe81bd66b76a732ea41cd0ed925da462b53be77..f4d59afb5b5d736bff7273d2cb6edf8fedb7bd6c 100644 --- a/checkpoint-8000/optimizer.bin +++ b/checkpoint-8000/optimizer.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:37aaefefdb6f70582ba4e0fe792f1815bf645c5274160d6be0122c97f33e6772 -size 73251413 +oid sha256:c3c7c8c2540bc30e8887baf4dbc7a59a0ba1d9eb84d5e3d91e089aee4e02b26d +size 126008021 diff --git a/checkpoint-8000/random_states_0.pkl b/checkpoint-8000/random_states_0.pkl index dad2b104071d7a7925e24545748cdc260a0e5938..35c242ceaf76dfd66c9e45c659670706f9f3de84 100644 --- a/checkpoint-8000/random_states_0.pkl +++ b/checkpoint-8000/random_states_0.pkl @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:33bbd98ce1aec386264aaa6446863e17dbd2c555965f9908e18cc125f7eaebcc -size 14727 +oid sha256:f4a0b8687a418440e9ae6e0335f66ce0f248cc3e43884dc41cf415017a8cd0be +size 14663 diff --git a/checkpoint-8000/scheduler.bin b/checkpoint-8000/scheduler.bin index 103cd7d39761333ae45241ced58d4939b8bbab1b..e9eb428f6f4977066d8f88cc1d5bdb8c4b2ff4e5 100644 --- a/checkpoint-8000/scheduler.bin +++ b/checkpoint-8000/scheduler.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:a5fce898e072e105d4a0326632e232d94fb92053d23ba0d2277130a1e6e426dd +oid sha256:432364952f2cf1a8b7fcc370bab1c6ec192d3cd9d5275dacdf4a8f6ec0daab56 size 563 diff --git a/checkpoint-8000/unet/config.json b/checkpoint-8000/unet/config.json index c2d7b8c0d490531013583af57a806d21893e494c..06fb387aa3ed37ab3b2f455b5469059d55604f74 100644 --- a/checkpoint-8000/unet/config.json +++ b/checkpoint-8000/unet/config.json @@ -6,8 +6,8 @@ "addition_embed_type_num_heads": 64, "attention_head_dim": 8, "block_out_channels": [ - 64, - 64, + 128, + 128, 128, 128 ], @@ -19,9 +19,9 @@ "cross_attention_dim": 256, "cross_attention_norm": null, "down_block_types": [ - "DownBlock2D", "DownBlock2D", "CrossAttnDownBlock2D", + "CrossAttnDownBlock2D", "DownBlock2D" ], "downsample_padding": 1, @@ -52,7 +52,7 @@ "up_block_types": [ "UpBlock2D", "CrossAttnUpBlock2D", - "UpBlock2D", + "CrossAttnUpBlock2D", "UpBlock2D" ], "upcast_attention": false, diff --git a/checkpoint-8000/unet/diffusion_pytorch_model.bin b/checkpoint-8000/unet/diffusion_pytorch_model.bin index 51c8b63fff747ff79f0bc1daea87f652a8e99e1a..f36158b2988ed0509d4971825eee3e1ce298334e 100644 --- a/checkpoint-8000/unet/diffusion_pytorch_model.bin +++ b/checkpoint-8000/unet/diffusion_pytorch_model.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:76e3eaf11f21b65844da6738ec4a96573c9fe7df121e89e0444f18f2ba57017b -size 36621969 +oid sha256:57e17c38d5103bc22dce6e906470b8dc137dd3880077c008821aefbe2cba09db +size 63003985 diff --git a/checkpoint-8000/unet_ema/config.json b/checkpoint-8000/unet_ema/config.json index f7be3d86d384d5425fcf72e9875c7d67720cc882..8a32b5ffb6296a0e068ee38acaf6121fda0c8a94 100644 --- a/checkpoint-8000/unet_ema/config.json +++ b/checkpoint-8000/unet_ema/config.json @@ -6,8 +6,8 @@ "addition_embed_type_num_heads": 64, "attention_head_dim": 8, "block_out_channels": [ - 64, - 64, + 128, + 128, 128, 128 ], @@ -20,9 +20,9 @@ "cross_attention_norm": null, "decay": 0.9999, "down_block_types": [ - "DownBlock2D", "DownBlock2D", "CrossAttnDownBlock2D", + "CrossAttnDownBlock2D", "DownBlock2D" ], "downsample_padding": 1, @@ -57,7 +57,7 @@ "up_block_types": [ "UpBlock2D", "CrossAttnUpBlock2D", - "UpBlock2D", + "CrossAttnUpBlock2D", "UpBlock2D" ], "upcast_attention": false, diff --git a/checkpoint-8000/unet_ema/diffusion_pytorch_model.bin b/checkpoint-8000/unet_ema/diffusion_pytorch_model.bin index fd1f346baa0fbbad3251e125e67fb5b932eb5451..628596af6720a0b6dac856b307b98b4da8dfbc67 100644 --- a/checkpoint-8000/unet_ema/diffusion_pytorch_model.bin +++ b/checkpoint-8000/unet_ema/diffusion_pytorch_model.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:87e293f8704763d09dcb6642636b05ba2a5bee750cb75c7c805100d1c9efe85f -size 36616081 +oid sha256:d8c607ae5a46adb211f469b1f193adf919e5e27137265772a50890e3952adbca +size 62996305 diff --git a/checkpoint-8500/optimizer.bin b/checkpoint-8500/optimizer.bin index f5a374e6b38a90f926376845d40a66f7448f6a3d..835b4d511ac1bac21fb3e873e657d4c574b88640 100644 --- a/checkpoint-8500/optimizer.bin +++ b/checkpoint-8500/optimizer.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:4e4a6d9ef511c9f1fa300fd2c9ec335b57809d9e0b505bddbcd1fd6376a0e241 -size 73251413 +oid sha256:c4801e7b196390298d36c695f47929b471c02438796987357157d34faa38ffbd +size 126008021 diff --git a/checkpoint-8500/random_states_0.pkl b/checkpoint-8500/random_states_0.pkl index 69e431344c127c29b78f6901bbd3fbcf47e60481..37dbbf0796b7cd9c91b0d19a8860fa7d2c25a25d 100644 --- a/checkpoint-8500/random_states_0.pkl +++ b/checkpoint-8500/random_states_0.pkl @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:661940e7ceee476e4baa48b5c506e1233a6791acf822701a3cd1ccd3894cbc40 -size 14727 +oid sha256:46b751ec571365e7eb57ec0553e8c1c4dfa5955fec012a04beecfafa259235e3 +size 14663 diff --git a/checkpoint-8500/scheduler.bin b/checkpoint-8500/scheduler.bin index 95a37b7829b192ae98e4de63e2b90689143e5cce..0709b5b50b0747cd6f14e5629d08c26a525cefac 100644 --- a/checkpoint-8500/scheduler.bin +++ b/checkpoint-8500/scheduler.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:7deb6dde713df800b42b2601c85ef11c1995f21b48bb09f9f50ea8955527b441 +oid sha256:50131b4a4fd3ce64e6cd851891b8e8e1b1948539a33084a234fe7699e196eb7a size 563 diff --git a/checkpoint-8500/unet/config.json b/checkpoint-8500/unet/config.json index c2d7b8c0d490531013583af57a806d21893e494c..06fb387aa3ed37ab3b2f455b5469059d55604f74 100644 --- a/checkpoint-8500/unet/config.json +++ b/checkpoint-8500/unet/config.json @@ -6,8 +6,8 @@ "addition_embed_type_num_heads": 64, "attention_head_dim": 8, "block_out_channels": [ - 64, - 64, + 128, + 128, 128, 128 ], @@ -19,9 +19,9 @@ "cross_attention_dim": 256, "cross_attention_norm": null, "down_block_types": [ - "DownBlock2D", "DownBlock2D", "CrossAttnDownBlock2D", + "CrossAttnDownBlock2D", "DownBlock2D" ], "downsample_padding": 1, @@ -52,7 +52,7 @@ "up_block_types": [ "UpBlock2D", "CrossAttnUpBlock2D", - "UpBlock2D", + "CrossAttnUpBlock2D", "UpBlock2D" ], "upcast_attention": false, diff --git a/checkpoint-8500/unet/diffusion_pytorch_model.bin b/checkpoint-8500/unet/diffusion_pytorch_model.bin index 19ea646dbf29165379b43d5dbfcd1440cd2903b7..dffa668502fac42f5a512392dc7175f7a73135bf 100644 --- a/checkpoint-8500/unet/diffusion_pytorch_model.bin +++ b/checkpoint-8500/unet/diffusion_pytorch_model.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:4db46db5a375d8f9c41e751fe456f37aae331de5b434aa7292833e82b773feca -size 36621969 +oid sha256:0de7fcb5ddf097d196d0907b957674b5bae099d69be98ecb330c6bcddc8e8dc8 +size 63003985 diff --git a/checkpoint-8500/unet_ema/config.json b/checkpoint-8500/unet_ema/config.json index 5c169e81ce8a9866f6f686733c3ce23f6f982fec..90e7dd84a2a93bf5ff964351761a3895ee3aa6ad 100644 --- a/checkpoint-8500/unet_ema/config.json +++ b/checkpoint-8500/unet_ema/config.json @@ -6,8 +6,8 @@ "addition_embed_type_num_heads": 64, "attention_head_dim": 8, "block_out_channels": [ - 64, - 64, + 128, + 128, 128, 128 ], @@ -20,9 +20,9 @@ "cross_attention_norm": null, "decay": 0.9999, "down_block_types": [ - "DownBlock2D", "DownBlock2D", "CrossAttnDownBlock2D", + "CrossAttnDownBlock2D", "DownBlock2D" ], "downsample_padding": 1, @@ -57,7 +57,7 @@ "up_block_types": [ "UpBlock2D", "CrossAttnUpBlock2D", - "UpBlock2D", + "CrossAttnUpBlock2D", "UpBlock2D" ], "upcast_attention": false, diff --git a/checkpoint-8500/unet_ema/diffusion_pytorch_model.bin b/checkpoint-8500/unet_ema/diffusion_pytorch_model.bin index f0f61f82c3ffa2f582f76c5c6a480f39f247082b..012248ff8a7cb255a8e7b1878c34aa9759fb3193 100644 --- a/checkpoint-8500/unet_ema/diffusion_pytorch_model.bin +++ b/checkpoint-8500/unet_ema/diffusion_pytorch_model.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:67ff45d8aa1829fb0bed3fd1c8f52edc3dfb4bb186d4fbb1cf892955361a4ae5 -size 36616081 +oid sha256:776906b13e4b30d951d8c4f23d57f6827ca98b72a80d72d3ef94e96ccd6c7c72 +size 62996305 diff --git a/unet/config.json b/unet/config.json index c2d7b8c0d490531013583af57a806d21893e494c..06fb387aa3ed37ab3b2f455b5469059d55604f74 100644 --- a/unet/config.json +++ b/unet/config.json @@ -6,8 +6,8 @@ "addition_embed_type_num_heads": 64, "attention_head_dim": 8, "block_out_channels": [ - 64, - 64, + 128, + 128, 128, 128 ], @@ -19,9 +19,9 @@ "cross_attention_dim": 256, "cross_attention_norm": null, "down_block_types": [ - "DownBlock2D", "DownBlock2D", "CrossAttnDownBlock2D", + "CrossAttnDownBlock2D", "DownBlock2D" ], "downsample_padding": 1, @@ -52,7 +52,7 @@ "up_block_types": [ "UpBlock2D", "CrossAttnUpBlock2D", - "UpBlock2D", + "CrossAttnUpBlock2D", "UpBlock2D" ], "upcast_attention": false, diff --git a/unet/diffusion_pytorch_model.bin b/unet/diffusion_pytorch_model.bin index e307f9e9dd49931b96cd1ec697be8482eddd8e86..770bffe917873f279a6058bb362e69916fe5f1c8 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:6344b8906475a63e0459683e6459f9fec1633c1b67c4a3d9bc180afea384c0d1 -size 36621969 +oid sha256:1443a9426caf351c4f893937e939a3d5433261debd53b385faaa0aa4a036b25e +size 63003985