diff --git a/checkpoint-110500/optimizer.bin b/checkpoint-110500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..02551f17b0f57c40ae37f29ba1ff3e7b59da489f --- /dev/null +++ b/checkpoint-110500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:4a6649ed291d269b15aabbc54a9d627da21ad25bc214ef5c7990977ff6ac8b94 +size 73251413 diff --git a/checkpoint-110500/random_states_0.pkl b/checkpoint-110500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..4ce2d1a9c756f087ea4b68c922a18ff9f82e1eff --- /dev/null +++ b/checkpoint-110500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:ef939377761817e7a909f925c8dfa6f17d4b27b434466845acf238154b4d8cef +size 14663 diff --git a/checkpoint-110500/scaler.pt b/checkpoint-110500/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..9680ae311afd0b4d4da162e91f57b1a57b7aaf56 --- /dev/null +++ b/checkpoint-110500/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:d2b1f049e7b1b84ca8f86f9bc103f41e6edefb714d4d8a80307ed41ead87d4a2 +size 557 diff --git a/checkpoint-110500/scheduler.bin b/checkpoint-110500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..9aa46fd561ef214a56b6b63486139b43f0c49e27 --- /dev/null +++ b/checkpoint-110500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:6aa13828e54ddc9a2a1b380db4cedb6e8f3aabe4b6a43424f23b93f1a7c4740a +size 563 diff --git a/checkpoint-110500/unet/config.json b/checkpoint-110500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..bb1cd664fc36d4dc821a4f0fdae4b94d7d99b356 --- /dev/null +++ b/checkpoint-110500/unet/config.json @@ -0,0 +1,60 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "attention_head_dim": 8, + "block_out_channels": [ + 64, + 64, + 128, + 128 + ], + "center_input_sample": false, + "class_embed_type": null, + "class_embeddings_concat": false, + "conv_in_kernel": 3, + "conv_out_kernel": 3, + "cross_attention_dim": 256, + "cross_attention_norm": null, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "CrossAttnDownBlock2D" + ], + "downsample_padding": 1, + "dual_cross_attention": false, + "encoder_hid_dim": null, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "layers_per_block": 2, + "mid_block_only_cross_attention": null, + "mid_block_scale_factor": 1, + "mid_block_type": "UNetMidBlock2DCrossAttn", + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "only_cross_attention": false, + "out_channels": 3, + "projection_class_embeddings_input_dim": null, + "resnet_out_scale_factor": 1.0, + "resnet_skip_time_act": false, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_cond_proj_dim": null, + "time_embedding_act_fn": null, + "time_embedding_dim": null, + "time_embedding_type": "positional", + "timestep_post_act": null, + "up_block_types": [ + "CrossAttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "use_linear_projection": false +} diff --git a/checkpoint-110500/unet/diffusion_pytorch_model.bin b/checkpoint-110500/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..74b326761ffa024e3d913d8fbfac1c178275e4d3 --- /dev/null +++ b/checkpoint-110500/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:76f071da8da9d57822b28c3a13e26586efca7005851e4aab1ca42c3138c55361 +size 36621969 diff --git a/checkpoint-110500/unet_ema/config.json b/checkpoint-110500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..c2d0240d83848bc77f0d03616208f5fa43a55623 --- /dev/null +++ b/checkpoint-110500/unet_ema/config.json @@ -0,0 +1,67 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "attention_head_dim": 8, + "block_out_channels": [ + 64, + 64, + 128, + 128 + ], + "center_input_sample": false, + "class_embed_type": null, + "class_embeddings_concat": false, + "conv_in_kernel": 3, + "conv_out_kernel": 3, + "cross_attention_dim": 256, + "cross_attention_norm": null, + "decay": 0.9999, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "CrossAttnDownBlock2D" + ], + "downsample_padding": 1, + "dual_cross_attention": false, + "encoder_hid_dim": null, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "inv_gamma": 1.0, + "layers_per_block": 2, + "mid_block_only_cross_attention": null, + "mid_block_scale_factor": 1, + "mid_block_type": "UNetMidBlock2DCrossAttn", + "min_decay": 0.0, + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "only_cross_attention": false, + "optimization_step": 110500, + "out_channels": 3, + "power": 0.75, + "projection_class_embeddings_input_dim": null, + "resnet_out_scale_factor": 1.0, + "resnet_skip_time_act": false, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_cond_proj_dim": null, + "time_embedding_act_fn": null, + "time_embedding_dim": null, + "time_embedding_type": "positional", + "timestep_post_act": null, + "up_block_types": [ + "CrossAttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "update_after_step": 0, + "use_ema_warmup": true, + "use_linear_projection": false +} diff --git a/checkpoint-110500/unet_ema/diffusion_pytorch_model.bin b/checkpoint-110500/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..6fc032486e36ed55eefe84bdd433050d0d0c6b53 --- /dev/null +++ b/checkpoint-110500/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c44e98516db8bf609ea61842d8504a599a254bf37642b62e58cf2398bce2f5ee +size 36616081 diff --git a/checkpoint-111000/optimizer.bin b/checkpoint-111000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..0d32ccd8601a5d6227845124458e90d2ed38960c --- /dev/null +++ b/checkpoint-111000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:f75d85ec8103060711748696aea8b702522fea823f67569f931053005498c894 +size 73251413 diff --git a/checkpoint-111000/random_states_0.pkl b/checkpoint-111000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..21e46fd5ad7e7b1b61d72d33a300e508b33e246c --- /dev/null +++ b/checkpoint-111000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:05f69bd5cdd686a82a4f2f21d832e4e8a528c7ca9cf6b0eac852c036244665be +size 14663 diff --git a/checkpoint-111000/scaler.pt b/checkpoint-111000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..9e48c1d99919013c4436b9e9c6cceb6c7775c045 --- /dev/null +++ b/checkpoint-111000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:d5ff8bf791b9314cb485359642c60fc7c5e30c00a9c2db7574d9aa93a2174b57 +size 557 diff --git a/checkpoint-111000/scheduler.bin b/checkpoint-111000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..1a695b49fa88f5a3e4ae233e32b9fbc2a97ba3b7 --- /dev/null +++ b/checkpoint-111000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e64a1caaca58d837ac9dadcf42d32ecf950f62d15c917421bede19b736cbf7f4 +size 563 diff --git a/checkpoint-111000/unet/config.json b/checkpoint-111000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..bb1cd664fc36d4dc821a4f0fdae4b94d7d99b356 --- /dev/null +++ b/checkpoint-111000/unet/config.json @@ -0,0 +1,60 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "attention_head_dim": 8, + "block_out_channels": [ + 64, + 64, + 128, + 128 + ], + "center_input_sample": false, + "class_embed_type": null, + "class_embeddings_concat": false, + "conv_in_kernel": 3, + "conv_out_kernel": 3, + "cross_attention_dim": 256, + "cross_attention_norm": null, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "CrossAttnDownBlock2D" + ], + "downsample_padding": 1, + "dual_cross_attention": false, + "encoder_hid_dim": null, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "layers_per_block": 2, + "mid_block_only_cross_attention": null, + "mid_block_scale_factor": 1, + "mid_block_type": "UNetMidBlock2DCrossAttn", + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "only_cross_attention": false, + "out_channels": 3, + "projection_class_embeddings_input_dim": null, + "resnet_out_scale_factor": 1.0, + "resnet_skip_time_act": false, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_cond_proj_dim": null, + "time_embedding_act_fn": null, + "time_embedding_dim": null, + "time_embedding_type": "positional", + "timestep_post_act": null, + "up_block_types": [ + "CrossAttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "use_linear_projection": false +} diff --git a/checkpoint-111000/unet/diffusion_pytorch_model.bin b/checkpoint-111000/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..e28358fac7fb6f0f6f0b4b422a091f80a19700ca --- /dev/null +++ b/checkpoint-111000/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:d44c56f4e8b67aeb7bcd05f0179e5aae8da95f9c516ec714c698991d445412b0 +size 36621969 diff --git a/checkpoint-111000/unet_ema/config.json b/checkpoint-111000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..e0177d9d45365ae75cfad549ae6e2deedebebf34 --- /dev/null +++ b/checkpoint-111000/unet_ema/config.json @@ -0,0 +1,67 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "attention_head_dim": 8, + "block_out_channels": [ + 64, + 64, + 128, + 128 + ], + "center_input_sample": false, + "class_embed_type": null, + "class_embeddings_concat": false, + "conv_in_kernel": 3, + "conv_out_kernel": 3, + "cross_attention_dim": 256, + "cross_attention_norm": null, + "decay": 0.9999, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "CrossAttnDownBlock2D" + ], + "downsample_padding": 1, + "dual_cross_attention": false, + "encoder_hid_dim": null, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "inv_gamma": 1.0, + "layers_per_block": 2, + "mid_block_only_cross_attention": null, + "mid_block_scale_factor": 1, + "mid_block_type": "UNetMidBlock2DCrossAttn", + "min_decay": 0.0, + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "only_cross_attention": false, + "optimization_step": 111000, + "out_channels": 3, + "power": 0.75, + "projection_class_embeddings_input_dim": null, + "resnet_out_scale_factor": 1.0, + "resnet_skip_time_act": false, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_cond_proj_dim": null, + "time_embedding_act_fn": null, + "time_embedding_dim": null, + "time_embedding_type": "positional", + "timestep_post_act": null, + "up_block_types": [ + "CrossAttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "update_after_step": 0, + "use_ema_warmup": true, + "use_linear_projection": false +} diff --git a/checkpoint-111000/unet_ema/diffusion_pytorch_model.bin b/checkpoint-111000/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..280351f5089edbecd21ee50c95117a4366af6419 --- /dev/null +++ b/checkpoint-111000/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7daaaf98ea0ed435583a3ab9baa9d78cb7eaa2f1c897a404eedf03482565b142 +size 36616081 diff --git a/checkpoint-111500/optimizer.bin b/checkpoint-111500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..b67fffbb795aaf0db0a3eaa1021ccc2b6cff84dc --- /dev/null +++ b/checkpoint-111500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:50d4c939eaf9582885d0970bf217b6224d30d9451dcff0e407a60f73a9b00972 +size 73251413 diff --git a/checkpoint-111500/random_states_0.pkl b/checkpoint-111500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..b333ce916407e71e87c090cbe692633bf30759b6 --- /dev/null +++ b/checkpoint-111500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b2d338dfd7747332ac13e4903ec422c05eb4b509457fef68358e2166fd672338 +size 14663 diff --git a/checkpoint-111500/scaler.pt b/checkpoint-111500/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..353d95865707f3a3b7f32700a9e4258c1339b5aa --- /dev/null +++ b/checkpoint-111500/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b8a4f9309bf0882863fb13466d88e4676de96132eeed730253fcc52ad60f1345 +size 557 diff --git a/checkpoint-111500/scheduler.bin b/checkpoint-111500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..eeb632436c5a1a29fca9a256f632bc9ec597e186 --- /dev/null +++ b/checkpoint-111500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:ec234658ebff831bf995f68e695f75bd9e0de5a31e7e2113632ddde9259703e6 +size 563 diff --git a/checkpoint-111500/unet/config.json b/checkpoint-111500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..bb1cd664fc36d4dc821a4f0fdae4b94d7d99b356 --- /dev/null +++ b/checkpoint-111500/unet/config.json @@ -0,0 +1,60 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "attention_head_dim": 8, + "block_out_channels": [ + 64, + 64, + 128, + 128 + ], + "center_input_sample": false, + "class_embed_type": null, + "class_embeddings_concat": false, + "conv_in_kernel": 3, + "conv_out_kernel": 3, + "cross_attention_dim": 256, + "cross_attention_norm": null, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "CrossAttnDownBlock2D" + ], + "downsample_padding": 1, + "dual_cross_attention": false, + "encoder_hid_dim": null, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "layers_per_block": 2, + "mid_block_only_cross_attention": null, + "mid_block_scale_factor": 1, + "mid_block_type": "UNetMidBlock2DCrossAttn", + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "only_cross_attention": false, + "out_channels": 3, + "projection_class_embeddings_input_dim": null, + "resnet_out_scale_factor": 1.0, + "resnet_skip_time_act": false, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_cond_proj_dim": null, + "time_embedding_act_fn": null, + "time_embedding_dim": null, + "time_embedding_type": "positional", + "timestep_post_act": null, + "up_block_types": [ + "CrossAttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "use_linear_projection": false +} diff --git a/checkpoint-111500/unet/diffusion_pytorch_model.bin b/checkpoint-111500/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..ae382163ab63a7231d3e7b3754a5bc47ddfe20fe --- /dev/null +++ b/checkpoint-111500/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:f701284194e7108ae4a84f05c19584d7f9b14ed6a88435462550abb947cf2e94 +size 36621969 diff --git a/checkpoint-111500/unet_ema/config.json b/checkpoint-111500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..73c48ee2c3e614923081d032fb389b7db8c177bb --- /dev/null +++ b/checkpoint-111500/unet_ema/config.json @@ -0,0 +1,67 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "attention_head_dim": 8, + "block_out_channels": [ + 64, + 64, + 128, + 128 + ], + "center_input_sample": false, + "class_embed_type": null, + "class_embeddings_concat": false, + "conv_in_kernel": 3, + "conv_out_kernel": 3, + "cross_attention_dim": 256, + "cross_attention_norm": null, + "decay": 0.9999, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "CrossAttnDownBlock2D" + ], + "downsample_padding": 1, + "dual_cross_attention": false, + "encoder_hid_dim": null, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "inv_gamma": 1.0, + "layers_per_block": 2, + "mid_block_only_cross_attention": null, + "mid_block_scale_factor": 1, + "mid_block_type": "UNetMidBlock2DCrossAttn", + "min_decay": 0.0, + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "only_cross_attention": false, + "optimization_step": 111500, + "out_channels": 3, + "power": 0.75, + "projection_class_embeddings_input_dim": null, + "resnet_out_scale_factor": 1.0, + "resnet_skip_time_act": false, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_cond_proj_dim": null, + "time_embedding_act_fn": null, + "time_embedding_dim": null, + "time_embedding_type": "positional", + "timestep_post_act": null, + "up_block_types": [ + "CrossAttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "update_after_step": 0, + "use_ema_warmup": true, + "use_linear_projection": false +} diff --git a/checkpoint-111500/unet_ema/diffusion_pytorch_model.bin b/checkpoint-111500/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..187228d90d198b7a0b5da1274b206f51339fa810 --- /dev/null +++ b/checkpoint-111500/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:6a9d679797fe93596f0d61cddec9e2f90e7d18a6c289d6d4756ebd5a98a4bef8 +size 36616081 diff --git a/checkpoint-112000/optimizer.bin b/checkpoint-112000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..299f7c7fe94ac56f4ccc09ff3f4114206d0c1d8d --- /dev/null +++ b/checkpoint-112000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:d1ef3fc3ba918eb17afd7179f2bdf0eaef2155ff2112d10d290b84bff516a8ba +size 73251413 diff --git a/checkpoint-112000/random_states_0.pkl b/checkpoint-112000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..120f7527db274492d8151871791b6a90c4ea2d8c --- /dev/null +++ b/checkpoint-112000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7e5b24870cf612a173a5e347d2947d17db73ecb9f9d502898be4bff0c59704ff +size 14663 diff --git a/checkpoint-112000/scaler.pt b/checkpoint-112000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..c649642ddc626879b427fb5cb1fba3513feef1cb --- /dev/null +++ b/checkpoint-112000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c43865450a2bbd02d451abe88dc222df0fd36c2792f7e09c6ecd768038fb9634 +size 557 diff --git a/checkpoint-112000/scheduler.bin b/checkpoint-112000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..5ed5db846ef70a539672df00b60ebc39c87ca518 --- /dev/null +++ b/checkpoint-112000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:1c43c30d81e508f1a3ff769b2fb7bb58f25456eae2c9f1b03491b80f6101c204 +size 563 diff --git a/checkpoint-112000/unet/config.json b/checkpoint-112000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..bb1cd664fc36d4dc821a4f0fdae4b94d7d99b356 --- /dev/null +++ b/checkpoint-112000/unet/config.json @@ -0,0 +1,60 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "attention_head_dim": 8, + "block_out_channels": [ + 64, + 64, + 128, + 128 + ], + "center_input_sample": false, + "class_embed_type": null, + "class_embeddings_concat": false, + "conv_in_kernel": 3, + "conv_out_kernel": 3, + "cross_attention_dim": 256, + "cross_attention_norm": null, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "CrossAttnDownBlock2D" + ], + "downsample_padding": 1, + "dual_cross_attention": false, + "encoder_hid_dim": null, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "layers_per_block": 2, + "mid_block_only_cross_attention": null, + "mid_block_scale_factor": 1, + "mid_block_type": "UNetMidBlock2DCrossAttn", + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "only_cross_attention": false, + "out_channels": 3, + "projection_class_embeddings_input_dim": null, + "resnet_out_scale_factor": 1.0, + "resnet_skip_time_act": false, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_cond_proj_dim": null, + "time_embedding_act_fn": null, + "time_embedding_dim": null, + "time_embedding_type": "positional", + "timestep_post_act": null, + "up_block_types": [ + "CrossAttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "use_linear_projection": false +} diff --git a/checkpoint-112000/unet/diffusion_pytorch_model.bin b/checkpoint-112000/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..559dfb5b4d34ac0a96882814cb52a520da8739c2 --- /dev/null +++ b/checkpoint-112000/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:9bc7dea0b0518dda64bb0b8f9c2a36e2c8aba36cba7a0cecc36e8ae1d8cc36a0 +size 36621969 diff --git a/checkpoint-112000/unet_ema/config.json b/checkpoint-112000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..b63aa7fa07bc9109a836c73eaaed2d48f5d814a0 --- /dev/null +++ b/checkpoint-112000/unet_ema/config.json @@ -0,0 +1,67 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "attention_head_dim": 8, + "block_out_channels": [ + 64, + 64, + 128, + 128 + ], + "center_input_sample": false, + "class_embed_type": null, + "class_embeddings_concat": false, + "conv_in_kernel": 3, + "conv_out_kernel": 3, + "cross_attention_dim": 256, + "cross_attention_norm": null, + "decay": 0.9999, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "CrossAttnDownBlock2D" + ], + "downsample_padding": 1, + "dual_cross_attention": false, + "encoder_hid_dim": null, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "inv_gamma": 1.0, + "layers_per_block": 2, + "mid_block_only_cross_attention": null, + "mid_block_scale_factor": 1, + "mid_block_type": "UNetMidBlock2DCrossAttn", + "min_decay": 0.0, + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "only_cross_attention": false, + "optimization_step": 112000, + "out_channels": 3, + "power": 0.75, + "projection_class_embeddings_input_dim": null, + "resnet_out_scale_factor": 1.0, + "resnet_skip_time_act": false, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_cond_proj_dim": null, + "time_embedding_act_fn": null, + "time_embedding_dim": null, + "time_embedding_type": "positional", + "timestep_post_act": null, + "up_block_types": [ + "CrossAttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "update_after_step": 0, + "use_ema_warmup": true, + "use_linear_projection": false +} diff --git a/checkpoint-112000/unet_ema/diffusion_pytorch_model.bin b/checkpoint-112000/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..a1912a7ecc23d49956957641c40d9f6cce017195 --- /dev/null +++ b/checkpoint-112000/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7d23bf0856bfe6019425c444ce4204f0edadd4eeaff226c2c901b33ab53c1fe2 +size 36616081 diff --git a/checkpoint-112500/optimizer.bin b/checkpoint-112500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..43bcd3a46f59b5debf4fd4160a6e72908bb019d8 --- /dev/null +++ b/checkpoint-112500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:dd4032e6278839405cc125674a4c8e74004a3acba6dba66650c095ebe1682e0b +size 73251413 diff --git a/checkpoint-112500/random_states_0.pkl b/checkpoint-112500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..24554ad7b444c91a24fef4fc5a2cd2af5852585d --- /dev/null +++ b/checkpoint-112500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:25be898e7b6362dc3bd490910ca9d95f6da2c3495c3a59ea3925ceaa2860a5e3 +size 14663 diff --git a/checkpoint-112500/scaler.pt b/checkpoint-112500/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..7b059743f07025acbadcc51875d9f70b2612bc21 --- /dev/null +++ b/checkpoint-112500/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:4df8675f9c506c7132b875d98f2043ebfe8c567255ee10fbe56eb145da8b60a5 +size 557 diff --git a/checkpoint-112500/scheduler.bin b/checkpoint-112500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..059dc232327a6e4de10da98b620906926161378e --- /dev/null +++ b/checkpoint-112500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:53e018137d97cca32e2218733d091612b9c9f5a29b38328d5119faeadaa9d83e +size 563 diff --git a/checkpoint-112500/unet/config.json b/checkpoint-112500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..bb1cd664fc36d4dc821a4f0fdae4b94d7d99b356 --- /dev/null +++ b/checkpoint-112500/unet/config.json @@ -0,0 +1,60 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "attention_head_dim": 8, + "block_out_channels": [ + 64, + 64, + 128, + 128 + ], + "center_input_sample": false, + "class_embed_type": null, + "class_embeddings_concat": false, + "conv_in_kernel": 3, + "conv_out_kernel": 3, + "cross_attention_dim": 256, + "cross_attention_norm": null, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "CrossAttnDownBlock2D" + ], + "downsample_padding": 1, + "dual_cross_attention": false, + "encoder_hid_dim": null, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "layers_per_block": 2, + "mid_block_only_cross_attention": null, + "mid_block_scale_factor": 1, + "mid_block_type": "UNetMidBlock2DCrossAttn", + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "only_cross_attention": false, + "out_channels": 3, + "projection_class_embeddings_input_dim": null, + "resnet_out_scale_factor": 1.0, + "resnet_skip_time_act": false, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_cond_proj_dim": null, + "time_embedding_act_fn": null, + "time_embedding_dim": null, + "time_embedding_type": "positional", + "timestep_post_act": null, + "up_block_types": [ + "CrossAttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "use_linear_projection": false +} diff --git a/checkpoint-112500/unet/diffusion_pytorch_model.bin b/checkpoint-112500/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..f3c1946ec9dfc447c0ac98fd080de344bc4385a9 --- /dev/null +++ b/checkpoint-112500/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:a76d65e5ed11c1f886c1fb2b7cc724093af6fa8f01095f24f1ed3e59e34ee6be +size 36621969 diff --git a/checkpoint-112500/unet_ema/config.json b/checkpoint-112500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..5ec068232d9eea1805fbad60a75679aed4ec96d3 --- /dev/null +++ b/checkpoint-112500/unet_ema/config.json @@ -0,0 +1,67 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "attention_head_dim": 8, + "block_out_channels": [ + 64, + 64, + 128, + 128 + ], + "center_input_sample": false, + "class_embed_type": null, + "class_embeddings_concat": false, + "conv_in_kernel": 3, + "conv_out_kernel": 3, + "cross_attention_dim": 256, + "cross_attention_norm": null, + "decay": 0.9999, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "CrossAttnDownBlock2D" + ], + "downsample_padding": 1, + "dual_cross_attention": false, + "encoder_hid_dim": null, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "inv_gamma": 1.0, + "layers_per_block": 2, + "mid_block_only_cross_attention": null, + "mid_block_scale_factor": 1, + "mid_block_type": "UNetMidBlock2DCrossAttn", + "min_decay": 0.0, + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "only_cross_attention": false, + "optimization_step": 112500, + "out_channels": 3, + "power": 0.75, + "projection_class_embeddings_input_dim": null, + "resnet_out_scale_factor": 1.0, + "resnet_skip_time_act": false, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_cond_proj_dim": null, + "time_embedding_act_fn": null, + "time_embedding_dim": null, + "time_embedding_type": "positional", + "timestep_post_act": null, + "up_block_types": [ + "CrossAttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "update_after_step": 0, + "use_ema_warmup": true, + "use_linear_projection": false +} diff --git a/checkpoint-112500/unet_ema/diffusion_pytorch_model.bin b/checkpoint-112500/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..9aca24924a4d841e92366c8b14b56146cb430b37 --- /dev/null +++ b/checkpoint-112500/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:8b2dde0220d6e957cad7caabf995bd43b0fba9fec45da03dcc84d7ec018c5545 +size 36616081 diff --git a/checkpoint-113000/optimizer.bin b/checkpoint-113000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..3621fc4983ee1e048d85195ed612832cef8012ee --- /dev/null +++ b/checkpoint-113000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:9a9e697e0aae39e2dc1579c00a5593c6c2865b8ac690e3a5f44079e4020af19f +size 73251413 diff --git a/checkpoint-113000/random_states_0.pkl b/checkpoint-113000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..cc0c5ee832f8f47d7f971f630aa5acd6ed432554 --- /dev/null +++ b/checkpoint-113000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:2e156fd25ee0417012bf544feb4bbcc8235bb7a937cde2af020db7c7ed866bd2 +size 14663 diff --git a/checkpoint-113000/scaler.pt b/checkpoint-113000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..c5131a42d7b36d1e75c8f8060dee0d7f9e69e589 --- /dev/null +++ b/checkpoint-113000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:8747714e82a7f2da46fca62c08326b64c562752fc5c35d5ae54ec135e70a9b0d +size 557 diff --git a/checkpoint-113000/scheduler.bin b/checkpoint-113000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..5a0266a209fb2a2cecd080ebe315297c16afd8b8 --- /dev/null +++ b/checkpoint-113000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:4903dbbcb9c87b421d84c8f55e335cea0b1d8afcefce30b38a8791afe60b649e +size 563 diff --git a/checkpoint-113000/unet/config.json b/checkpoint-113000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..bb1cd664fc36d4dc821a4f0fdae4b94d7d99b356 --- /dev/null +++ b/checkpoint-113000/unet/config.json @@ -0,0 +1,60 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "attention_head_dim": 8, + "block_out_channels": [ + 64, + 64, + 128, + 128 + ], + "center_input_sample": false, + "class_embed_type": null, + "class_embeddings_concat": false, + "conv_in_kernel": 3, + "conv_out_kernel": 3, + "cross_attention_dim": 256, + "cross_attention_norm": null, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "CrossAttnDownBlock2D" + ], + "downsample_padding": 1, + "dual_cross_attention": false, + "encoder_hid_dim": null, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "layers_per_block": 2, + "mid_block_only_cross_attention": null, + "mid_block_scale_factor": 1, + "mid_block_type": "UNetMidBlock2DCrossAttn", + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "only_cross_attention": false, + "out_channels": 3, + "projection_class_embeddings_input_dim": null, + "resnet_out_scale_factor": 1.0, + "resnet_skip_time_act": false, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_cond_proj_dim": null, + "time_embedding_act_fn": null, + "time_embedding_dim": null, + "time_embedding_type": "positional", + "timestep_post_act": null, + "up_block_types": [ + "CrossAttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "use_linear_projection": false +} diff --git a/checkpoint-113000/unet/diffusion_pytorch_model.bin b/checkpoint-113000/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..8ecead60630fd0f68dc95e9d5a2f00c2c81ec0f0 --- /dev/null +++ b/checkpoint-113000/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:a6c45ee6491bcf2c9a578f069bc6f0ab7567c5b2fc5241bc2d9eec8dcb65d045 +size 36621969 diff --git a/checkpoint-113000/unet_ema/config.json b/checkpoint-113000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..6eb9e37e9564745bc27265a6145e41cbfcf1d0b9 --- /dev/null +++ b/checkpoint-113000/unet_ema/config.json @@ -0,0 +1,67 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "attention_head_dim": 8, + "block_out_channels": [ + 64, + 64, + 128, + 128 + ], + "center_input_sample": false, + "class_embed_type": null, + "class_embeddings_concat": false, + "conv_in_kernel": 3, + "conv_out_kernel": 3, + "cross_attention_dim": 256, + "cross_attention_norm": null, + "decay": 0.9999, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "CrossAttnDownBlock2D" + ], + "downsample_padding": 1, + "dual_cross_attention": false, + "encoder_hid_dim": null, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "inv_gamma": 1.0, + "layers_per_block": 2, + "mid_block_only_cross_attention": null, + "mid_block_scale_factor": 1, + "mid_block_type": "UNetMidBlock2DCrossAttn", + "min_decay": 0.0, + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "only_cross_attention": false, + "optimization_step": 113000, + "out_channels": 3, + "power": 0.75, + "projection_class_embeddings_input_dim": null, + "resnet_out_scale_factor": 1.0, + "resnet_skip_time_act": false, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_cond_proj_dim": null, + "time_embedding_act_fn": null, + "time_embedding_dim": null, + "time_embedding_type": "positional", + "timestep_post_act": null, + "up_block_types": [ + "CrossAttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "update_after_step": 0, + "use_ema_warmup": true, + "use_linear_projection": false +} diff --git a/checkpoint-113000/unet_ema/diffusion_pytorch_model.bin b/checkpoint-113000/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..a5ba1839752aa820f9575d29b364214c2cb95993 --- /dev/null +++ b/checkpoint-113000/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:0f205cc25db693e5d7c7bc17daef0c77023ff1a07f84fcafbebf51c040690e79 +size 36616081 diff --git a/checkpoint-113500/optimizer.bin b/checkpoint-113500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..b25045c36e30af5c3cf3487eeff544331c933666 --- /dev/null +++ b/checkpoint-113500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:f350e61f1dd06cced50f39aee2791f62701c2a438a30e0ab39835a75b2ebf751 +size 73251413 diff --git a/checkpoint-113500/random_states_0.pkl b/checkpoint-113500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..bd22806321e4dced8e1de0bac3ab07210786c96d --- /dev/null +++ b/checkpoint-113500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c79e43d4652f76723666a8f85ac8f92ca9d9d01eccdd6ea45d4894d3e31c8381 +size 14663 diff --git a/checkpoint-113500/scaler.pt b/checkpoint-113500/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..7b6b0dd7bc1172186356a65c4b17a56fd7e7bd49 --- /dev/null +++ b/checkpoint-113500/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:90fe833d1dc6f63c37613775b16a564aeb78f8cf103770e81f0fbb4839db398a +size 557 diff --git a/checkpoint-113500/scheduler.bin b/checkpoint-113500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..1f4618a69f6ce5f12cf1d7eacea3943994968fc0 --- /dev/null +++ b/checkpoint-113500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:5a90486265695d3763c89a82c4ef6d6b40012525e28dddd6a0c1ad05996481b0 +size 563 diff --git a/checkpoint-113500/unet/config.json b/checkpoint-113500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..bb1cd664fc36d4dc821a4f0fdae4b94d7d99b356 --- /dev/null +++ b/checkpoint-113500/unet/config.json @@ -0,0 +1,60 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "attention_head_dim": 8, + "block_out_channels": [ + 64, + 64, + 128, + 128 + ], + "center_input_sample": false, + "class_embed_type": null, + "class_embeddings_concat": false, + "conv_in_kernel": 3, + "conv_out_kernel": 3, + "cross_attention_dim": 256, + "cross_attention_norm": null, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "CrossAttnDownBlock2D" + ], + "downsample_padding": 1, + "dual_cross_attention": false, + "encoder_hid_dim": null, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "layers_per_block": 2, + "mid_block_only_cross_attention": null, + "mid_block_scale_factor": 1, + "mid_block_type": "UNetMidBlock2DCrossAttn", + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "only_cross_attention": false, + "out_channels": 3, + "projection_class_embeddings_input_dim": null, + "resnet_out_scale_factor": 1.0, + "resnet_skip_time_act": false, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_cond_proj_dim": null, + "time_embedding_act_fn": null, + "time_embedding_dim": null, + "time_embedding_type": "positional", + "timestep_post_act": null, + "up_block_types": [ + "CrossAttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "use_linear_projection": false +} diff --git a/checkpoint-113500/unet/diffusion_pytorch_model.bin b/checkpoint-113500/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..a6e7df50d861c194c10aec46ccee92517a2cc490 --- /dev/null +++ b/checkpoint-113500/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:2a6fbb6cefa4253c7481b160ed168affe39624d84f6c640f340f4520f0c9b923 +size 36621969 diff --git a/checkpoint-113500/unet_ema/config.json b/checkpoint-113500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..242a73016220fe0bc8f9e87d717e6c8782babd2b --- /dev/null +++ b/checkpoint-113500/unet_ema/config.json @@ -0,0 +1,67 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "attention_head_dim": 8, + "block_out_channels": [ + 64, + 64, + 128, + 128 + ], + "center_input_sample": false, + "class_embed_type": null, + "class_embeddings_concat": false, + "conv_in_kernel": 3, + "conv_out_kernel": 3, + "cross_attention_dim": 256, + "cross_attention_norm": null, + "decay": 0.9999, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "CrossAttnDownBlock2D" + ], + "downsample_padding": 1, + "dual_cross_attention": false, + "encoder_hid_dim": null, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "inv_gamma": 1.0, + "layers_per_block": 2, + "mid_block_only_cross_attention": null, + "mid_block_scale_factor": 1, + "mid_block_type": "UNetMidBlock2DCrossAttn", + "min_decay": 0.0, + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "only_cross_attention": false, + "optimization_step": 113500, + "out_channels": 3, + "power": 0.75, + "projection_class_embeddings_input_dim": null, + "resnet_out_scale_factor": 1.0, + "resnet_skip_time_act": false, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_cond_proj_dim": null, + "time_embedding_act_fn": null, + "time_embedding_dim": null, + "time_embedding_type": "positional", + "timestep_post_act": null, + "up_block_types": [ + "CrossAttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "update_after_step": 0, + "use_ema_warmup": true, + "use_linear_projection": false +} diff --git a/checkpoint-113500/unet_ema/diffusion_pytorch_model.bin b/checkpoint-113500/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..b307b218677a8a9e5b04ded3b934ba5271aa1ce2 --- /dev/null +++ b/checkpoint-113500/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e6c7274f0ce52289536a39df8d4241ef1b3cefdc3766d902d41eb99f48ef5926 +size 36616081 diff --git a/unet/diffusion_pytorch_model.bin b/unet/diffusion_pytorch_model.bin index bf5cb83b0fdfa22f710e51586d363e4edb728a60..d24e6a17b5a3736604d0772c9fbafaae5b037568 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:4d9eb8e1f582b69f1f0e0d12ced0b21327190f8ff0c87b93b6280235618af99f +oid sha256:81e83a20d8724f28b61af7497a83fb3120748fbf48c76f6013715e3ee6cd5b42 size 36621969