diff --git a/.gitignore b/.gitignore new file mode 100644 index 0000000000000000000000000000000000000000..79d1facbe641aa7da396a0875b650788e3edef65 --- /dev/null +++ b/.gitignore @@ -0,0 +1,2 @@ +step_* +epoch_* diff --git a/checkpoint-1000/optimizer.bin b/checkpoint-1000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..132521989a5dfea11f35201b2674d16ee2547ed2 --- /dev/null +++ b/checkpoint-1000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:d69bd67808710d2b0ec62dc379f55793f4d7e24ad1a392243149c2659f9a25af +size 57349893 diff --git a/checkpoint-1000/random_states_0.pkl b/checkpoint-1000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..a657ce8b5f9bfa584165c77584c0da682c7315ac --- /dev/null +++ b/checkpoint-1000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:ffe6f0ec5b4ec531c5ffb781160ad26d28e673da2e19fabec62eb84b83fe4284 +size 14599 diff --git a/checkpoint-1000/scaler.pt b/checkpoint-1000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..efdbd3c795f6b0d4144e68355e99c220ccdedd09 --- /dev/null +++ b/checkpoint-1000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:68cff80b680ddf6e7abbef98b5f336b97f9b5963e2209307f639383870e8cc71 +size 557 diff --git a/checkpoint-1000/scheduler.bin b/checkpoint-1000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..95f1aa22a9a5b388c51fc3ec8fdaba73b8f70966 --- /dev/null +++ b/checkpoint-1000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:53ad4638aaf323211ac1f24bdf9754cac9b86c6770c93f327301f1be9d439da8 +size 563 diff --git a/checkpoint-1000/unet/config.json b/checkpoint-1000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..99ebec7e97d8731e19f09a61fd292f2b02845211 --- /dev/null +++ b/checkpoint-1000/unet/config.json @@ -0,0 +1,40 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "add_attention": true, + "attention_head_dim": 8, + "block_out_channels": [ + 64, + 64, + 128, + 128 + ], + "center_input_sample": false, + "class_embed_type": null, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "AttnDownBlock2D" + ], + "downsample_padding": 1, + "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": 256, + "time_embedding_type": "positional", + "up_block_types": [ + "AttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ] +} diff --git a/checkpoint-1000/unet/diffusion_pytorch_model.bin b/checkpoint-1000/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..8e8bcd32bb969674fee534bb1d3a06ea3f0a1af4 --- /dev/null +++ b/checkpoint-1000/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:2af1183038222a1d46766f65448cccbfe6f0263e5e17422a926916e81e8df340 +size 28664529 diff --git a/checkpoint-1000/unet_ema/config.json b/checkpoint-1000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..9c3e57882f07feed46bc5e5529b2fbab608f2ee2 --- /dev/null +++ b/checkpoint-1000/unet_ema/config.json @@ -0,0 +1,47 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "add_attention": true, + "attention_head_dim": 8, + "block_out_channels": [ + 64, + 64, + 128, + 128 + ], + "center_input_sample": false, + "class_embed_type": null, + "decay": 0.9999, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "AttnDownBlock2D" + ], + "downsample_padding": 1, + "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": 1000, + "out_channels": 3, + "power": 0.75, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_embedding_type": "positional", + "up_block_types": [ + "AttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "update_after_step": 0, + "use_ema_warmup": true +} diff --git a/checkpoint-1000/unet_ema/diffusion_pytorch_model.bin b/checkpoint-1000/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..1fdf07f2a71187530090997b14e9ffe998453df2 --- /dev/null +++ b/checkpoint-1000/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:626d3c16a3f28fee65a62be25236dd9fff77bc91aebe306df752c38126a5c4cc +size 28659921 diff --git a/checkpoint-1500/optimizer.bin b/checkpoint-1500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..c5d12d85de2957a68dd68456a1cd6d0b6c96a2a4 --- /dev/null +++ b/checkpoint-1500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:ad43519962c3e068976172fb9aab746a8f15237286282063693896264482ff79 +size 57349893 diff --git a/checkpoint-1500/random_states_0.pkl b/checkpoint-1500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..4b03a090e79ac185968bdafc30533efb51ec4de8 --- /dev/null +++ b/checkpoint-1500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c2b5322996d86576b10a46a10c902f121097533613600d5fdbe2073907832559 +size 14599 diff --git a/checkpoint-1500/scaler.pt b/checkpoint-1500/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..69c76b8f27f096d1d1a3d9d0e387af1a9cf5028d --- /dev/null +++ b/checkpoint-1500/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:203a72d6c29f42a0e2964fdddc8d7a98df1eccee78fea9de0fa416613390f5c6 +size 557 diff --git a/checkpoint-1500/scheduler.bin b/checkpoint-1500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..5f7d612985238320c8b82d6e70d8d10255125e75 --- /dev/null +++ b/checkpoint-1500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:96780da0007a39f94691a9591ca3c6703a49160a753b6b144dbba68be213815d +size 563 diff --git a/checkpoint-1500/unet/config.json b/checkpoint-1500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..99ebec7e97d8731e19f09a61fd292f2b02845211 --- /dev/null +++ b/checkpoint-1500/unet/config.json @@ -0,0 +1,40 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "add_attention": true, + "attention_head_dim": 8, + "block_out_channels": [ + 64, + 64, + 128, + 128 + ], + "center_input_sample": false, + "class_embed_type": null, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "AttnDownBlock2D" + ], + "downsample_padding": 1, + "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": 256, + "time_embedding_type": "positional", + "up_block_types": [ + "AttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ] +} diff --git a/checkpoint-1500/unet/diffusion_pytorch_model.bin b/checkpoint-1500/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..6ec0d5f0235f8128a6955f407f2863e7f287cc56 --- /dev/null +++ b/checkpoint-1500/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b76f7137e2408f6030a4480a97b321d68428eecd23dcec69262b6fe6c4bcd758 +size 28664529 diff --git a/checkpoint-1500/unet_ema/config.json b/checkpoint-1500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..d23262f1df2b0677c63c7a7d25d03c24083d6ccc --- /dev/null +++ b/checkpoint-1500/unet_ema/config.json @@ -0,0 +1,47 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "add_attention": true, + "attention_head_dim": 8, + "block_out_channels": [ + 64, + 64, + 128, + 128 + ], + "center_input_sample": false, + "class_embed_type": null, + "decay": 0.9999, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "AttnDownBlock2D" + ], + "downsample_padding": 1, + "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": 1500, + "out_channels": 3, + "power": 0.75, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_embedding_type": "positional", + "up_block_types": [ + "AttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "update_after_step": 0, + "use_ema_warmup": true +} diff --git a/checkpoint-1500/unet_ema/diffusion_pytorch_model.bin b/checkpoint-1500/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..86ab33907b9ce5d2252927e41b4aef3f2136dda8 --- /dev/null +++ b/checkpoint-1500/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:3effbd9636bb02708055b89e22fa4cacf732956bcf2fb96a2be4a0f298851b99 +size 28659921 diff --git a/checkpoint-2000/optimizer.bin b/checkpoint-2000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..65fada13525e2fd1f264958327f9b0cf39acfa47 --- /dev/null +++ b/checkpoint-2000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:8dfc68c6103da8db2b20e56a2fbed6da3a89778a3cf2cdef3382096715bff81b +size 57349893 diff --git a/checkpoint-2000/random_states_0.pkl b/checkpoint-2000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..9bd4482420b9d66f33ae548abc5ff9b2880ffe4f --- /dev/null +++ b/checkpoint-2000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e0eb6aed3447200ab080c4795ea41be4bad9a7d682b2ad0b05e790b91cab9951 +size 14599 diff --git a/checkpoint-2000/scaler.pt b/checkpoint-2000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..21ece35416ce79724d347155f11cfa297b97cabc --- /dev/null +++ b/checkpoint-2000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:dd2de9749828adacdf103bf6e9592702bb7067a2c1df27dd62ab38c1eb8c070f +size 557 diff --git a/checkpoint-2000/scheduler.bin b/checkpoint-2000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..c78c51a014283675416ea39f563e8c48c566bad4 --- /dev/null +++ b/checkpoint-2000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:fc1b02b51d165d3fbe6ffe2e2469da70ae15ea26b940456dd33a2859bbc8d145 +size 563 diff --git a/checkpoint-2000/unet/config.json b/checkpoint-2000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..99ebec7e97d8731e19f09a61fd292f2b02845211 --- /dev/null +++ b/checkpoint-2000/unet/config.json @@ -0,0 +1,40 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "add_attention": true, + "attention_head_dim": 8, + "block_out_channels": [ + 64, + 64, + 128, + 128 + ], + "center_input_sample": false, + "class_embed_type": null, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "AttnDownBlock2D" + ], + "downsample_padding": 1, + "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": 256, + "time_embedding_type": "positional", + "up_block_types": [ + "AttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ] +} diff --git a/checkpoint-2000/unet/diffusion_pytorch_model.bin b/checkpoint-2000/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..b9119ad52ccfa5401594a7489ef91b4aa5f4bd8e --- /dev/null +++ b/checkpoint-2000/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:a97096dd6845998c073c1615491dcfd0bdaf6a94d1fbaebe63dac3d6d945c8bf +size 28664529 diff --git a/checkpoint-2000/unet_ema/config.json b/checkpoint-2000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..1bb62012916b2d0e47637fa3899e11928655f327 --- /dev/null +++ b/checkpoint-2000/unet_ema/config.json @@ -0,0 +1,47 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "add_attention": true, + "attention_head_dim": 8, + "block_out_channels": [ + 64, + 64, + 128, + 128 + ], + "center_input_sample": false, + "class_embed_type": null, + "decay": 0.9999, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "AttnDownBlock2D" + ], + "downsample_padding": 1, + "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": 2000, + "out_channels": 3, + "power": 0.75, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_embedding_type": "positional", + "up_block_types": [ + "AttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "update_after_step": 0, + "use_ema_warmup": true +} diff --git a/checkpoint-2000/unet_ema/diffusion_pytorch_model.bin b/checkpoint-2000/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..e23a3499f44dbe10992e08e52921b25cd2fe37f6 --- /dev/null +++ b/checkpoint-2000/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:bcc503d5242a1124ed86f308e191f1661cbee5728935120099798613db0cf410 +size 28659921 diff --git a/checkpoint-2500/optimizer.bin b/checkpoint-2500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..92d991b28ff16981310428556ea7938b237567a6 --- /dev/null +++ b/checkpoint-2500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:61a2198e5a088aed75904de57deb9979bdf197ff24d97bb9ec50df5a4d1f9622 +size 57349893 diff --git a/checkpoint-2500/random_states_0.pkl b/checkpoint-2500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..3c7eb20e0bda22ea1aaead2f98c83048e6d7dd06 --- /dev/null +++ b/checkpoint-2500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:cdd489c1d119caa585ef3d054394450822d1ef315c33e481176b55db47d4fce2 +size 14599 diff --git a/checkpoint-2500/scaler.pt b/checkpoint-2500/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..9a3e6e139e26b18feddbe2f66ba98b73890dacfa --- /dev/null +++ b/checkpoint-2500/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:0fbcebc8f5487b0c117b5dd47f2ea304af3eebf408d297118d9307e1223927e1 +size 557 diff --git a/checkpoint-2500/scheduler.bin b/checkpoint-2500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..0d807ee84c824a9678bf408d92ec980af8d26690 --- /dev/null +++ b/checkpoint-2500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:8ad9051d458b8f5dd4f53e95e49d24070f8c7b8cb8c5e154c42b8db78b18ea96 +size 563 diff --git a/checkpoint-2500/unet/config.json b/checkpoint-2500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..99ebec7e97d8731e19f09a61fd292f2b02845211 --- /dev/null +++ b/checkpoint-2500/unet/config.json @@ -0,0 +1,40 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "add_attention": true, + "attention_head_dim": 8, + "block_out_channels": [ + 64, + 64, + 128, + 128 + ], + "center_input_sample": false, + "class_embed_type": null, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "AttnDownBlock2D" + ], + "downsample_padding": 1, + "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": 256, + "time_embedding_type": "positional", + "up_block_types": [ + "AttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ] +} diff --git a/checkpoint-2500/unet/diffusion_pytorch_model.bin b/checkpoint-2500/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..9eca64e7bf4e0dcec8652534302944148e153ea9 --- /dev/null +++ b/checkpoint-2500/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:9f97d50f2555add0c0c81345dfa1e1a17d6566461f9e5ffea7b0c950dab24da8 +size 28664529 diff --git a/checkpoint-2500/unet_ema/config.json b/checkpoint-2500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..924073f12aa1cc04e33a6106288495203fa809f0 --- /dev/null +++ b/checkpoint-2500/unet_ema/config.json @@ -0,0 +1,47 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "add_attention": true, + "attention_head_dim": 8, + "block_out_channels": [ + 64, + 64, + 128, + 128 + ], + "center_input_sample": false, + "class_embed_type": null, + "decay": 0.9999, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "AttnDownBlock2D" + ], + "downsample_padding": 1, + "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": 2500, + "out_channels": 3, + "power": 0.75, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_embedding_type": "positional", + "up_block_types": [ + "AttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "update_after_step": 0, + "use_ema_warmup": true +} diff --git a/checkpoint-2500/unet_ema/diffusion_pytorch_model.bin b/checkpoint-2500/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..5d1e511278a88c4350709f6bf4b06e42b2b0c290 --- /dev/null +++ b/checkpoint-2500/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:37f50e52d9b991efe74ca656fac3482371b547ab2100165937d12cc0434eb94b +size 28659921 diff --git a/checkpoint-3000/optimizer.bin b/checkpoint-3000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..5948402da1e698d9c4ce26e92360a7784ea4b6d2 --- /dev/null +++ b/checkpoint-3000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:2ae1c7b32669d46450187e475f72afb30f73d0c744cc44c5e6efe201145a6ec2 +size 57349893 diff --git a/checkpoint-3000/random_states_0.pkl b/checkpoint-3000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..9f5ac0e392325c5698365dfc41eb12f3cbcfd175 --- /dev/null +++ b/checkpoint-3000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:fa413cbdaaca415106f5fb388092c43818464ecff4df802cf4d6e4376ad4aa6e +size 14599 diff --git a/checkpoint-3000/scaler.pt b/checkpoint-3000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..2888c9ddc2c30e6b47ce0e6abcc8e2d1303aab00 --- /dev/null +++ b/checkpoint-3000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:fb1f9398b77268202e8e1465734a63d123b1ef11c27f20f2473677e9883a6869 +size 557 diff --git a/checkpoint-3000/scheduler.bin b/checkpoint-3000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..cc6efc64e8f3cc68e22cdb229488b9c33d5f8361 --- /dev/null +++ b/checkpoint-3000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e3041709dacccb9a76ff9a1c838cb6b3e20a766059ec16220100860d7429ea43 +size 563 diff --git a/checkpoint-3000/unet/config.json b/checkpoint-3000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..99ebec7e97d8731e19f09a61fd292f2b02845211 --- /dev/null +++ b/checkpoint-3000/unet/config.json @@ -0,0 +1,40 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "add_attention": true, + "attention_head_dim": 8, + "block_out_channels": [ + 64, + 64, + 128, + 128 + ], + "center_input_sample": false, + "class_embed_type": null, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "AttnDownBlock2D" + ], + "downsample_padding": 1, + "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": 256, + "time_embedding_type": "positional", + "up_block_types": [ + "AttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ] +} diff --git a/checkpoint-3000/unet/diffusion_pytorch_model.bin b/checkpoint-3000/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..9a5a51f811f7568fe63f8379e0ed2ad56e7a16f7 --- /dev/null +++ b/checkpoint-3000/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:9d9094780ae7861708b0e479d72a4942fa4f6cc40a90919f6dca2584b3695117 +size 28664529 diff --git a/checkpoint-3000/unet_ema/config.json b/checkpoint-3000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..4dd0a59d17c7ba6d67d6d6b0a511896c8126a4bc --- /dev/null +++ b/checkpoint-3000/unet_ema/config.json @@ -0,0 +1,47 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "add_attention": true, + "attention_head_dim": 8, + "block_out_channels": [ + 64, + 64, + 128, + 128 + ], + "center_input_sample": false, + "class_embed_type": null, + "decay": 0.9999, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "AttnDownBlock2D" + ], + "downsample_padding": 1, + "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": 3000, + "out_channels": 3, + "power": 0.75, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_embedding_type": "positional", + "up_block_types": [ + "AttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "update_after_step": 0, + "use_ema_warmup": true +} diff --git a/checkpoint-3000/unet_ema/diffusion_pytorch_model.bin b/checkpoint-3000/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..49a0eb977d533531c8d293a3e2d6d745d939122e --- /dev/null +++ b/checkpoint-3000/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:0d729de9a5d30e3d5c5c267e170ac1e79867fa754af78bf74b53fede74c4a1d2 +size 28659921 diff --git a/checkpoint-500/optimizer.bin b/checkpoint-500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..ef33749dc8a3b2b4d66bd0cca2a39e8f914167cf --- /dev/null +++ b/checkpoint-500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:8898c675565e78977c9a6705043a6652fdc2274103294c1a9fb9d33bc212a4a7 +size 57349893 diff --git a/checkpoint-500/random_states_0.pkl b/checkpoint-500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..35c63ac4f21d443d69f108313fe23725a18a09d7 --- /dev/null +++ b/checkpoint-500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e13ed61612ddb11e52a1a2b2609ef6285f41b5d7488a4ba210691fb441af1496 +size 14599 diff --git a/checkpoint-500/scaler.pt b/checkpoint-500/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..b5f2e5ee77c2cef5eff816170558ce9209855685 --- /dev/null +++ b/checkpoint-500/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:a3f196a54202bb4ba1220e8c59f42f9cda0702d68ea83147d814c2fb2f36b8f2 +size 557 diff --git a/checkpoint-500/scheduler.bin b/checkpoint-500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..a547e968db171c97dcae1ef98b0109041a3e6600 --- /dev/null +++ b/checkpoint-500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:caa855a58bc71a6fa89384f52cf73f72b07ee47729f7813ad04419a2d25fe313 +size 563 diff --git a/checkpoint-500/unet/config.json b/checkpoint-500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..99ebec7e97d8731e19f09a61fd292f2b02845211 --- /dev/null +++ b/checkpoint-500/unet/config.json @@ -0,0 +1,40 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "add_attention": true, + "attention_head_dim": 8, + "block_out_channels": [ + 64, + 64, + 128, + 128 + ], + "center_input_sample": false, + "class_embed_type": null, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "AttnDownBlock2D" + ], + "downsample_padding": 1, + "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": 256, + "time_embedding_type": "positional", + "up_block_types": [ + "AttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ] +} diff --git a/checkpoint-500/unet/diffusion_pytorch_model.bin b/checkpoint-500/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..635e04e759531c35b6796ad909a500dc7506594b --- /dev/null +++ b/checkpoint-500/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c8a47caed9f4a872ae695036aa213acf391cd75bb7999c619aafb17f83d7d7fa +size 28664529 diff --git a/checkpoint-500/unet_ema/config.json b/checkpoint-500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..345ceeff249bf4ed094871f75ef6da85c4d3d55d --- /dev/null +++ b/checkpoint-500/unet_ema/config.json @@ -0,0 +1,47 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "add_attention": true, + "attention_head_dim": 8, + "block_out_channels": [ + 64, + 64, + 128, + 128 + ], + "center_input_sample": false, + "class_embed_type": null, + "decay": 0.9999, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "AttnDownBlock2D" + ], + "downsample_padding": 1, + "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": 500, + "out_channels": 3, + "power": 0.75, + "resnet_time_scale_shift": "default", + "sample_size": 256, + "time_embedding_type": "positional", + "up_block_types": [ + "AttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "update_after_step": 0, + "use_ema_warmup": true +} diff --git a/checkpoint-500/unet_ema/diffusion_pytorch_model.bin b/checkpoint-500/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..79f121ec90e25011ec14b8165ec1a87dcae34616 --- /dev/null +++ b/checkpoint-500/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:188e8a283d8bd859797f836874f76a4c61e83678e7c8a8e2adadb91186ef696b +size 28659921 diff --git a/model_index.json b/model_index.json new file mode 100644 index 0000000000000000000000000000000000000000..89361bf58519544e36a6c57ed097ea6aeb9b250b --- /dev/null +++ b/model_index.json @@ -0,0 +1,12 @@ +{ + "_class_name": "DDPMPipeline", + "_diffusers_version": "0.16.1", + "scheduler": [ + "diffusers", + "DDPMScheduler" + ], + "unet": [ + "diffusers", + "UNet2DModel" + ] +} diff --git a/scheduler/scheduler_config.json b/scheduler/scheduler_config.json new file mode 100644 index 0000000000000000000000000000000000000000..b7485e2539b46708b51281a0d29449fc08a32b61 --- /dev/null +++ b/scheduler/scheduler_config.json @@ -0,0 +1,16 @@ +{ + "_class_name": "DDPMScheduler", + "_diffusers_version": "0.16.1", + "beta_end": 0.02, + "beta_schedule": "linear", + "beta_start": 0.0001, + "clip_sample": true, + "clip_sample_range": 1.0, + "dynamic_thresholding_ratio": 0.995, + "num_train_timesteps": 1000, + "prediction_type": "epsilon", + "sample_max_value": 1.0, + "thresholding": false, + "trained_betas": null, + "variance_type": "fixed_small" +} diff --git a/unet/config.json b/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..99ebec7e97d8731e19f09a61fd292f2b02845211 --- /dev/null +++ b/unet/config.json @@ -0,0 +1,40 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.16.1", + "act_fn": "silu", + "add_attention": true, + "attention_head_dim": 8, + "block_out_channels": [ + 64, + 64, + 128, + 128 + ], + "center_input_sample": false, + "class_embed_type": null, + "down_block_types": [ + "DownBlock2D", + "DownBlock2D", + "DownBlock2D", + "AttnDownBlock2D" + ], + "downsample_padding": 1, + "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": 256, + "time_embedding_type": "positional", + "up_block_types": [ + "AttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ] +} diff --git a/unet/diffusion_pytorch_model.bin b/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..ab290db3936c1d4d9e9fe61185098277a33f6d68 --- /dev/null +++ b/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:0df3c8a6c84a0db7ad7898d6446e5cd05c68405d54065e15bd992c29ac273d3d +size 28664529