diff --git a/checkpoint-96500/optimizer.bin b/checkpoint-96500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..cbd8c7be110df9a42c550329103c115f119a6d9a --- /dev/null +++ b/checkpoint-96500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:eeb7472fe1bbcaebb4d0fa60f197ed919f28971797f77a456a8d03c4c410993a +size 73251413 diff --git a/checkpoint-96500/random_states_0.pkl b/checkpoint-96500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..fd3b04e1eac93d034995ff5f413bbdccc7547279 --- /dev/null +++ b/checkpoint-96500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:caf1c2dbd7ddf878f45e4aebc0fddc1444097c09afb1192f59b33e4499cb96d6 +size 14663 diff --git a/checkpoint-96500/scaler.pt b/checkpoint-96500/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..9c99b6362636272fd469e20bbe1fcccf686f8718 --- /dev/null +++ b/checkpoint-96500/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:9d9756d13e8e34dd6adb1439cd2ea674283a46767bf060f96463edc13fb94cfb +size 557 diff --git a/checkpoint-96500/scheduler.bin b/checkpoint-96500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..a8023d6741def9b0dab0b84df6919863b60f1c4b --- /dev/null +++ b/checkpoint-96500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e10c2104cc6edd7e6b2c76e71b0aaa59c5007be539002845c0bb6fdca42b4c14 +size 563 diff --git a/checkpoint-96500/unet/config.json b/checkpoint-96500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..bb1cd664fc36d4dc821a4f0fdae4b94d7d99b356 --- /dev/null +++ b/checkpoint-96500/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-96500/unet/diffusion_pytorch_model.bin b/checkpoint-96500/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..a555ade089927a0387d01755c092376d209bff54 --- /dev/null +++ b/checkpoint-96500/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:54db5c4659801945edebf25e942c8d74e3090d57c522b627794e0de63dce4843 +size 36621969 diff --git a/checkpoint-96500/unet_ema/config.json b/checkpoint-96500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..e90a2d1aae3fdeab30722cfd753ff45c2f6f6b46 --- /dev/null +++ b/checkpoint-96500/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": 96500, + "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-96500/unet_ema/diffusion_pytorch_model.bin b/checkpoint-96500/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..c01c7b3596053ae58d8f01fc38e3a90a1a8e99ba --- /dev/null +++ b/checkpoint-96500/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:45975121c4e1524f3d792c28bd5cafd547c9312560f0a389bce640dd83a29a53 +size 36616081 diff --git a/checkpoint-97000/optimizer.bin b/checkpoint-97000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..4f14f41a9310e1a41ab057cef464204295a5d15c --- /dev/null +++ b/checkpoint-97000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:dd20ecfec3911d7913d2f4b76b11bfabf5668c86905b7fd6dd0658defc16167e +size 73251413 diff --git a/checkpoint-97000/random_states_0.pkl b/checkpoint-97000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..3c2e92e94803446badd9a5345995a8cf35515648 --- /dev/null +++ b/checkpoint-97000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:aa36efc3722662c3da7e37241d64db0727aa27244b545c9d501c82dd9e3c2f2b +size 14663 diff --git a/checkpoint-97000/scaler.pt b/checkpoint-97000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..12acadb6fdf877fa0b72f85421fe63a1c0d39f91 --- /dev/null +++ b/checkpoint-97000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:ae0209e3afa094f53863400aaad2c6c19704855c0500b56695968f15eee60edb +size 557 diff --git a/checkpoint-97000/scheduler.bin b/checkpoint-97000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..a98c09751cd3f916e4b98c249369669229fee537 --- /dev/null +++ b/checkpoint-97000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:85973dacab208ce21b3410718acb20d2b9e571fa7294e1649accc590b5449b92 +size 563 diff --git a/checkpoint-97000/unet/config.json b/checkpoint-97000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..bb1cd664fc36d4dc821a4f0fdae4b94d7d99b356 --- /dev/null +++ b/checkpoint-97000/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-97000/unet/diffusion_pytorch_model.bin b/checkpoint-97000/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..80c0111be71034d86931b82e4fef28ac92c9ca75 --- /dev/null +++ b/checkpoint-97000/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:5bfc0c3674b2c87bbd218665015a8d18e54857e5e86e3b0f66d0f0b9768fe1ff +size 36621969 diff --git a/checkpoint-97000/unet_ema/config.json b/checkpoint-97000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..4320ea8f7fe88386a0b13f74ede1da7c7960cfe1 --- /dev/null +++ b/checkpoint-97000/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": 97000, + "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-97000/unet_ema/diffusion_pytorch_model.bin b/checkpoint-97000/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..9e87f8d8f81ab89428314ae28249db1517916086 --- /dev/null +++ b/checkpoint-97000/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:494b77808f1f0ec69aeddf9a79d975eaea85801f4a190c23007d886d11700146 +size 36616081 diff --git a/checkpoint-97500/optimizer.bin b/checkpoint-97500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..92af16477211739ea08468fb02db251a70eb6a40 --- /dev/null +++ b/checkpoint-97500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:db50221c44894ab9c8182c567dd8964a7a8cded516fc12d297aa528e29fb4b91 +size 73251413 diff --git a/checkpoint-97500/random_states_0.pkl b/checkpoint-97500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..3d679c57566dfdd2d0910f0bb504923187013c83 --- /dev/null +++ b/checkpoint-97500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:fd7cd06c66d8870d68808906a41a73b0df9a0c1c00ddc9e0550c74556bdd7a3f +size 14663 diff --git a/checkpoint-97500/scaler.pt b/checkpoint-97500/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..642fdd883fb1c6a748f7d8daa4287298f8be5b5a --- /dev/null +++ b/checkpoint-97500/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:69b45f863cb57c699573bb1f24bd161ce84503115499ba768ced3d5a49c9b752 +size 557 diff --git a/checkpoint-97500/scheduler.bin b/checkpoint-97500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..36cb6422e3c09c4c6b0d06e31800ac1fbc6c60a8 --- /dev/null +++ b/checkpoint-97500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:a2c34db67bf6c1b20876279507cb9ea8dd29a8b76aa65055270d44005316588e +size 563 diff --git a/checkpoint-97500/unet/config.json b/checkpoint-97500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..bb1cd664fc36d4dc821a4f0fdae4b94d7d99b356 --- /dev/null +++ b/checkpoint-97500/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-97500/unet/diffusion_pytorch_model.bin b/checkpoint-97500/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..1c4750c94255fc1d7c6766294c9d506e6ba6d42a --- /dev/null +++ b/checkpoint-97500/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:89a88cf44843ced513e4b0d4ae42d1fbd562de75238193ffeac15a70c91cc7f3 +size 36621969 diff --git a/checkpoint-97500/unet_ema/config.json b/checkpoint-97500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..11c59b220d38cfaeaf9fb66a7aad1e53a0ec71d7 --- /dev/null +++ b/checkpoint-97500/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": 97500, + "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-97500/unet_ema/diffusion_pytorch_model.bin b/checkpoint-97500/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..49fa927b096ef7f2527e8bf53119f55c08bdd16a --- /dev/null +++ b/checkpoint-97500/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:1e75b8018e48df841d5a33651f8c53b7a9b3444d7a0896cf74c6fb6fd71c4d49 +size 36616081 diff --git a/checkpoint-98000/optimizer.bin b/checkpoint-98000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..bfc42ef98c4b81a870529e20483bb7c63d0c7faa --- /dev/null +++ b/checkpoint-98000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:85b2adbd81c6228fa4026af8460db22d3aa230ceae71d11decf1c72cb27fe834 +size 73251413 diff --git a/checkpoint-98000/random_states_0.pkl b/checkpoint-98000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..85666a3ee5b1bcef891d7da01eb299180826e883 --- /dev/null +++ b/checkpoint-98000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:1114efe694154d32f841e0c45c7d483b1478b347dff0d1a4f5592d76c4beeb62 +size 14663 diff --git a/checkpoint-98000/scaler.pt b/checkpoint-98000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..ea4f75be77a8e15bba214f75537ac29dc052b71a --- /dev/null +++ b/checkpoint-98000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:0107d303ad22166162d69c282f368aa90168e2d8062d685e6a3fd7426463baef +size 557 diff --git a/checkpoint-98000/scheduler.bin b/checkpoint-98000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..845ba2bc711df567530448feed40ec5ea7b77927 --- /dev/null +++ b/checkpoint-98000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:fe9441ef61b5fab947558b6a0c880e248297409703f9a1309237679477ead090 +size 563 diff --git a/checkpoint-98000/unet/config.json b/checkpoint-98000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..bb1cd664fc36d4dc821a4f0fdae4b94d7d99b356 --- /dev/null +++ b/checkpoint-98000/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-98000/unet/diffusion_pytorch_model.bin b/checkpoint-98000/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..bd8fddc76a1eb23ded7bbde64217402b6b837cb6 --- /dev/null +++ b/checkpoint-98000/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:78c98f74a5075f52764804a75a0f61f244f38ade30e0119ff4793d76d9f8c664 +size 36621969 diff --git a/checkpoint-98000/unet_ema/config.json b/checkpoint-98000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..06edb61ebd82bd8a05be06ca3b94cead1232e366 --- /dev/null +++ b/checkpoint-98000/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": 98000, + "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-98000/unet_ema/diffusion_pytorch_model.bin b/checkpoint-98000/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..f53a29bcef4e4fd53733882aeb5681bdaebe09e3 --- /dev/null +++ b/checkpoint-98000/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:a3d64f076844cb51a8bc0741e89f254dcff72ca88dd70bf6f4b56299bbcf141a +size 36616081 diff --git a/checkpoint-98500/optimizer.bin b/checkpoint-98500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..e711c464af8966ca70ab5fedb0e5860fcce4cfc0 --- /dev/null +++ b/checkpoint-98500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:a6b7bde2a0f6627a6fe698749a08b028f2c12a6bca8020989bb2b09e5e84c495 +size 73251413 diff --git a/checkpoint-98500/random_states_0.pkl b/checkpoint-98500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..1d76ef8ec7eb947946e106b4e46d149a6c01ca29 --- /dev/null +++ b/checkpoint-98500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:753517a9d0f61e5ccd43d99bd935d4e197e10cd9d5009a961bdc1aee25e96d50 +size 14663 diff --git a/checkpoint-98500/scaler.pt b/checkpoint-98500/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..3c571fe2a946a6d42f7cdb0102de2ced22207bc8 --- /dev/null +++ b/checkpoint-98500/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c3da6be6da37c65735e33862c6a101d3afd9474152b356f832a9b23d09aaaa1f +size 557 diff --git a/checkpoint-98500/scheduler.bin b/checkpoint-98500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..33b5c44a719bd12598ec6262ff27233d48b1ab91 --- /dev/null +++ b/checkpoint-98500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:884a61008f1c29f0de246842b0e32b54046a885e350d4f37e37bceff34577f31 +size 563 diff --git a/checkpoint-98500/unet/config.json b/checkpoint-98500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..bb1cd664fc36d4dc821a4f0fdae4b94d7d99b356 --- /dev/null +++ b/checkpoint-98500/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-98500/unet/diffusion_pytorch_model.bin b/checkpoint-98500/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..4e02f90a7e57a51b8de94f1c143a24a3aca266b1 --- /dev/null +++ b/checkpoint-98500/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:76cd18adcb88cbd3ee136b10f0657a5df84893ab17218a0759b6c0e15a6f80a1 +size 36621969 diff --git a/checkpoint-98500/unet_ema/config.json b/checkpoint-98500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..83ee0d082ad5af93fe78a40d428a0570c51a3086 --- /dev/null +++ b/checkpoint-98500/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": 98500, + "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-98500/unet_ema/diffusion_pytorch_model.bin b/checkpoint-98500/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..960fddb39ffc1d672b93eb11651fa8963537c108 --- /dev/null +++ b/checkpoint-98500/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:db6d77683fab0c95e6a4ab7fc3f794cff1d010832bae80aa20faf62da71ec289 +size 36616081 diff --git a/checkpoint-99000/optimizer.bin b/checkpoint-99000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..b46a2817f2317529f7d2e954240517a76c4eebc7 --- /dev/null +++ b/checkpoint-99000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:ea00d5bc8c4037bd0e352d086b2d0deef4811f85378792f194a5e990a5c2584d +size 73251413 diff --git a/checkpoint-99000/random_states_0.pkl b/checkpoint-99000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..2acc669e734bbb026b30d215fd958fe9bad9c918 --- /dev/null +++ b/checkpoint-99000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:d7346840bcee9767c7caecbe950e4220c962a0ae2b72f565750a17f75b8887c9 +size 14663 diff --git a/checkpoint-99000/scaler.pt b/checkpoint-99000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..d4c85217fa50e125826ea26b030a860a0bc561cc --- /dev/null +++ b/checkpoint-99000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:8fb0866b3263b8a7f6df8f6684d51a31ad3a6d048580fb3026cb3f9cf3113c39 +size 557 diff --git a/checkpoint-99000/scheduler.bin b/checkpoint-99000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..9c4c9295649c1d8c8f33f00c2f734e433df4d043 --- /dev/null +++ b/checkpoint-99000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7748647449176778a70a9129479eaad00d75e8a35aa6a23ea23f5d4579b237d4 +size 563 diff --git a/checkpoint-99000/unet/config.json b/checkpoint-99000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..bb1cd664fc36d4dc821a4f0fdae4b94d7d99b356 --- /dev/null +++ b/checkpoint-99000/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-99000/unet/diffusion_pytorch_model.bin b/checkpoint-99000/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..c3eff03434ae74e4b1239c07e77b51a96e09d4d7 --- /dev/null +++ b/checkpoint-99000/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:119dea9110bd7de7dea45e084f9a798a368915a70488d8482269744540d8b12c +size 36621969 diff --git a/checkpoint-99000/unet_ema/config.json b/checkpoint-99000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..e04878e275dfe1ad72cad111cb0693e290027b46 --- /dev/null +++ b/checkpoint-99000/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": 99000, + "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-99000/unet_ema/diffusion_pytorch_model.bin b/checkpoint-99000/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..f9cbaf0f78c366a16839f6801684312fc7118edf --- /dev/null +++ b/checkpoint-99000/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:04e922ad50f7b513a7a0a823e4132df3f7147e80a88f791fa765664ee577d3ef +size 36616081 diff --git a/checkpoint-99500/optimizer.bin b/checkpoint-99500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..0bfa23a4cf3c04290494d3f6574b73b982f92c37 --- /dev/null +++ b/checkpoint-99500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:89edad94b1ae7acd1079e8c88c97025dc06d23d2341ed02e5860df12fbf7140d +size 73251413 diff --git a/checkpoint-99500/random_states_0.pkl b/checkpoint-99500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..b5e4492a2fb5af8c36fc0e2e295444de6f1b26ea --- /dev/null +++ b/checkpoint-99500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b7d92e21e96b655483568ac54f5e133f0fc168809f9685f1f3ec86a44078cf6b +size 14663 diff --git a/checkpoint-99500/scaler.pt b/checkpoint-99500/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..859655f8165b1edf31820003d42735b852be12ca --- /dev/null +++ b/checkpoint-99500/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:9985a34f823a2e3dc21d11f8813202c5e32b92fe21411f3ffb89c7a8253731f1 +size 557 diff --git a/checkpoint-99500/scheduler.bin b/checkpoint-99500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..6ed6f6f853b55b620e5fa37a40b096352f6f56c7 --- /dev/null +++ b/checkpoint-99500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:4402874564140fac947a2df2860e31cfcbb59accee7117ed55fdc8bbbb7d4720 +size 563 diff --git a/checkpoint-99500/unet/config.json b/checkpoint-99500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..bb1cd664fc36d4dc821a4f0fdae4b94d7d99b356 --- /dev/null +++ b/checkpoint-99500/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-99500/unet/diffusion_pytorch_model.bin b/checkpoint-99500/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..7bdc538b227e2b4caee3d6c4529490e3171448a2 --- /dev/null +++ b/checkpoint-99500/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:212ce1cee702691be7764e303a94456b22b227080e818798718ec89d73a5a556 +size 36621969 diff --git a/checkpoint-99500/unet_ema/config.json b/checkpoint-99500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..b0adb662c3edd0b3aeb20eb0b96113857015494b --- /dev/null +++ b/checkpoint-99500/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": 99500, + "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-99500/unet_ema/diffusion_pytorch_model.bin b/checkpoint-99500/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..1ede7ab02fc0c5f59e2589da5f65e4d24d22da6a --- /dev/null +++ b/checkpoint-99500/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e1678b70f0807142a589ef111bc7d445dfe2d89a1149a7c43bf5f8b02ef491b6 +size 36616081 diff --git a/unet/diffusion_pytorch_model.bin b/unet/diffusion_pytorch_model.bin index d148acd223507b1ffc7201ba91a7aeea1fc6483c..6f35a6f5487be3b10b964835377a188f0bad8993 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:09452ab0c28b9d9b2a1919692b7188954970b55f7f1687dbd26dbc4c360b60d5 +oid sha256:3b03b5607f964c5f4e4b22320d5bf54627aeeb0ecaa052d3a330673287014162 size 36621969