diff --git a/checkpoint-19500/optimizer.bin b/checkpoint-19500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..18bb2469d8ba14187a160ee55240e6e156799b5b --- /dev/null +++ b/checkpoint-19500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:2c31b5e3715edbe35792da5858d05f04d820fcb711bdc834f2730d953e0536f8 +size 909760147 diff --git a/checkpoint-19500/random_states_0.pkl b/checkpoint-19500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..fd952611551dee34ce6709b1660ea6845fddbe40 --- /dev/null +++ b/checkpoint-19500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:8775b48e519f46489ad302f8c3ae21aa952f1dd35528656b85e8d22479c16af1 +size 17563 diff --git a/checkpoint-19500/scheduler.bin b/checkpoint-19500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..e8a670cb57c3534f93dbcf0df9894cdd1801c876 --- /dev/null +++ b/checkpoint-19500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:fe08a246d09060d1541389aeb144b56a437b717ed48787a12e02f099d780876c +size 559 diff --git a/checkpoint-19500/unet/config.json b/checkpoint-19500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..8798549599c81da847196c0825621ca9a4784799 --- /dev/null +++ b/checkpoint-19500/unet/config.json @@ -0,0 +1,50 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.22.0.dev0", + "act_fn": "silu", + "add_attention": true, + "attention_head_dim": 8, + "attn_norm_num_groups": null, + "block_out_channels": [ + 128, + 128, + 256, + 256, + 512, + 512 + ], + "center_input_sample": false, + "class_embed_type": null, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "AttnDownBlock2D", + "DownBlock2D" + ], + "downsample_padding": 1, + "downsample_type": "conv", + "dropout": 0.0, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "layers_per_block": 2, + "mid_block_scale_factor": 1, + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "out_channels": 3, + "resnet_time_scale_shift": "default", + "sample_size": 512, + "time_embedding_type": "positional", + "up_block_types": [ + "UpBlock2D", + "AttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upsample_type": "conv" +} diff --git a/checkpoint-19500/unet/diffusion_pytorch_model.safetensors b/checkpoint-19500/unet/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..a677a9e7aafd3448db142f980a02a5a8049c1373 --- /dev/null +++ b/checkpoint-19500/unet/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:fb8e0a8c45affb699e4662fdb481b44071421406abe17d8e9deec0f6d188ce7d +size 454741108 diff --git a/checkpoint-19500/unet_ema/config.json b/checkpoint-19500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..af53e7328549ff6f7e3ceda9ed51b15d3f59dbe7 --- /dev/null +++ b/checkpoint-19500/unet_ema/config.json @@ -0,0 +1,57 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.22.0.dev0", + "act_fn": "silu", + "add_attention": true, + "attention_head_dim": 8, + "attn_norm_num_groups": null, + "block_out_channels": [ + 128, + 128, + 256, + 256, + 512, + 512 + ], + "center_input_sample": false, + "class_embed_type": null, + "decay": 0.9999, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "AttnDownBlock2D", + "DownBlock2D" + ], + "downsample_padding": 1, + "downsample_type": "conv", + "dropout": 0.0, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "inv_gamma": 1.0, + "layers_per_block": 2, + "mid_block_scale_factor": 1, + "min_decay": 0.0, + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "optimization_step": 19500, + "out_channels": 3, + "power": 0.75, + "resnet_time_scale_shift": "default", + "sample_size": 512, + "time_embedding_type": "positional", + "up_block_types": [ + "UpBlock2D", + "AttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "update_after_step": 0, + "upsample_type": "conv", + "use_ema_warmup": true +} diff --git a/checkpoint-19500/unet_ema/diffusion_pytorch_model.safetensors b/checkpoint-19500/unet_ema/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..3f92846069917907e86d5369aacce70ea1ec48b9 --- /dev/null +++ b/checkpoint-19500/unet_ema/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:bfc0cbef3f2191e944220483b2b966194fdeb52301eeccd33d0eaa9cb44ed5ec +size 454741108 diff --git a/checkpoint-20000/optimizer.bin b/checkpoint-20000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..68afba95f3d1bc8603087383cbb345d3700082aa --- /dev/null +++ b/checkpoint-20000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:59c03d6188af8e0b437eb1e7d504f7199b7cfd48e2953c4b09b97e3cdb049554 +size 909760147 diff --git a/checkpoint-20000/random_states_0.pkl b/checkpoint-20000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..7a214f8dd587f72992692d17b63fd3fff427f9c5 --- /dev/null +++ b/checkpoint-20000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:9be8f20b67c11d9b60edf55f51c793529d29534c606b88fc745d853b897c004e +size 17563 diff --git a/checkpoint-20000/scheduler.bin b/checkpoint-20000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..283c9099343f8778cd7101d09f30ddcb73141b6c --- /dev/null +++ b/checkpoint-20000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:267d033f39bcd1f55664cafb8f44da465efcf4ed3aae6f8864c2ea1121559674 +size 559 diff --git a/checkpoint-20000/unet/config.json b/checkpoint-20000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..8798549599c81da847196c0825621ca9a4784799 --- /dev/null +++ b/checkpoint-20000/unet/config.json @@ -0,0 +1,50 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.22.0.dev0", + "act_fn": "silu", + "add_attention": true, + "attention_head_dim": 8, + "attn_norm_num_groups": null, + "block_out_channels": [ + 128, + 128, + 256, + 256, + 512, + 512 + ], + "center_input_sample": false, + "class_embed_type": null, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "AttnDownBlock2D", + "DownBlock2D" + ], + "downsample_padding": 1, + "downsample_type": "conv", + "dropout": 0.0, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "layers_per_block": 2, + "mid_block_scale_factor": 1, + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "out_channels": 3, + "resnet_time_scale_shift": "default", + "sample_size": 512, + "time_embedding_type": "positional", + "up_block_types": [ + "UpBlock2D", + "AttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upsample_type": "conv" +} diff --git a/checkpoint-20000/unet/diffusion_pytorch_model.safetensors b/checkpoint-20000/unet/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..1984ad63cdd16b5968d26e5eab4db4542b293e81 --- /dev/null +++ b/checkpoint-20000/unet/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:d5bf7d4aaefcacf1b00b5d0c9b515a27d04dd2aa4c1ac2ac4423a615c8ed5d7d +size 454741108 diff --git a/checkpoint-20000/unet_ema/config.json b/checkpoint-20000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..5da6c853f44e2eff4e4e9a07cc261bdcf5218846 --- /dev/null +++ b/checkpoint-20000/unet_ema/config.json @@ -0,0 +1,57 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.22.0.dev0", + "act_fn": "silu", + "add_attention": true, + "attention_head_dim": 8, + "attn_norm_num_groups": null, + "block_out_channels": [ + 128, + 128, + 256, + 256, + 512, + 512 + ], + "center_input_sample": false, + "class_embed_type": null, + "decay": 0.9999, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "AttnDownBlock2D", + "DownBlock2D" + ], + "downsample_padding": 1, + "downsample_type": "conv", + "dropout": 0.0, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "inv_gamma": 1.0, + "layers_per_block": 2, + "mid_block_scale_factor": 1, + "min_decay": 0.0, + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "optimization_step": 20000, + "out_channels": 3, + "power": 0.75, + "resnet_time_scale_shift": "default", + "sample_size": 512, + "time_embedding_type": "positional", + "up_block_types": [ + "UpBlock2D", + "AttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "update_after_step": 0, + "upsample_type": "conv", + "use_ema_warmup": true +} diff --git a/checkpoint-20000/unet_ema/diffusion_pytorch_model.safetensors b/checkpoint-20000/unet_ema/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..980f6d2c80edd256b5205a72fcb53650f1074f1a --- /dev/null +++ b/checkpoint-20000/unet_ema/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:ae5ac8489c46a2fd07b60cd6e765a64e72060fafb81b24323854697c6d0f6485 +size 454741108 diff --git a/checkpoint-20500/optimizer.bin b/checkpoint-20500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..bfb460b6f2b9be53f6e2b9a5bd782416feb1083a --- /dev/null +++ b/checkpoint-20500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:107d1c60c13279c4124b480b88a5e14518e336b34d11a3151fb078b93d952e1d +size 909760147 diff --git a/checkpoint-20500/random_states_0.pkl b/checkpoint-20500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..548aba2ff90a64b0a51b1659d5b726474289db33 --- /dev/null +++ b/checkpoint-20500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:726cb434d7cc48ae6a0bdddc17a8221c955bda9688d0f3d95fa2b8cee005d488 +size 17563 diff --git a/checkpoint-20500/scheduler.bin b/checkpoint-20500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..dac178b73b7e5c9f2a55caa0652c7ee90fee6840 --- /dev/null +++ b/checkpoint-20500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:68c780e749313837d7646d32e756d8d6889b45b2501cfc2d475be6f3f9521672 +size 559 diff --git a/checkpoint-20500/unet/config.json b/checkpoint-20500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..8798549599c81da847196c0825621ca9a4784799 --- /dev/null +++ b/checkpoint-20500/unet/config.json @@ -0,0 +1,50 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.22.0.dev0", + "act_fn": "silu", + "add_attention": true, + "attention_head_dim": 8, + "attn_norm_num_groups": null, + "block_out_channels": [ + 128, + 128, + 256, + 256, + 512, + 512 + ], + "center_input_sample": false, + "class_embed_type": null, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "AttnDownBlock2D", + "DownBlock2D" + ], + "downsample_padding": 1, + "downsample_type": "conv", + "dropout": 0.0, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "layers_per_block": 2, + "mid_block_scale_factor": 1, + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "out_channels": 3, + "resnet_time_scale_shift": "default", + "sample_size": 512, + "time_embedding_type": "positional", + "up_block_types": [ + "UpBlock2D", + "AttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upsample_type": "conv" +} diff --git a/checkpoint-20500/unet/diffusion_pytorch_model.safetensors b/checkpoint-20500/unet/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..215280f98985cde032e0447c51be09c4c04fe98b --- /dev/null +++ b/checkpoint-20500/unet/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:5601680e60589dd174348c80697e595b4b14206b6a728d2c0e894e541e486321 +size 454741108 diff --git a/checkpoint-20500/unet_ema/config.json b/checkpoint-20500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..74a8985dbdd9731054f9fa84da061da3a847bfc0 --- /dev/null +++ b/checkpoint-20500/unet_ema/config.json @@ -0,0 +1,57 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.22.0.dev0", + "act_fn": "silu", + "add_attention": true, + "attention_head_dim": 8, + "attn_norm_num_groups": null, + "block_out_channels": [ + 128, + 128, + 256, + 256, + 512, + 512 + ], + "center_input_sample": false, + "class_embed_type": null, + "decay": 0.9999, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "AttnDownBlock2D", + "DownBlock2D" + ], + "downsample_padding": 1, + "downsample_type": "conv", + "dropout": 0.0, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "inv_gamma": 1.0, + "layers_per_block": 2, + "mid_block_scale_factor": 1, + "min_decay": 0.0, + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "optimization_step": 20500, + "out_channels": 3, + "power": 0.75, + "resnet_time_scale_shift": "default", + "sample_size": 512, + "time_embedding_type": "positional", + "up_block_types": [ + "UpBlock2D", + "AttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "update_after_step": 0, + "upsample_type": "conv", + "use_ema_warmup": true +} diff --git a/checkpoint-20500/unet_ema/diffusion_pytorch_model.safetensors b/checkpoint-20500/unet_ema/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..35a8fbb0a6778f0b963f37649c96275b7924a575 --- /dev/null +++ b/checkpoint-20500/unet_ema/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:aacbe38ecf942d26f09e20725b20f6599695a337e4491712403af970e021daa9 +size 454741108 diff --git a/checkpoint-21000/optimizer.bin b/checkpoint-21000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..69c3e530371054f57612d25d7240e53604dd8277 --- /dev/null +++ b/checkpoint-21000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:2fbdd8749a1bb5613c5bef9f7b1682fbeb885fe7c4f5f60967ff36457e51fcaa +size 909760147 diff --git a/checkpoint-21000/random_states_0.pkl b/checkpoint-21000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..12f64923bc3873368d50231a30fd449d5646c9b5 --- /dev/null +++ b/checkpoint-21000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b1f9e9c293fa6a089d03058e5cd01acc35092e8887891c7f4b78c2478c2c0ef5 +size 17563 diff --git a/checkpoint-21000/scheduler.bin b/checkpoint-21000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..25f994946bc38c429a7dc10626dddc4c990c2029 --- /dev/null +++ b/checkpoint-21000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:002a4625d0fd66d5be44c1d4c2cdef931195f28bf106fc54b5fc1d12f38322f3 +size 559 diff --git a/checkpoint-21000/unet/config.json b/checkpoint-21000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..8798549599c81da847196c0825621ca9a4784799 --- /dev/null +++ b/checkpoint-21000/unet/config.json @@ -0,0 +1,50 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.22.0.dev0", + "act_fn": "silu", + "add_attention": true, + "attention_head_dim": 8, + "attn_norm_num_groups": null, + "block_out_channels": [ + 128, + 128, + 256, + 256, + 512, + 512 + ], + "center_input_sample": false, + "class_embed_type": null, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "AttnDownBlock2D", + "DownBlock2D" + ], + "downsample_padding": 1, + "downsample_type": "conv", + "dropout": 0.0, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "layers_per_block": 2, + "mid_block_scale_factor": 1, + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "out_channels": 3, + "resnet_time_scale_shift": "default", + "sample_size": 512, + "time_embedding_type": "positional", + "up_block_types": [ + "UpBlock2D", + "AttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upsample_type": "conv" +} diff --git a/checkpoint-21000/unet/diffusion_pytorch_model.safetensors b/checkpoint-21000/unet/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..b9e03aececca9488be00955c336701e9333e64e7 --- /dev/null +++ b/checkpoint-21000/unet/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:6bad1f9ecc1df675b686dd78945c1ba2c5ef0a2f062e44cd685b2fd530f0d9a1 +size 454741108 diff --git a/checkpoint-21000/unet_ema/config.json b/checkpoint-21000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..1513044deee4a3d6b8b6a382be006c738c2cfc9b --- /dev/null +++ b/checkpoint-21000/unet_ema/config.json @@ -0,0 +1,57 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.22.0.dev0", + "act_fn": "silu", + "add_attention": true, + "attention_head_dim": 8, + "attn_norm_num_groups": null, + "block_out_channels": [ + 128, + 128, + 256, + 256, + 512, + 512 + ], + "center_input_sample": false, + "class_embed_type": null, + "decay": 0.9999, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "AttnDownBlock2D", + "DownBlock2D" + ], + "downsample_padding": 1, + "downsample_type": "conv", + "dropout": 0.0, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "inv_gamma": 1.0, + "layers_per_block": 2, + "mid_block_scale_factor": 1, + "min_decay": 0.0, + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "optimization_step": 21000, + "out_channels": 3, + "power": 0.75, + "resnet_time_scale_shift": "default", + "sample_size": 512, + "time_embedding_type": "positional", + "up_block_types": [ + "UpBlock2D", + "AttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "update_after_step": 0, + "upsample_type": "conv", + "use_ema_warmup": true +} diff --git a/checkpoint-21000/unet_ema/diffusion_pytorch_model.safetensors b/checkpoint-21000/unet_ema/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..d85ab9a6de3a187f21fa5c38a7b569af3b6b93c6 --- /dev/null +++ b/checkpoint-21000/unet_ema/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:2edb6f038485ccc3685c8079027bde8aafb49cf66e708c0d6daa02700b7ff0d4 +size 454741108 diff --git a/checkpoint-21500/optimizer.bin b/checkpoint-21500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..88cfefcd2f267ffbf52e1d1971bc147c214d4eed --- /dev/null +++ b/checkpoint-21500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:73c39dfadd66e2d608b8547996ab2fd3ab652e9f3ac97ffa9a30092cc8cea18e +size 909760147 diff --git a/checkpoint-21500/random_states_0.pkl b/checkpoint-21500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..6e750091db76a2c1b29e5439aafc00ec2c5be6ff --- /dev/null +++ b/checkpoint-21500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:0114a12469025dd321febe716f8649a933aa65003bdb49cc5e42d6c4ed1167bb +size 17563 diff --git a/checkpoint-21500/scheduler.bin b/checkpoint-21500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..1bcc8d52ebb91546490dff80ecef4e72dc9886bf --- /dev/null +++ b/checkpoint-21500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:1833f067607031db91019c607f129265b13e6c3e5fc7740212e97f01a1e1981c +size 559 diff --git a/checkpoint-21500/unet/config.json b/checkpoint-21500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..8798549599c81da847196c0825621ca9a4784799 --- /dev/null +++ b/checkpoint-21500/unet/config.json @@ -0,0 +1,50 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.22.0.dev0", + "act_fn": "silu", + "add_attention": true, + "attention_head_dim": 8, + "attn_norm_num_groups": null, + "block_out_channels": [ + 128, + 128, + 256, + 256, + 512, + 512 + ], + "center_input_sample": false, + "class_embed_type": null, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "AttnDownBlock2D", + "DownBlock2D" + ], + "downsample_padding": 1, + "downsample_type": "conv", + "dropout": 0.0, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "layers_per_block": 2, + "mid_block_scale_factor": 1, + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "out_channels": 3, + "resnet_time_scale_shift": "default", + "sample_size": 512, + "time_embedding_type": "positional", + "up_block_types": [ + "UpBlock2D", + "AttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upsample_type": "conv" +} diff --git a/checkpoint-21500/unet/diffusion_pytorch_model.safetensors b/checkpoint-21500/unet/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..00affd48ea2625711cfe014b2ca27dc3e948acb4 --- /dev/null +++ b/checkpoint-21500/unet/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:77cda02e7bd39a167fcd9d85637a15526c356082b086ca7258960e5c543e1943 +size 454741108 diff --git a/checkpoint-21500/unet_ema/config.json b/checkpoint-21500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..957caccf05fd19bf1e5fb09c968ae4d85518ff60 --- /dev/null +++ b/checkpoint-21500/unet_ema/config.json @@ -0,0 +1,57 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.22.0.dev0", + "act_fn": "silu", + "add_attention": true, + "attention_head_dim": 8, + "attn_norm_num_groups": null, + "block_out_channels": [ + 128, + 128, + 256, + 256, + 512, + 512 + ], + "center_input_sample": false, + "class_embed_type": null, + "decay": 0.9999, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "AttnDownBlock2D", + "DownBlock2D" + ], + "downsample_padding": 1, + "downsample_type": "conv", + "dropout": 0.0, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "inv_gamma": 1.0, + "layers_per_block": 2, + "mid_block_scale_factor": 1, + "min_decay": 0.0, + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "optimization_step": 21500, + "out_channels": 3, + "power": 0.75, + "resnet_time_scale_shift": "default", + "sample_size": 512, + "time_embedding_type": "positional", + "up_block_types": [ + "UpBlock2D", + "AttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "update_after_step": 0, + "upsample_type": "conv", + "use_ema_warmup": true +} diff --git a/checkpoint-21500/unet_ema/diffusion_pytorch_model.safetensors b/checkpoint-21500/unet_ema/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..dd0a843fd016a25433937eb4f2ceabc0d8175254 --- /dev/null +++ b/checkpoint-21500/unet_ema/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:f6a4a3b33bdd95c0fa42541ea166a2bbb95318a6d6ad992d9589221b35cebe9a +size 454741108 diff --git a/checkpoint-22000/optimizer.bin b/checkpoint-22000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..975297ea8782b93adc211f8ff8a639403087af76 --- /dev/null +++ b/checkpoint-22000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:42f40e43292cdd65fe4a0fa79dc644d1e6a8a6454102bc64229672d2417d42c5 +size 909760147 diff --git a/checkpoint-22000/random_states_0.pkl b/checkpoint-22000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..ba5bfe7959b6b9d194f3f50565000db8c04040b8 --- /dev/null +++ b/checkpoint-22000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:5d31fa4963728197ea6ab8b93efd42396dbbd7be6b7e21dcaaac884740c44b2b +size 17563 diff --git a/checkpoint-22000/scheduler.bin b/checkpoint-22000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..1959ac6ae6050688f711a3e65a62b76a6b9cb194 --- /dev/null +++ b/checkpoint-22000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:4f383d5dc8e222d67bdb6e8392b8ef0cc45a3b7971ab8c35f663a9fc7f0fb1eb +size 559 diff --git a/checkpoint-22000/unet/config.json b/checkpoint-22000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..8798549599c81da847196c0825621ca9a4784799 --- /dev/null +++ b/checkpoint-22000/unet/config.json @@ -0,0 +1,50 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.22.0.dev0", + "act_fn": "silu", + "add_attention": true, + "attention_head_dim": 8, + "attn_norm_num_groups": null, + "block_out_channels": [ + 128, + 128, + 256, + 256, + 512, + 512 + ], + "center_input_sample": false, + "class_embed_type": null, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "AttnDownBlock2D", + "DownBlock2D" + ], + "downsample_padding": 1, + "downsample_type": "conv", + "dropout": 0.0, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "layers_per_block": 2, + "mid_block_scale_factor": 1, + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "out_channels": 3, + "resnet_time_scale_shift": "default", + "sample_size": 512, + "time_embedding_type": "positional", + "up_block_types": [ + "UpBlock2D", + "AttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upsample_type": "conv" +} diff --git a/checkpoint-22000/unet/diffusion_pytorch_model.safetensors b/checkpoint-22000/unet/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..cc32689359e78e0e5817da2237d37c2fc2c58156 --- /dev/null +++ b/checkpoint-22000/unet/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:6f6032e4fe6c86adc0d4aac9073048e24f3e33e92cf39ba141f2738686ae060c +size 454741108 diff --git a/checkpoint-22000/unet_ema/config.json b/checkpoint-22000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..e63ae4b70740ffd598812bda5dcd17db2fa409fd --- /dev/null +++ b/checkpoint-22000/unet_ema/config.json @@ -0,0 +1,57 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.22.0.dev0", + "act_fn": "silu", + "add_attention": true, + "attention_head_dim": 8, + "attn_norm_num_groups": null, + "block_out_channels": [ + 128, + 128, + 256, + 256, + 512, + 512 + ], + "center_input_sample": false, + "class_embed_type": null, + "decay": 0.9999, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "AttnDownBlock2D", + "DownBlock2D" + ], + "downsample_padding": 1, + "downsample_type": "conv", + "dropout": 0.0, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "inv_gamma": 1.0, + "layers_per_block": 2, + "mid_block_scale_factor": 1, + "min_decay": 0.0, + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "optimization_step": 22000, + "out_channels": 3, + "power": 0.75, + "resnet_time_scale_shift": "default", + "sample_size": 512, + "time_embedding_type": "positional", + "up_block_types": [ + "UpBlock2D", + "AttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "update_after_step": 0, + "upsample_type": "conv", + "use_ema_warmup": true +} diff --git a/checkpoint-22000/unet_ema/diffusion_pytorch_model.safetensors b/checkpoint-22000/unet_ema/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..0cdb54abf57e6c47c6347eac979fe4db7f058494 --- /dev/null +++ b/checkpoint-22000/unet_ema/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:db7269df7f05dd1c5418df8282bb12c5c98c2f7cd2c069fd182c21decd3ad455 +size 454741108 diff --git a/checkpoint-22500/optimizer.bin b/checkpoint-22500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..247935ec275f5004a38fb9772e18f60816464310 --- /dev/null +++ b/checkpoint-22500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:ae262f68f49193fa7481825a21752827e0699f1f564f8cc0026b9381f4a7d195 +size 909760147 diff --git a/checkpoint-22500/random_states_0.pkl b/checkpoint-22500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..bb114301d7c62a715e91440fd0823e199dfa807c --- /dev/null +++ b/checkpoint-22500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:31ba9fac92fe32d73cf2fe55ea4a900dccc86d87f7f59b6b6855aacf643f5c02 +size 17563 diff --git a/checkpoint-22500/scheduler.bin b/checkpoint-22500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..85e949356c51542fd321e2a849e72a5c2d31e1b8 --- /dev/null +++ b/checkpoint-22500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:39b0ddbd236cad5f857ee9a5d33386aca41732f459cf6a984e627ec0a79ef838 +size 559 diff --git a/checkpoint-22500/unet/config.json b/checkpoint-22500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..8798549599c81da847196c0825621ca9a4784799 --- /dev/null +++ b/checkpoint-22500/unet/config.json @@ -0,0 +1,50 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.22.0.dev0", + "act_fn": "silu", + "add_attention": true, + "attention_head_dim": 8, + "attn_norm_num_groups": null, + "block_out_channels": [ + 128, + 128, + 256, + 256, + 512, + 512 + ], + "center_input_sample": false, + "class_embed_type": null, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "AttnDownBlock2D", + "DownBlock2D" + ], + "downsample_padding": 1, + "downsample_type": "conv", + "dropout": 0.0, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "layers_per_block": 2, + "mid_block_scale_factor": 1, + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "out_channels": 3, + "resnet_time_scale_shift": "default", + "sample_size": 512, + "time_embedding_type": "positional", + "up_block_types": [ + "UpBlock2D", + "AttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upsample_type": "conv" +} diff --git a/checkpoint-22500/unet/diffusion_pytorch_model.safetensors b/checkpoint-22500/unet/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..01cd228e98cac306f9c86c87cecb11294c7a9f73 --- /dev/null +++ b/checkpoint-22500/unet/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:3327aa1ad6b6728e7a378db6f0bd4f1191c648e7faec8cbd2cb8904fc48d6940 +size 454741108 diff --git a/checkpoint-22500/unet_ema/config.json b/checkpoint-22500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..c5dd2667868a2415bcd78be18d7c5d495ceccab6 --- /dev/null +++ b/checkpoint-22500/unet_ema/config.json @@ -0,0 +1,57 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.22.0.dev0", + "act_fn": "silu", + "add_attention": true, + "attention_head_dim": 8, + "attn_norm_num_groups": null, + "block_out_channels": [ + 128, + 128, + 256, + 256, + 512, + 512 + ], + "center_input_sample": false, + "class_embed_type": null, + "decay": 0.9999, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "AttnDownBlock2D", + "DownBlock2D" + ], + "downsample_padding": 1, + "downsample_type": "conv", + "dropout": 0.0, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "inv_gamma": 1.0, + "layers_per_block": 2, + "mid_block_scale_factor": 1, + "min_decay": 0.0, + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "optimization_step": 22500, + "out_channels": 3, + "power": 0.75, + "resnet_time_scale_shift": "default", + "sample_size": 512, + "time_embedding_type": "positional", + "up_block_types": [ + "UpBlock2D", + "AttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "update_after_step": 0, + "upsample_type": "conv", + "use_ema_warmup": true +} diff --git a/checkpoint-22500/unet_ema/diffusion_pytorch_model.safetensors b/checkpoint-22500/unet_ema/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..4a5931ae1102c68b03cf77d64d45472e56ee740e --- /dev/null +++ b/checkpoint-22500/unet_ema/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:121ad77f14af7362c6b6c722ad81e5734bf7298110d9f475b770a249e9a04055 +size 454741108 diff --git a/checkpoint-23000/optimizer.bin b/checkpoint-23000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..0fcf31eca58dd38ce12d03ccc3eee9dbdc00a1d5 --- /dev/null +++ b/checkpoint-23000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:d1c87fb75a5ba5129223dcfb3bd3858404048406734b85090f04f963b0a3b58c +size 909760147 diff --git a/checkpoint-23000/random_states_0.pkl b/checkpoint-23000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..ad2132ab926c0594c557689d04496156db8d3450 --- /dev/null +++ b/checkpoint-23000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:2fdc4167c21c5a0ac9fb9c2177c0e46c82844d8b50d6e8996ffbf01003ec884c +size 17563 diff --git a/checkpoint-23000/scheduler.bin b/checkpoint-23000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..9465601b3a5fd42e4955ee3621bc55f28e22f415 --- /dev/null +++ b/checkpoint-23000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:ceb46abef0b21bdccf3f4f7a7019122718eca4abea2bc03f53c4e9a15820bd23 +size 559 diff --git a/checkpoint-23000/unet/config.json b/checkpoint-23000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..8798549599c81da847196c0825621ca9a4784799 --- /dev/null +++ b/checkpoint-23000/unet/config.json @@ -0,0 +1,50 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.22.0.dev0", + "act_fn": "silu", + "add_attention": true, + "attention_head_dim": 8, + "attn_norm_num_groups": null, + "block_out_channels": [ + 128, + 128, + 256, + 256, + 512, + 512 + ], + "center_input_sample": false, + "class_embed_type": null, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "AttnDownBlock2D", + "DownBlock2D" + ], + "downsample_padding": 1, + "downsample_type": "conv", + "dropout": 0.0, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "layers_per_block": 2, + "mid_block_scale_factor": 1, + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "out_channels": 3, + "resnet_time_scale_shift": "default", + "sample_size": 512, + "time_embedding_type": "positional", + "up_block_types": [ + "UpBlock2D", + "AttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upsample_type": "conv" +} diff --git a/checkpoint-23000/unet/diffusion_pytorch_model.safetensors b/checkpoint-23000/unet/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..7bc29dffe4c2e8737f5adb1401dd5299d0d00192 --- /dev/null +++ b/checkpoint-23000/unet/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:a4e3d0d58e88f78b1858d6b15d7e53cd5572ea94e8613c67f48076a71171c96c +size 454741108 diff --git a/checkpoint-23000/unet_ema/config.json b/checkpoint-23000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..ae183fc410d1e6340ee10fb791f33268a3a95a7c --- /dev/null +++ b/checkpoint-23000/unet_ema/config.json @@ -0,0 +1,57 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.22.0.dev0", + "act_fn": "silu", + "add_attention": true, + "attention_head_dim": 8, + "attn_norm_num_groups": null, + "block_out_channels": [ + 128, + 128, + 256, + 256, + 512, + 512 + ], + "center_input_sample": false, + "class_embed_type": null, + "decay": 0.9999, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "AttnDownBlock2D", + "DownBlock2D" + ], + "downsample_padding": 1, + "downsample_type": "conv", + "dropout": 0.0, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "inv_gamma": 1.0, + "layers_per_block": 2, + "mid_block_scale_factor": 1, + "min_decay": 0.0, + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "optimization_step": 23000, + "out_channels": 3, + "power": 0.75, + "resnet_time_scale_shift": "default", + "sample_size": 512, + "time_embedding_type": "positional", + "up_block_types": [ + "UpBlock2D", + "AttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "update_after_step": 0, + "upsample_type": "conv", + "use_ema_warmup": true +} diff --git a/checkpoint-23000/unet_ema/diffusion_pytorch_model.safetensors b/checkpoint-23000/unet_ema/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..d3a2ff06b028082ce233719cffea136c4f24d3b9 --- /dev/null +++ b/checkpoint-23000/unet_ema/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:133d511380dcff5c23c20e625c47e37e67d0b49282b281ad284ad448d334c414 +size 454741108 diff --git a/checkpoint-23500/optimizer.bin b/checkpoint-23500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..f3823028994e41c56e4480d4e670244695d49975 --- /dev/null +++ b/checkpoint-23500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b1c66ea21db79f0b09327f98fdb9ed09decd136c6f3952411572005ad28fab70 +size 909760147 diff --git a/checkpoint-23500/random_states_0.pkl b/checkpoint-23500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..150e10fadaf84ba362be59ab05c37fddb7cba94b --- /dev/null +++ b/checkpoint-23500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:09bbcc1f8b7acf7591762cd3b0a8ff45dc2cf0dc74cc5602df596fb262303112 +size 17563 diff --git a/checkpoint-23500/scheduler.bin b/checkpoint-23500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..cad5b31f9311efffc1f81e8824b3f87ae867f3c5 --- /dev/null +++ b/checkpoint-23500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:9df5470b0968f3b94d19d7496fd3552eb5ed10c1e2633651ce844121594600a3 +size 559 diff --git a/checkpoint-23500/unet/config.json b/checkpoint-23500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..8798549599c81da847196c0825621ca9a4784799 --- /dev/null +++ b/checkpoint-23500/unet/config.json @@ -0,0 +1,50 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.22.0.dev0", + "act_fn": "silu", + "add_attention": true, + "attention_head_dim": 8, + "attn_norm_num_groups": null, + "block_out_channels": [ + 128, + 128, + 256, + 256, + 512, + 512 + ], + "center_input_sample": false, + "class_embed_type": null, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "AttnDownBlock2D", + "DownBlock2D" + ], + "downsample_padding": 1, + "downsample_type": "conv", + "dropout": 0.0, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "layers_per_block": 2, + "mid_block_scale_factor": 1, + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "out_channels": 3, + "resnet_time_scale_shift": "default", + "sample_size": 512, + "time_embedding_type": "positional", + "up_block_types": [ + "UpBlock2D", + "AttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upsample_type": "conv" +} diff --git a/checkpoint-23500/unet/diffusion_pytorch_model.safetensors b/checkpoint-23500/unet/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..43891d2a7b2f6e93c3139a5165765b79b4e3ec5a --- /dev/null +++ b/checkpoint-23500/unet/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:31a9a925c00444b86a917f2a0bd7b84b8cce2548da10083ba1efca8bcfd888c2 +size 454741108 diff --git a/checkpoint-23500/unet_ema/config.json b/checkpoint-23500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..6d707b3022e673ae672e56df9ed0d6a877fad6bb --- /dev/null +++ b/checkpoint-23500/unet_ema/config.json @@ -0,0 +1,57 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.22.0.dev0", + "act_fn": "silu", + "add_attention": true, + "attention_head_dim": 8, + "attn_norm_num_groups": null, + "block_out_channels": [ + 128, + 128, + 256, + 256, + 512, + 512 + ], + "center_input_sample": false, + "class_embed_type": null, + "decay": 0.9999, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "AttnDownBlock2D", + "DownBlock2D" + ], + "downsample_padding": 1, + "downsample_type": "conv", + "dropout": 0.0, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "inv_gamma": 1.0, + "layers_per_block": 2, + "mid_block_scale_factor": 1, + "min_decay": 0.0, + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "optimization_step": 23500, + "out_channels": 3, + "power": 0.75, + "resnet_time_scale_shift": "default", + "sample_size": 512, + "time_embedding_type": "positional", + "up_block_types": [ + "UpBlock2D", + "AttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "update_after_step": 0, + "upsample_type": "conv", + "use_ema_warmup": true +} diff --git a/checkpoint-23500/unet_ema/diffusion_pytorch_model.safetensors b/checkpoint-23500/unet_ema/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..de5c4a18232cd76b418b913494ace43b5fb59dbb --- /dev/null +++ b/checkpoint-23500/unet_ema/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:4c07b3ac0210cd0eb48d6a485ef16262633d1bd4a46320371a3c1a8f6e2237a6 +size 454741108 diff --git a/checkpoint-24000/optimizer.bin b/checkpoint-24000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..9e701c921dbcaa097650996e5b8c2dc39384a087 --- /dev/null +++ b/checkpoint-24000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:252bf0bce2d1ebe44caf04cc46939199b5b0ca8770c115edb470f8fb0bd5ea4c +size 909760147 diff --git a/checkpoint-24000/random_states_0.pkl b/checkpoint-24000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..322ef907ed48e29fc1983d9fdc9fd3b3c79b6987 --- /dev/null +++ b/checkpoint-24000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:42b44bde4d5dc11d6a20cf884c39478ff03cbd660dcf9ad557e0c2fe07c0d591 +size 17563 diff --git a/checkpoint-24000/scheduler.bin b/checkpoint-24000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..68b311240bb26b3d12f0555d6f261ab3b485c0b1 --- /dev/null +++ b/checkpoint-24000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:0f1343f02b97dc07e714edfb38c8dc2a9fa29b5226f2a2868358b58be3e7827e +size 559 diff --git a/checkpoint-24000/unet/config.json b/checkpoint-24000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..8798549599c81da847196c0825621ca9a4784799 --- /dev/null +++ b/checkpoint-24000/unet/config.json @@ -0,0 +1,50 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.22.0.dev0", + "act_fn": "silu", + "add_attention": true, + "attention_head_dim": 8, + "attn_norm_num_groups": null, + "block_out_channels": [ + 128, + 128, + 256, + 256, + 512, + 512 + ], + "center_input_sample": false, + "class_embed_type": null, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "AttnDownBlock2D", + "DownBlock2D" + ], + "downsample_padding": 1, + "downsample_type": "conv", + "dropout": 0.0, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "layers_per_block": 2, + "mid_block_scale_factor": 1, + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "out_channels": 3, + "resnet_time_scale_shift": "default", + "sample_size": 512, + "time_embedding_type": "positional", + "up_block_types": [ + "UpBlock2D", + "AttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upsample_type": "conv" +} diff --git a/checkpoint-24000/unet/diffusion_pytorch_model.safetensors b/checkpoint-24000/unet/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..eae0ca88f86f602aca078cc19d430eaf0375ff99 --- /dev/null +++ b/checkpoint-24000/unet/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:dea5238161957df04f215c18abc83b1264213988fb4ce586cfca16f3ca79cc0d +size 454741108 diff --git a/checkpoint-24000/unet_ema/config.json b/checkpoint-24000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..037ebb8d6f7c36619d85b77707db447e423ba28a --- /dev/null +++ b/checkpoint-24000/unet_ema/config.json @@ -0,0 +1,57 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.22.0.dev0", + "act_fn": "silu", + "add_attention": true, + "attention_head_dim": 8, + "attn_norm_num_groups": null, + "block_out_channels": [ + 128, + 128, + 256, + 256, + 512, + 512 + ], + "center_input_sample": false, + "class_embed_type": null, + "decay": 0.9999, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "AttnDownBlock2D", + "DownBlock2D" + ], + "downsample_padding": 1, + "downsample_type": "conv", + "dropout": 0.0, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 3, + "inv_gamma": 1.0, + "layers_per_block": 2, + "mid_block_scale_factor": 1, + "min_decay": 0.0, + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "optimization_step": 24000, + "out_channels": 3, + "power": 0.75, + "resnet_time_scale_shift": "default", + "sample_size": 512, + "time_embedding_type": "positional", + "up_block_types": [ + "UpBlock2D", + "AttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "update_after_step": 0, + "upsample_type": "conv", + "use_ema_warmup": true +} diff --git a/checkpoint-24000/unet_ema/diffusion_pytorch_model.safetensors b/checkpoint-24000/unet_ema/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..1b3c6a6bcd32b6cbef4e773c0a44cbdb383a2bb8 --- /dev/null +++ b/checkpoint-24000/unet_ema/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c8c1139fe21b97ad7ab757480f4739e0c71181aad3d15e59ca223526bf0db98a +size 454741108 diff --git a/logs/train_unconditional/events.out.tfevents.1696511328.jupiter.4184.0 b/logs/train_unconditional/events.out.tfevents.1696511328.jupiter.4184.0 index bbd0957526ddd20c0b2c2d4db12b7fb0eb5cebe3..a0d918619319c5f4db98b928a1cef6da1233ab8d 100644 --- a/logs/train_unconditional/events.out.tfevents.1696511328.jupiter.4184.0 +++ b/logs/train_unconditional/events.out.tfevents.1696511328.jupiter.4184.0 @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:7c07c97cc516a5edf10b137a55945e3d0e501cac3893c1a48d48ab0298559e84 -size 22830013 +oid sha256:46eaf55ff6b33cf2ceecd59aa6190a321d8ce9ebc0d11c19deef7e478c9dfff6 +size 26168541 diff --git a/unet/diffusion_pytorch_model.safetensors b/unet/diffusion_pytorch_model.safetensors index dc39878e2b5ad4b22a17e9dfdccef79831e15dcd..a6b075684b2fa0b563c1459ac24af4340f834c33 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:8e45a9b32cc67dccb66e3b70f8dab3cc33b915aace9230dc0a3f776e54588623 +oid sha256:82311685434502c03943f310aadf766a8db0d7ce628ccbadc6c8f23a49ee5f6a size 454741108