diff --git a/checkpoint-10000/optimizer.bin b/checkpoint-10000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..7eb013b2b3e3d1587dc58fa150b2311b0da27d09 --- /dev/null +++ b/checkpoint-10000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:dc213ea7911887962275ee036cdaca175e56d3fe1ad54427f4ee2f25c1a86dbb +size 6876851924 diff --git a/checkpoint-10000/random_states_0.pkl b/checkpoint-10000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..df08d009f5eed5558fbc1a30099831c85c15f02d --- /dev/null +++ b/checkpoint-10000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:1559c72680a9165d815e4ac590abadc4f1e98c68808f34a20efbb3dfe262692b +size 16036 diff --git a/checkpoint-10000/scheduler.bin b/checkpoint-10000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..eddbbf8cb6c157e1aae29c85d87cc021a5a2ab0d --- /dev/null +++ b/checkpoint-10000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:0f807739dfe462022eca2fd8aeecb73007e24eb43ddd85aa50b49a8b692e0fb5 +size 1000 diff --git a/checkpoint-10000/unet/config.json b/checkpoint-10000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..ccb8474073252815aa720c280c23b28669d7a727 --- /dev/null +++ b/checkpoint-10000/unet/config.json @@ -0,0 +1,68 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.33.1", + "_name_or_path": "output_lr1e4_accstep1/checkpoint-2500", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "addition_time_embed_dim": null, + "attention_head_dim": 8, + "attention_type": "default", + "block_out_channels": [ + 320, + 640, + 1280, + 1280 + ], + "center_input_sample": false, + "class_embed_type": null, + "class_embeddings_concat": false, + "conv_in_kernel": 3, + "conv_out_kernel": 3, + "cross_attention_dim": 768, + "cross_attention_norm": null, + "down_block_types": [ + "CrossAttnDownBlock2D", + "CrossAttnDownBlock2D", + "CrossAttnDownBlock2D", + "DownBlock2D" + ], + "downsample_padding": 1, + "dropout": 0.0, + "dual_cross_attention": false, + "encoder_hid_dim": null, + "encoder_hid_dim_type": null, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 8, + "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_attention_heads": null, + "num_class_embeds": null, + "only_cross_attention": false, + "out_channels": 4, + "projection_class_embeddings_input_dim": null, + "resnet_out_scale_factor": 1.0, + "resnet_skip_time_act": false, + "resnet_time_scale_shift": "default", + "reverse_transformer_layers_per_block": null, + "sample_size": 64, + "time_cond_proj_dim": null, + "time_embedding_act_fn": null, + "time_embedding_dim": null, + "time_embedding_type": "positional", + "timestep_post_act": null, + "transformer_layers_per_block": 1, + "up_block_types": [ + "UpBlock2D", + "CrossAttnUpBlock2D", + "CrossAttnUpBlock2D", + "CrossAttnUpBlock2D" + ], + "upcast_attention": false, + "use_linear_projection": false +} diff --git a/checkpoint-10000/unet/diffusion_pytorch_model.safetensors b/checkpoint-10000/unet/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..e7da2172a1864b41333cba08782833ffdb279e5a --- /dev/null +++ b/checkpoint-10000/unet/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:6a57a4c6cf76e35b0080ceebdf52fa343ffc4b161b86297f71ea9fd348bc8eed +size 3438213624 diff --git a/checkpoint-3000/optimizer.bin b/checkpoint-3000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..a7ac6893ddb0acf1dc183f082de6a42bbbc0bb39 --- /dev/null +++ b/checkpoint-3000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:707c2796fab1c9c3df94c4f8bb22912472d716067e59af766e2205e23a127175 +size 6876851924 diff --git a/checkpoint-3000/random_states_0.pkl b/checkpoint-3000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..33f04b666ab0e887115e271ec9e813ae032799b6 --- /dev/null +++ b/checkpoint-3000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:220b232be9bbfe853811f01ac489a5119d52cec68731bb25417c1be078c78e6b +size 16036 diff --git a/checkpoint-3000/scheduler.bin b/checkpoint-3000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..1c18059bf450239f69cc8214bdd26a7c5638e541 --- /dev/null +++ b/checkpoint-3000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:9bcaa0ec9565c3bec056c6136ed24ab62a762aedcf6cdfb8053f701a68b6d056 +size 1000 diff --git a/checkpoint-3000/unet/config.json b/checkpoint-3000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..ccb8474073252815aa720c280c23b28669d7a727 --- /dev/null +++ b/checkpoint-3000/unet/config.json @@ -0,0 +1,68 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.33.1", + "_name_or_path": "output_lr1e4_accstep1/checkpoint-2500", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "addition_time_embed_dim": null, + "attention_head_dim": 8, + "attention_type": "default", + "block_out_channels": [ + 320, + 640, + 1280, + 1280 + ], + "center_input_sample": false, + "class_embed_type": null, + "class_embeddings_concat": false, + "conv_in_kernel": 3, + "conv_out_kernel": 3, + "cross_attention_dim": 768, + "cross_attention_norm": null, + "down_block_types": [ + "CrossAttnDownBlock2D", + "CrossAttnDownBlock2D", + "CrossAttnDownBlock2D", + "DownBlock2D" + ], + "downsample_padding": 1, + "dropout": 0.0, + "dual_cross_attention": false, + "encoder_hid_dim": null, + "encoder_hid_dim_type": null, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 8, + "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_attention_heads": null, + "num_class_embeds": null, + "only_cross_attention": false, + "out_channels": 4, + "projection_class_embeddings_input_dim": null, + "resnet_out_scale_factor": 1.0, + "resnet_skip_time_act": false, + "resnet_time_scale_shift": "default", + "reverse_transformer_layers_per_block": null, + "sample_size": 64, + "time_cond_proj_dim": null, + "time_embedding_act_fn": null, + "time_embedding_dim": null, + "time_embedding_type": "positional", + "timestep_post_act": null, + "transformer_layers_per_block": 1, + "up_block_types": [ + "UpBlock2D", + "CrossAttnUpBlock2D", + "CrossAttnUpBlock2D", + "CrossAttnUpBlock2D" + ], + "upcast_attention": false, + "use_linear_projection": false +} diff --git a/checkpoint-3000/unet/diffusion_pytorch_model.safetensors b/checkpoint-3000/unet/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..8512ff227183e91dbef446ef028aa61ab10a25b3 --- /dev/null +++ b/checkpoint-3000/unet/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:54e5fb0b20cdad7b1281731c1cdcfc4c7a5f6cedc2ff2581f68a9ffc5cdff593 +size 3438213624 diff --git a/checkpoint-3500/optimizer.bin b/checkpoint-3500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..1bd598c87920c31593ae73306f095d9027a76019 --- /dev/null +++ b/checkpoint-3500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:5e9cd05979fbd4c5b84d3c39993d89717a616d73fdc7ce64396890c88836302c +size 6876851924 diff --git a/checkpoint-3500/random_states_0.pkl b/checkpoint-3500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..37953a14af86567c8c84ac6fd3b70ca8be7eb05b --- /dev/null +++ b/checkpoint-3500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:da83a3d3f9519f2b1b86b36375b3ed32528f1b77a7e3db4283ac73219d56be3d +size 16036 diff --git a/checkpoint-3500/scheduler.bin b/checkpoint-3500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..1b973fb9993a974476663dc6de24b347bf4f4ee5 --- /dev/null +++ b/checkpoint-3500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e6dd533cc093c46d0767cb255242c6e7c5706aaffabe304eeb0030f224e5cb82 +size 1000 diff --git a/checkpoint-3500/unet/config.json b/checkpoint-3500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..ccb8474073252815aa720c280c23b28669d7a727 --- /dev/null +++ b/checkpoint-3500/unet/config.json @@ -0,0 +1,68 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.33.1", + "_name_or_path": "output_lr1e4_accstep1/checkpoint-2500", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "addition_time_embed_dim": null, + "attention_head_dim": 8, + "attention_type": "default", + "block_out_channels": [ + 320, + 640, + 1280, + 1280 + ], + "center_input_sample": false, + "class_embed_type": null, + "class_embeddings_concat": false, + "conv_in_kernel": 3, + "conv_out_kernel": 3, + "cross_attention_dim": 768, + "cross_attention_norm": null, + "down_block_types": [ + "CrossAttnDownBlock2D", + "CrossAttnDownBlock2D", + "CrossAttnDownBlock2D", + "DownBlock2D" + ], + "downsample_padding": 1, + "dropout": 0.0, + "dual_cross_attention": false, + "encoder_hid_dim": null, + "encoder_hid_dim_type": null, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 8, + "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_attention_heads": null, + "num_class_embeds": null, + "only_cross_attention": false, + "out_channels": 4, + "projection_class_embeddings_input_dim": null, + "resnet_out_scale_factor": 1.0, + "resnet_skip_time_act": false, + "resnet_time_scale_shift": "default", + "reverse_transformer_layers_per_block": null, + "sample_size": 64, + "time_cond_proj_dim": null, + "time_embedding_act_fn": null, + "time_embedding_dim": null, + "time_embedding_type": "positional", + "timestep_post_act": null, + "transformer_layers_per_block": 1, + "up_block_types": [ + "UpBlock2D", + "CrossAttnUpBlock2D", + "CrossAttnUpBlock2D", + "CrossAttnUpBlock2D" + ], + "upcast_attention": false, + "use_linear_projection": false +} diff --git a/checkpoint-3500/unet/diffusion_pytorch_model.safetensors b/checkpoint-3500/unet/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..6970a76c128e7bebd8c461c83a393a27519d85a9 --- /dev/null +++ b/checkpoint-3500/unet/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c435c1082274dafc018638ff7840a45d0558ed06169782b3b81889d307615b0a +size 3438213624 diff --git a/checkpoint-4000/optimizer.bin b/checkpoint-4000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..86ea558fea389c1da5f90ff409a2c14ce8fe569c --- /dev/null +++ b/checkpoint-4000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:a50743c3359be0033d140cb875d2d3fee75ef3c947bee401a9765674b9da9be4 +size 6876851924 diff --git a/checkpoint-4000/random_states_0.pkl b/checkpoint-4000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..a3ddbdeb31128313046def4fd1fe22842a039399 --- /dev/null +++ b/checkpoint-4000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:1fcdfa275b6f2387d23af677716c224c5371737f2be51d61d3297763c86113ef +size 16036 diff --git a/checkpoint-4000/scheduler.bin b/checkpoint-4000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..76902b6d725b8df171a141e9018f3d3ecd64921a --- /dev/null +++ b/checkpoint-4000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:2bbb61109d01da8a9c381c0e2f077875e9118afeba0c4ed8b289ee114f675793 +size 1000 diff --git a/checkpoint-4000/unet/config.json b/checkpoint-4000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..ccb8474073252815aa720c280c23b28669d7a727 --- /dev/null +++ b/checkpoint-4000/unet/config.json @@ -0,0 +1,68 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.33.1", + "_name_or_path": "output_lr1e4_accstep1/checkpoint-2500", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "addition_time_embed_dim": null, + "attention_head_dim": 8, + "attention_type": "default", + "block_out_channels": [ + 320, + 640, + 1280, + 1280 + ], + "center_input_sample": false, + "class_embed_type": null, + "class_embeddings_concat": false, + "conv_in_kernel": 3, + "conv_out_kernel": 3, + "cross_attention_dim": 768, + "cross_attention_norm": null, + "down_block_types": [ + "CrossAttnDownBlock2D", + "CrossAttnDownBlock2D", + "CrossAttnDownBlock2D", + "DownBlock2D" + ], + "downsample_padding": 1, + "dropout": 0.0, + "dual_cross_attention": false, + "encoder_hid_dim": null, + "encoder_hid_dim_type": null, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 8, + "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_attention_heads": null, + "num_class_embeds": null, + "only_cross_attention": false, + "out_channels": 4, + "projection_class_embeddings_input_dim": null, + "resnet_out_scale_factor": 1.0, + "resnet_skip_time_act": false, + "resnet_time_scale_shift": "default", + "reverse_transformer_layers_per_block": null, + "sample_size": 64, + "time_cond_proj_dim": null, + "time_embedding_act_fn": null, + "time_embedding_dim": null, + "time_embedding_type": "positional", + "timestep_post_act": null, + "transformer_layers_per_block": 1, + "up_block_types": [ + "UpBlock2D", + "CrossAttnUpBlock2D", + "CrossAttnUpBlock2D", + "CrossAttnUpBlock2D" + ], + "upcast_attention": false, + "use_linear_projection": false +} diff --git a/checkpoint-4000/unet/diffusion_pytorch_model.safetensors b/checkpoint-4000/unet/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..e9cc36e1009a85db77f1eab247a49b9b80791dc8 --- /dev/null +++ b/checkpoint-4000/unet/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:8b93ecda3f8ff10b0746f846028b798da5257eefc6994a9f30415dfe37b9ee20 +size 3438213624 diff --git a/checkpoint-4500/optimizer.bin b/checkpoint-4500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..aba343ae87835bb8eb6d41c01d72447fcc59dd0b --- /dev/null +++ b/checkpoint-4500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:f2fd48a173e9fc3721051c157f53995acef3da258ed583f3bb4ba6499575d287 +size 6876851924 diff --git a/checkpoint-4500/random_states_0.pkl b/checkpoint-4500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..7f6c8d196d89539e457aba44f8f784a764f2c078 --- /dev/null +++ b/checkpoint-4500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:1b431d09474525c108cee6293a8cd5ec5bab166969f3d8059e44def498599a0d +size 16036 diff --git a/checkpoint-4500/scheduler.bin b/checkpoint-4500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..99dac86b7e870102e7082b39c5f9bb61b560c706 --- /dev/null +++ b/checkpoint-4500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c053c1662beab785ce3130a131593c5ee64e664223403b2aab6ac3c68b0dd77b +size 1000 diff --git a/checkpoint-4500/unet/config.json b/checkpoint-4500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..ccb8474073252815aa720c280c23b28669d7a727 --- /dev/null +++ b/checkpoint-4500/unet/config.json @@ -0,0 +1,68 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.33.1", + "_name_or_path": "output_lr1e4_accstep1/checkpoint-2500", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "addition_time_embed_dim": null, + "attention_head_dim": 8, + "attention_type": "default", + "block_out_channels": [ + 320, + 640, + 1280, + 1280 + ], + "center_input_sample": false, + "class_embed_type": null, + "class_embeddings_concat": false, + "conv_in_kernel": 3, + "conv_out_kernel": 3, + "cross_attention_dim": 768, + "cross_attention_norm": null, + "down_block_types": [ + "CrossAttnDownBlock2D", + "CrossAttnDownBlock2D", + "CrossAttnDownBlock2D", + "DownBlock2D" + ], + "downsample_padding": 1, + "dropout": 0.0, + "dual_cross_attention": false, + "encoder_hid_dim": null, + "encoder_hid_dim_type": null, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 8, + "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_attention_heads": null, + "num_class_embeds": null, + "only_cross_attention": false, + "out_channels": 4, + "projection_class_embeddings_input_dim": null, + "resnet_out_scale_factor": 1.0, + "resnet_skip_time_act": false, + "resnet_time_scale_shift": "default", + "reverse_transformer_layers_per_block": null, + "sample_size": 64, + "time_cond_proj_dim": null, + "time_embedding_act_fn": null, + "time_embedding_dim": null, + "time_embedding_type": "positional", + "timestep_post_act": null, + "transformer_layers_per_block": 1, + "up_block_types": [ + "UpBlock2D", + "CrossAttnUpBlock2D", + "CrossAttnUpBlock2D", + "CrossAttnUpBlock2D" + ], + "upcast_attention": false, + "use_linear_projection": false +} diff --git a/checkpoint-4500/unet/diffusion_pytorch_model.safetensors b/checkpoint-4500/unet/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..50158680bbc32e73bff7f9d89b73a17ff8ba034a --- /dev/null +++ b/checkpoint-4500/unet/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:6bb72de479461831b078df52b5b4f52947df4751a0e1208f50a969fe4aefbd70 +size 3438213624 diff --git a/checkpoint-5000/optimizer.bin b/checkpoint-5000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..59eef2679af4325858c573853e7559bf7ebbfc8e --- /dev/null +++ b/checkpoint-5000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:092acab10ec2828546df5f7597546d03474165e53f9c19440635182b411dcd65 +size 6876851924 diff --git a/checkpoint-5000/random_states_0.pkl b/checkpoint-5000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..dfec15ede7305006ab91871a754ae645c537ab91 --- /dev/null +++ b/checkpoint-5000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c5edd3c41f82b01d67bef5f6644b2c773a8d1575cddd575484f6182be16f4869 +size 16036 diff --git a/checkpoint-5000/scheduler.bin b/checkpoint-5000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..b67428c3bc5f73921cbbe89ecc130a0dc966ea0e --- /dev/null +++ b/checkpoint-5000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:0ac5722a50941bc1353846f80a544eb191fa869ae625cb63833f25ef78114c2d +size 1000 diff --git a/checkpoint-5000/unet/config.json b/checkpoint-5000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..ccb8474073252815aa720c280c23b28669d7a727 --- /dev/null +++ b/checkpoint-5000/unet/config.json @@ -0,0 +1,68 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.33.1", + "_name_or_path": "output_lr1e4_accstep1/checkpoint-2500", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "addition_time_embed_dim": null, + "attention_head_dim": 8, + "attention_type": "default", + "block_out_channels": [ + 320, + 640, + 1280, + 1280 + ], + "center_input_sample": false, + "class_embed_type": null, + "class_embeddings_concat": false, + "conv_in_kernel": 3, + "conv_out_kernel": 3, + "cross_attention_dim": 768, + "cross_attention_norm": null, + "down_block_types": [ + "CrossAttnDownBlock2D", + "CrossAttnDownBlock2D", + "CrossAttnDownBlock2D", + "DownBlock2D" + ], + "downsample_padding": 1, + "dropout": 0.0, + "dual_cross_attention": false, + "encoder_hid_dim": null, + "encoder_hid_dim_type": null, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 8, + "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_attention_heads": null, + "num_class_embeds": null, + "only_cross_attention": false, + "out_channels": 4, + "projection_class_embeddings_input_dim": null, + "resnet_out_scale_factor": 1.0, + "resnet_skip_time_act": false, + "resnet_time_scale_shift": "default", + "reverse_transformer_layers_per_block": null, + "sample_size": 64, + "time_cond_proj_dim": null, + "time_embedding_act_fn": null, + "time_embedding_dim": null, + "time_embedding_type": "positional", + "timestep_post_act": null, + "transformer_layers_per_block": 1, + "up_block_types": [ + "UpBlock2D", + "CrossAttnUpBlock2D", + "CrossAttnUpBlock2D", + "CrossAttnUpBlock2D" + ], + "upcast_attention": false, + "use_linear_projection": false +} diff --git a/checkpoint-5000/unet/diffusion_pytorch_model.safetensors b/checkpoint-5000/unet/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..cfffb32a57f7b30a4b56783179a46907b7e184cb --- /dev/null +++ b/checkpoint-5000/unet/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:43d5668157c9f4a4ba4ae19e1da3358dd4bfb53080952f3059b8f610e0b5e451 +size 3438213624 diff --git a/checkpoint-5500/optimizer.bin b/checkpoint-5500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..621c5c36b31c108b9d23ca3b0885cdcc350659f9 --- /dev/null +++ b/checkpoint-5500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:cbe59c58e3f1fb4236cea0852759011559cce25419539f8fc181766c85e1dc6c +size 6876851924 diff --git a/checkpoint-5500/random_states_0.pkl b/checkpoint-5500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..a01101f96d38768a622e3bd1a49b43fdee373ce9 --- /dev/null +++ b/checkpoint-5500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:8a18bddc91be7b30dc7afacd088c0d3f0f2031ab6678eaf70facae3ddf525a8c +size 16036 diff --git a/checkpoint-5500/scheduler.bin b/checkpoint-5500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..062f1a3a77ad5b39fa2fed00cdabc1932345f88a --- /dev/null +++ b/checkpoint-5500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:516c02083a8c2efe70f2b16f163598b1a81f26e2dfe21e359792352144a2ecd7 +size 1000 diff --git a/checkpoint-5500/unet/config.json b/checkpoint-5500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..ccb8474073252815aa720c280c23b28669d7a727 --- /dev/null +++ b/checkpoint-5500/unet/config.json @@ -0,0 +1,68 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.33.1", + "_name_or_path": "output_lr1e4_accstep1/checkpoint-2500", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "addition_time_embed_dim": null, + "attention_head_dim": 8, + "attention_type": "default", + "block_out_channels": [ + 320, + 640, + 1280, + 1280 + ], + "center_input_sample": false, + "class_embed_type": null, + "class_embeddings_concat": false, + "conv_in_kernel": 3, + "conv_out_kernel": 3, + "cross_attention_dim": 768, + "cross_attention_norm": null, + "down_block_types": [ + "CrossAttnDownBlock2D", + "CrossAttnDownBlock2D", + "CrossAttnDownBlock2D", + "DownBlock2D" + ], + "downsample_padding": 1, + "dropout": 0.0, + "dual_cross_attention": false, + "encoder_hid_dim": null, + "encoder_hid_dim_type": null, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 8, + "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_attention_heads": null, + "num_class_embeds": null, + "only_cross_attention": false, + "out_channels": 4, + "projection_class_embeddings_input_dim": null, + "resnet_out_scale_factor": 1.0, + "resnet_skip_time_act": false, + "resnet_time_scale_shift": "default", + "reverse_transformer_layers_per_block": null, + "sample_size": 64, + "time_cond_proj_dim": null, + "time_embedding_act_fn": null, + "time_embedding_dim": null, + "time_embedding_type": "positional", + "timestep_post_act": null, + "transformer_layers_per_block": 1, + "up_block_types": [ + "UpBlock2D", + "CrossAttnUpBlock2D", + "CrossAttnUpBlock2D", + "CrossAttnUpBlock2D" + ], + "upcast_attention": false, + "use_linear_projection": false +} diff --git a/checkpoint-5500/unet/diffusion_pytorch_model.safetensors b/checkpoint-5500/unet/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..051381842ea82d583dbdbe4e44b733fa45183149 --- /dev/null +++ b/checkpoint-5500/unet/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c5af09d691d14b564893972ec371e4e3495814abebd5ce592b3e9111be8e4c7d +size 3438213624 diff --git a/checkpoint-6000/optimizer.bin b/checkpoint-6000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..3f7611defb36063a5be7fe44308ca4c81558fa6d --- /dev/null +++ b/checkpoint-6000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:da2e6b0bd8f081e935a1e58a451c1e911ea30ea780c3aeda7c0fe91287ee2431 +size 6876851924 diff --git a/checkpoint-6000/random_states_0.pkl b/checkpoint-6000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..0dcc9ca117eb0d686b8aebbd59d298b0ba68baf8 --- /dev/null +++ b/checkpoint-6000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:2274f79941a4f0c4280617724cb50185791dc99a8104be4ccbdbed9aa6140f07 +size 16036 diff --git a/checkpoint-6000/scheduler.bin b/checkpoint-6000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..f516aa6739513829cc858088b23d507c5ad68799 --- /dev/null +++ b/checkpoint-6000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:255cbb0c1e6c238bd2b866da2c61bb79c63e9f06e564149d80740a7a1ead0f1a +size 1000 diff --git a/checkpoint-6000/unet/config.json b/checkpoint-6000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..ccb8474073252815aa720c280c23b28669d7a727 --- /dev/null +++ b/checkpoint-6000/unet/config.json @@ -0,0 +1,68 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.33.1", + "_name_or_path": "output_lr1e4_accstep1/checkpoint-2500", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "addition_time_embed_dim": null, + "attention_head_dim": 8, + "attention_type": "default", + "block_out_channels": [ + 320, + 640, + 1280, + 1280 + ], + "center_input_sample": false, + "class_embed_type": null, + "class_embeddings_concat": false, + "conv_in_kernel": 3, + "conv_out_kernel": 3, + "cross_attention_dim": 768, + "cross_attention_norm": null, + "down_block_types": [ + "CrossAttnDownBlock2D", + "CrossAttnDownBlock2D", + "CrossAttnDownBlock2D", + "DownBlock2D" + ], + "downsample_padding": 1, + "dropout": 0.0, + "dual_cross_attention": false, + "encoder_hid_dim": null, + "encoder_hid_dim_type": null, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 8, + "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_attention_heads": null, + "num_class_embeds": null, + "only_cross_attention": false, + "out_channels": 4, + "projection_class_embeddings_input_dim": null, + "resnet_out_scale_factor": 1.0, + "resnet_skip_time_act": false, + "resnet_time_scale_shift": "default", + "reverse_transformer_layers_per_block": null, + "sample_size": 64, + "time_cond_proj_dim": null, + "time_embedding_act_fn": null, + "time_embedding_dim": null, + "time_embedding_type": "positional", + "timestep_post_act": null, + "transformer_layers_per_block": 1, + "up_block_types": [ + "UpBlock2D", + "CrossAttnUpBlock2D", + "CrossAttnUpBlock2D", + "CrossAttnUpBlock2D" + ], + "upcast_attention": false, + "use_linear_projection": false +} diff --git a/checkpoint-6000/unet/diffusion_pytorch_model.safetensors b/checkpoint-6000/unet/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..4b0ec938c6bd0493a3a785f458c3277e1b6ea290 --- /dev/null +++ b/checkpoint-6000/unet/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:6770a2f9beae12fc9a320bbecb6d5bbf49c967e52bbdc670f7b904ca47b30d64 +size 3438213624 diff --git a/checkpoint-6500/optimizer.bin b/checkpoint-6500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..845e697051ebb646fc1a1f713313d77fe97014b8 --- /dev/null +++ b/checkpoint-6500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:6e6a4ee215f3ced6730af0e2029c6d6263caf19c394bb8b405590ae7cbb59291 +size 6876851924 diff --git a/checkpoint-6500/random_states_0.pkl b/checkpoint-6500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..4899dc83be878526ebdba86c942299948c884220 --- /dev/null +++ b/checkpoint-6500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:6e5b816b3dac981a404250238b660b57ac0028ee19606c0d7b29f89df9b7e6f8 +size 16036 diff --git a/checkpoint-6500/scheduler.bin b/checkpoint-6500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..e8e7366b44264aa6f5dc67d253fb876256dc6cf4 --- /dev/null +++ b/checkpoint-6500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:2eb0d3104e9fdadbf301a53e4752edbcf847bbbd972015cca0c16a25d4dfb728 +size 1000 diff --git a/checkpoint-6500/unet/config.json b/checkpoint-6500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..ccb8474073252815aa720c280c23b28669d7a727 --- /dev/null +++ b/checkpoint-6500/unet/config.json @@ -0,0 +1,68 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.33.1", + "_name_or_path": "output_lr1e4_accstep1/checkpoint-2500", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "addition_time_embed_dim": null, + "attention_head_dim": 8, + "attention_type": "default", + "block_out_channels": [ + 320, + 640, + 1280, + 1280 + ], + "center_input_sample": false, + "class_embed_type": null, + "class_embeddings_concat": false, + "conv_in_kernel": 3, + "conv_out_kernel": 3, + "cross_attention_dim": 768, + "cross_attention_norm": null, + "down_block_types": [ + "CrossAttnDownBlock2D", + "CrossAttnDownBlock2D", + "CrossAttnDownBlock2D", + "DownBlock2D" + ], + "downsample_padding": 1, + "dropout": 0.0, + "dual_cross_attention": false, + "encoder_hid_dim": null, + "encoder_hid_dim_type": null, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 8, + "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_attention_heads": null, + "num_class_embeds": null, + "only_cross_attention": false, + "out_channels": 4, + "projection_class_embeddings_input_dim": null, + "resnet_out_scale_factor": 1.0, + "resnet_skip_time_act": false, + "resnet_time_scale_shift": "default", + "reverse_transformer_layers_per_block": null, + "sample_size": 64, + "time_cond_proj_dim": null, + "time_embedding_act_fn": null, + "time_embedding_dim": null, + "time_embedding_type": "positional", + "timestep_post_act": null, + "transformer_layers_per_block": 1, + "up_block_types": [ + "UpBlock2D", + "CrossAttnUpBlock2D", + "CrossAttnUpBlock2D", + "CrossAttnUpBlock2D" + ], + "upcast_attention": false, + "use_linear_projection": false +} diff --git a/checkpoint-6500/unet/diffusion_pytorch_model.safetensors b/checkpoint-6500/unet/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..34d8c02089a75b336bc36d5a38d2bac0dce3e78b --- /dev/null +++ b/checkpoint-6500/unet/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:f439239f4fe9d15017ff76589113a86145cd2d4e723d47e4324001bbc2bdd58e +size 3438213624 diff --git a/checkpoint-7000/optimizer.bin b/checkpoint-7000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..e426a84c017af534562c1c1bc66eaf70f63570f7 --- /dev/null +++ b/checkpoint-7000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:3f4aa50b9a48fd697fa7d69f9062565465bea68bdccbdb7351da572c21e010b9 +size 6876851924 diff --git a/checkpoint-7000/random_states_0.pkl b/checkpoint-7000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..ce0558d3b622e0a221358f9f92fdc87d839bbfdd --- /dev/null +++ b/checkpoint-7000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:d360a7b6036461aad269b99053532375b86cc2ad817802417ae0d72b346530ab +size 16036 diff --git a/checkpoint-7000/scheduler.bin b/checkpoint-7000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..ebb5bc8361c344dc6fb0dbb1440e37c6a1633549 --- /dev/null +++ b/checkpoint-7000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:62df1a56ed6220fcb39701058bc5e392456feb08d80b106051c1759c5678e983 +size 1000 diff --git a/checkpoint-7000/unet/config.json b/checkpoint-7000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..ccb8474073252815aa720c280c23b28669d7a727 --- /dev/null +++ b/checkpoint-7000/unet/config.json @@ -0,0 +1,68 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.33.1", + "_name_or_path": "output_lr1e4_accstep1/checkpoint-2500", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "addition_time_embed_dim": null, + "attention_head_dim": 8, + "attention_type": "default", + "block_out_channels": [ + 320, + 640, + 1280, + 1280 + ], + "center_input_sample": false, + "class_embed_type": null, + "class_embeddings_concat": false, + "conv_in_kernel": 3, + "conv_out_kernel": 3, + "cross_attention_dim": 768, + "cross_attention_norm": null, + "down_block_types": [ + "CrossAttnDownBlock2D", + "CrossAttnDownBlock2D", + "CrossAttnDownBlock2D", + "DownBlock2D" + ], + "downsample_padding": 1, + "dropout": 0.0, + "dual_cross_attention": false, + "encoder_hid_dim": null, + "encoder_hid_dim_type": null, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 8, + "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_attention_heads": null, + "num_class_embeds": null, + "only_cross_attention": false, + "out_channels": 4, + "projection_class_embeddings_input_dim": null, + "resnet_out_scale_factor": 1.0, + "resnet_skip_time_act": false, + "resnet_time_scale_shift": "default", + "reverse_transformer_layers_per_block": null, + "sample_size": 64, + "time_cond_proj_dim": null, + "time_embedding_act_fn": null, + "time_embedding_dim": null, + "time_embedding_type": "positional", + "timestep_post_act": null, + "transformer_layers_per_block": 1, + "up_block_types": [ + "UpBlock2D", + "CrossAttnUpBlock2D", + "CrossAttnUpBlock2D", + "CrossAttnUpBlock2D" + ], + "upcast_attention": false, + "use_linear_projection": false +} diff --git a/checkpoint-7000/unet/diffusion_pytorch_model.safetensors b/checkpoint-7000/unet/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..2fa3446036ecce3edbd68de2b3eb637efc1db378 --- /dev/null +++ b/checkpoint-7000/unet/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:1a73868c220b4300bf274975e69471d92e179975a9d4ff6d0a11a0490d5aa11f +size 3438213624 diff --git a/checkpoint-7500/optimizer.bin b/checkpoint-7500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..f79b2869cd5b541735c7a97dc00d0d7e43f0e1c1 --- /dev/null +++ b/checkpoint-7500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:f4fe8e804913ba5e3567a1dd905f6e4f61d294dba9f526feb888db3224080274 +size 6876851924 diff --git a/checkpoint-7500/random_states_0.pkl b/checkpoint-7500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..5a5312d8577015b6f7f6bed6e4e4cfeeeb1fd711 --- /dev/null +++ b/checkpoint-7500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7bd5b658ae4cc0bc641631ea5fa8dffa98c1c64c067f3ef90124531651ee061c +size 16036 diff --git a/checkpoint-7500/scheduler.bin b/checkpoint-7500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..9a1013505b642b3d8e1144b4030e3996aec79ea3 --- /dev/null +++ b/checkpoint-7500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:18cd505f54c1eee6105391ce0a3f1b567a5b10c4bef08f9a5176846f72f5d495 +size 1000 diff --git a/checkpoint-7500/unet/config.json b/checkpoint-7500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..ccb8474073252815aa720c280c23b28669d7a727 --- /dev/null +++ b/checkpoint-7500/unet/config.json @@ -0,0 +1,68 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.33.1", + "_name_or_path": "output_lr1e4_accstep1/checkpoint-2500", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "addition_time_embed_dim": null, + "attention_head_dim": 8, + "attention_type": "default", + "block_out_channels": [ + 320, + 640, + 1280, + 1280 + ], + "center_input_sample": false, + "class_embed_type": null, + "class_embeddings_concat": false, + "conv_in_kernel": 3, + "conv_out_kernel": 3, + "cross_attention_dim": 768, + "cross_attention_norm": null, + "down_block_types": [ + "CrossAttnDownBlock2D", + "CrossAttnDownBlock2D", + "CrossAttnDownBlock2D", + "DownBlock2D" + ], + "downsample_padding": 1, + "dropout": 0.0, + "dual_cross_attention": false, + "encoder_hid_dim": null, + "encoder_hid_dim_type": null, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 8, + "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_attention_heads": null, + "num_class_embeds": null, + "only_cross_attention": false, + "out_channels": 4, + "projection_class_embeddings_input_dim": null, + "resnet_out_scale_factor": 1.0, + "resnet_skip_time_act": false, + "resnet_time_scale_shift": "default", + "reverse_transformer_layers_per_block": null, + "sample_size": 64, + "time_cond_proj_dim": null, + "time_embedding_act_fn": null, + "time_embedding_dim": null, + "time_embedding_type": "positional", + "timestep_post_act": null, + "transformer_layers_per_block": 1, + "up_block_types": [ + "UpBlock2D", + "CrossAttnUpBlock2D", + "CrossAttnUpBlock2D", + "CrossAttnUpBlock2D" + ], + "upcast_attention": false, + "use_linear_projection": false +} diff --git a/checkpoint-7500/unet/diffusion_pytorch_model.safetensors b/checkpoint-7500/unet/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..19307045b46de566f343803353846183848f3b11 --- /dev/null +++ b/checkpoint-7500/unet/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:13120705f0bee7767e6033b2de83031fdf4a2498929e6beffd07071b25bc563e +size 3438213624 diff --git a/checkpoint-8000/optimizer.bin b/checkpoint-8000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..cba56fcbb80a0e70c7ac6e7c73871a80129e2aad --- /dev/null +++ b/checkpoint-8000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:dacb84418960b36e43d46d59dc3a7553060c4bb629d8c00effb03a1e43190b24 +size 6876851924 diff --git a/checkpoint-8000/random_states_0.pkl b/checkpoint-8000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..7323a2fad1ebc267e016ca128ce5c9b591cdd724 --- /dev/null +++ b/checkpoint-8000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:2e20488220ee2b716dcadf924a0981b011769c97f38296ec37d5d1416bdc157a +size 16036 diff --git a/checkpoint-8000/scheduler.bin b/checkpoint-8000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..947c59bb44d0234fc7ae4ed4803916c1b7692340 --- /dev/null +++ b/checkpoint-8000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7409d6638c846703c58e7ad4675c17dcf2e6a81b4a44c75ee478ef77e141fa91 +size 1000 diff --git a/checkpoint-8000/unet/config.json b/checkpoint-8000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..ccb8474073252815aa720c280c23b28669d7a727 --- /dev/null +++ b/checkpoint-8000/unet/config.json @@ -0,0 +1,68 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.33.1", + "_name_or_path": "output_lr1e4_accstep1/checkpoint-2500", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "addition_time_embed_dim": null, + "attention_head_dim": 8, + "attention_type": "default", + "block_out_channels": [ + 320, + 640, + 1280, + 1280 + ], + "center_input_sample": false, + "class_embed_type": null, + "class_embeddings_concat": false, + "conv_in_kernel": 3, + "conv_out_kernel": 3, + "cross_attention_dim": 768, + "cross_attention_norm": null, + "down_block_types": [ + "CrossAttnDownBlock2D", + "CrossAttnDownBlock2D", + "CrossAttnDownBlock2D", + "DownBlock2D" + ], + "downsample_padding": 1, + "dropout": 0.0, + "dual_cross_attention": false, + "encoder_hid_dim": null, + "encoder_hid_dim_type": null, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 8, + "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_attention_heads": null, + "num_class_embeds": null, + "only_cross_attention": false, + "out_channels": 4, + "projection_class_embeddings_input_dim": null, + "resnet_out_scale_factor": 1.0, + "resnet_skip_time_act": false, + "resnet_time_scale_shift": "default", + "reverse_transformer_layers_per_block": null, + "sample_size": 64, + "time_cond_proj_dim": null, + "time_embedding_act_fn": null, + "time_embedding_dim": null, + "time_embedding_type": "positional", + "timestep_post_act": null, + "transformer_layers_per_block": 1, + "up_block_types": [ + "UpBlock2D", + "CrossAttnUpBlock2D", + "CrossAttnUpBlock2D", + "CrossAttnUpBlock2D" + ], + "upcast_attention": false, + "use_linear_projection": false +} diff --git a/checkpoint-8000/unet/diffusion_pytorch_model.safetensors b/checkpoint-8000/unet/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..4e93630a189e9286bf452b8c2ffbf82f93b6d1a2 --- /dev/null +++ b/checkpoint-8000/unet/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:3419f75f6453b672a8b73c2bb28b8851d36c504978174da8f36de05124109e2a +size 3438213624 diff --git a/checkpoint-8500/optimizer.bin b/checkpoint-8500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..61bc362ba9e9ae763b6ed586855eac14c478f808 --- /dev/null +++ b/checkpoint-8500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:360e4cdcb44041ead974c50467b6768b51ef7d1d1cc57ceda6dbd7a368ee3707 +size 6876851924 diff --git a/checkpoint-8500/random_states_0.pkl b/checkpoint-8500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..ed89f97bc3968bef40dac6bed22b1f5d37586fd1 --- /dev/null +++ b/checkpoint-8500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7353fb9c0990af43f5f39f0943eb83f8bafe8acb353e237449f8fe7a6f7f97f9 +size 16036 diff --git a/checkpoint-8500/scheduler.bin b/checkpoint-8500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..58e144c9d4c082667bef559fb5b9e5fe3d8f45db --- /dev/null +++ b/checkpoint-8500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:9c89d8d8bccec36ab9e88319aaec755ad9fa4195782c3cca5ea61d0e809ff57e +size 1000 diff --git a/checkpoint-8500/unet/config.json b/checkpoint-8500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..ccb8474073252815aa720c280c23b28669d7a727 --- /dev/null +++ b/checkpoint-8500/unet/config.json @@ -0,0 +1,68 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.33.1", + "_name_or_path": "output_lr1e4_accstep1/checkpoint-2500", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "addition_time_embed_dim": null, + "attention_head_dim": 8, + "attention_type": "default", + "block_out_channels": [ + 320, + 640, + 1280, + 1280 + ], + "center_input_sample": false, + "class_embed_type": null, + "class_embeddings_concat": false, + "conv_in_kernel": 3, + "conv_out_kernel": 3, + "cross_attention_dim": 768, + "cross_attention_norm": null, + "down_block_types": [ + "CrossAttnDownBlock2D", + "CrossAttnDownBlock2D", + "CrossAttnDownBlock2D", + "DownBlock2D" + ], + "downsample_padding": 1, + "dropout": 0.0, + "dual_cross_attention": false, + "encoder_hid_dim": null, + "encoder_hid_dim_type": null, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 8, + "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_attention_heads": null, + "num_class_embeds": null, + "only_cross_attention": false, + "out_channels": 4, + "projection_class_embeddings_input_dim": null, + "resnet_out_scale_factor": 1.0, + "resnet_skip_time_act": false, + "resnet_time_scale_shift": "default", + "reverse_transformer_layers_per_block": null, + "sample_size": 64, + "time_cond_proj_dim": null, + "time_embedding_act_fn": null, + "time_embedding_dim": null, + "time_embedding_type": "positional", + "timestep_post_act": null, + "transformer_layers_per_block": 1, + "up_block_types": [ + "UpBlock2D", + "CrossAttnUpBlock2D", + "CrossAttnUpBlock2D", + "CrossAttnUpBlock2D" + ], + "upcast_attention": false, + "use_linear_projection": false +} diff --git a/checkpoint-8500/unet/diffusion_pytorch_model.safetensors b/checkpoint-8500/unet/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..a5e2bfcd552a81c8d3afa85b1204c903b431477f --- /dev/null +++ b/checkpoint-8500/unet/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:bd39eb1d743179c0484288ee3283b41e3598367b2a05e34d94aed1276c8557d0 +size 3438213624 diff --git a/checkpoint-9000/optimizer.bin b/checkpoint-9000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..3d84f76e8af6411375506086c34c85afc63a7239 --- /dev/null +++ b/checkpoint-9000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:ca3f5f11930a1898028292ea243af6d3b5c4f2a6a77d9212df41fee9b4b95342 +size 6876851924 diff --git a/checkpoint-9000/random_states_0.pkl b/checkpoint-9000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..a3eb50be34fe9a198a41e2adcbccee941e048caf --- /dev/null +++ b/checkpoint-9000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:9b1523712da8b6b8ff191d93e2946a347a2dc8f9f0fd85f9e03fb1d9d96f92c7 +size 16036 diff --git a/checkpoint-9000/scheduler.bin b/checkpoint-9000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..2bb9b060c40343368e24e37c5f842ad05065982f --- /dev/null +++ b/checkpoint-9000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e02848355874392e2ebe0a08e996864dadbcedb2c14ac9b2c84334c77a73e38f +size 1000 diff --git a/checkpoint-9000/unet/config.json b/checkpoint-9000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..ccb8474073252815aa720c280c23b28669d7a727 --- /dev/null +++ b/checkpoint-9000/unet/config.json @@ -0,0 +1,68 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.33.1", + "_name_or_path": "output_lr1e4_accstep1/checkpoint-2500", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "addition_time_embed_dim": null, + "attention_head_dim": 8, + "attention_type": "default", + "block_out_channels": [ + 320, + 640, + 1280, + 1280 + ], + "center_input_sample": false, + "class_embed_type": null, + "class_embeddings_concat": false, + "conv_in_kernel": 3, + "conv_out_kernel": 3, + "cross_attention_dim": 768, + "cross_attention_norm": null, + "down_block_types": [ + "CrossAttnDownBlock2D", + "CrossAttnDownBlock2D", + "CrossAttnDownBlock2D", + "DownBlock2D" + ], + "downsample_padding": 1, + "dropout": 0.0, + "dual_cross_attention": false, + "encoder_hid_dim": null, + "encoder_hid_dim_type": null, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 8, + "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_attention_heads": null, + "num_class_embeds": null, + "only_cross_attention": false, + "out_channels": 4, + "projection_class_embeddings_input_dim": null, + "resnet_out_scale_factor": 1.0, + "resnet_skip_time_act": false, + "resnet_time_scale_shift": "default", + "reverse_transformer_layers_per_block": null, + "sample_size": 64, + "time_cond_proj_dim": null, + "time_embedding_act_fn": null, + "time_embedding_dim": null, + "time_embedding_type": "positional", + "timestep_post_act": null, + "transformer_layers_per_block": 1, + "up_block_types": [ + "UpBlock2D", + "CrossAttnUpBlock2D", + "CrossAttnUpBlock2D", + "CrossAttnUpBlock2D" + ], + "upcast_attention": false, + "use_linear_projection": false +} diff --git a/checkpoint-9000/unet/diffusion_pytorch_model.safetensors b/checkpoint-9000/unet/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..91efb6fcf9b99980aeecba45d246845f06b6acd8 --- /dev/null +++ b/checkpoint-9000/unet/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7ace86dbea48146baa845431af2dd59213d6273f219c3fdc5e224f3e440bc47b +size 3438213624 diff --git a/checkpoint-9500/optimizer.bin b/checkpoint-9500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..13f224fd6cf336e9978344324a87cb9f7eb23609 --- /dev/null +++ b/checkpoint-9500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:761e1df426af24df0ad77d873ea2e15c6808319d8acbb50bf53476e0d2cfd9f5 +size 6876851924 diff --git a/checkpoint-9500/random_states_0.pkl b/checkpoint-9500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..1dc41ee9ccfc27386bc6a1dbbfc2d80e066c6bbc --- /dev/null +++ b/checkpoint-9500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:d4481b5418cb09c3c8ba7ca0d06da12f61174527e60a318ff9a0702ec5abfe8c +size 16036 diff --git a/checkpoint-9500/scheduler.bin b/checkpoint-9500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..059600b42824453568909f3ad65f713882dccace --- /dev/null +++ b/checkpoint-9500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:aedbb1214c5360db1fde7920a74e80e35cb2bfe32c2adcbeb16da50f933981fd +size 1000 diff --git a/checkpoint-9500/unet/config.json b/checkpoint-9500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..ccb8474073252815aa720c280c23b28669d7a727 --- /dev/null +++ b/checkpoint-9500/unet/config.json @@ -0,0 +1,68 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.33.1", + "_name_or_path": "output_lr1e4_accstep1/checkpoint-2500", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "addition_time_embed_dim": null, + "attention_head_dim": 8, + "attention_type": "default", + "block_out_channels": [ + 320, + 640, + 1280, + 1280 + ], + "center_input_sample": false, + "class_embed_type": null, + "class_embeddings_concat": false, + "conv_in_kernel": 3, + "conv_out_kernel": 3, + "cross_attention_dim": 768, + "cross_attention_norm": null, + "down_block_types": [ + "CrossAttnDownBlock2D", + "CrossAttnDownBlock2D", + "CrossAttnDownBlock2D", + "DownBlock2D" + ], + "downsample_padding": 1, + "dropout": 0.0, + "dual_cross_attention": false, + "encoder_hid_dim": null, + "encoder_hid_dim_type": null, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 8, + "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_attention_heads": null, + "num_class_embeds": null, + "only_cross_attention": false, + "out_channels": 4, + "projection_class_embeddings_input_dim": null, + "resnet_out_scale_factor": 1.0, + "resnet_skip_time_act": false, + "resnet_time_scale_shift": "default", + "reverse_transformer_layers_per_block": null, + "sample_size": 64, + "time_cond_proj_dim": null, + "time_embedding_act_fn": null, + "time_embedding_dim": null, + "time_embedding_type": "positional", + "timestep_post_act": null, + "transformer_layers_per_block": 1, + "up_block_types": [ + "UpBlock2D", + "CrossAttnUpBlock2D", + "CrossAttnUpBlock2D", + "CrossAttnUpBlock2D" + ], + "upcast_attention": false, + "use_linear_projection": false +} diff --git a/checkpoint-9500/unet/diffusion_pytorch_model.safetensors b/checkpoint-9500/unet/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..3879923e93e31bf7387ed4c1f5c87eb70707dae1 --- /dev/null +++ b/checkpoint-9500/unet/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b3ebbd21f057392b1279038ccf3bd32992103087bbb534e72b82d66633430d6a +size 3438213624 diff --git a/readme.txt b/readme.txt new file mode 100644 index 0000000000000000000000000000000000000000..26b14fc9e17a7d7ff9639a7653ac090494c9240f --- /dev/null +++ b/readme.txt @@ -0,0 +1,5 @@ +目前,output_lr1e4_accstep1里面,直到2500步,学习率都是1e-4,梯度累计步长为1步,未使用角度信息。 +对应的参数为:Namespace(adam_beta1=0.9, adam_beta2=0.999, adam_epsilon=1e-08, adam_weight_decay=0.01, allow_tf32=False, cache_dir=None, center_crop=False, checkpointing_steps=500, checkpoints_total_limit=None, conditioning_dropout_prob=None, data_file_records='/home/ach17654gk/code/train_sdxl_gg/metadata.jsonl', dataloader_num_workers=0, dataset_config_name=None, dataset_name=None, edit_prompt_column='text', edited_image_column='conditioning_image', enable_xformers_memory_efficient_attention=False, gradient_accumulation_steps=1, gradient_checkpointing=False, hub_model_id=None, hub_token=None, learning_rate=0.0001, local_rank=1, logging_dir='logs', lr_scheduler='constant', lr_warmup_steps=500, max_grad_norm=1.0, max_train_samples=None, max_train_steps=2500, mixed_precision=None, non_ema_revision=None, num_train_epochs=100, num_validation_images=4, original_image_column='file_name', output_dir='output_lr1e4_accstep1', pretrained_model_name_or_path='/home/ach17654gk/.cache/huggingface/hub/models--runwayml--stable-diffusion-v1-5/snapshots/451f4fe16113bff5a5d2269ed5ad43b0592e9a14', push_to_hub=True, random_flip=False, report_to='wandb', resolution=512, resume_from_checkpoint=None, revision=None, scale_lr=False, seed=0, train_batch_size=16, train_data_dir='/home/ach17654gk/download/dataset/DFC/Track_1', use_8bit_adam=False, use_ema=False, val_image_url='/home/ach17654gk/download/dataset/DFC/Track_1/png/rgb_images/TrainArea_1551.png', validation_epochs=1, validation_prompt='A high-resolution synthetic aperture radar (SAR) satellite image, heavy speckle noise, monochrome, moderate contrast, detailed geometric patterns, realistic radar texture, remote sensing style, urban area, distinct linear and rectangular structures', variant=None) + + +接下来,在2500步的基础上,我将尝试学习率缩小3倍,即3e-5,训练到10000步,依旧不加入角度信息。 \ No newline at end of file diff --git a/unet/config.json b/unet/config.json index a0cfd35c2efd29641de60c2814431ed57d25ed46..ccb8474073252815aa720c280c23b28669d7a727 100644 --- a/unet/config.json +++ b/unet/config.json @@ -1,7 +1,7 @@ { "_class_name": "UNet2DConditionModel", "_diffusers_version": "0.33.1", - "_name_or_path": "/home/ach17654gk/.cache/huggingface/hub/models--runwayml--stable-diffusion-v1-5/snapshots/451f4fe16113bff5a5d2269ed5ad43b0592e9a14", + "_name_or_path": "output_lr1e4_accstep1/checkpoint-2500", "act_fn": "silu", "addition_embed_type": null, "addition_embed_type_num_heads": 64, diff --git a/unet/diffusion_pytorch_model.safetensors b/unet/diffusion_pytorch_model.safetensors index 3b141fc6f5e2eaabec81a8985c595a3774c0af6c..e7da2172a1864b41333cba08782833ffdb279e5a 100644 --- a/unet/diffusion_pytorch_model.safetensors +++ b/unet/diffusion_pytorch_model.safetensors @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:55c2cca9d4b62bc4385f06c3325e2d0b39416fc244f834b90bc04b7ac3f32196 +oid sha256:6a57a4c6cf76e35b0080ceebdf52fa343ffc4b161b86297f71ea9fd348bc8eed size 3438213624