diff --git a/checkpoint-1000/optimizer.bin b/checkpoint-1000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..4565980e1621902b7665c477d26f0eee2d11a92c --- /dev/null +++ b/checkpoint-1000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:55dd041e098bf1662537521ee8ae6e739be470ee70c48215c1a625a0e46ecd4d +size 909763736 diff --git a/checkpoint-1000/random_states_0.pkl b/checkpoint-1000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..d4b2550ae37771928e9c6b462dcfe1b145f270db --- /dev/null +++ b/checkpoint-1000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:6ae49d5c00bdd0152a46d50c5685497796de4507aeeb9c40cf11e75848e7832d +size 14344 diff --git a/checkpoint-1000/scheduler.bin b/checkpoint-1000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..2c58b4305488b7a0416e65d6abcc70c413a72819 --- /dev/null +++ b/checkpoint-1000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:8335f33cf06b56dcfbc64c7f56367a00b7aeb4ff84848bd20b9f72327b549dfb +size 1000 diff --git a/checkpoint-1000/unet/config.json b/checkpoint-1000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..52d7f3f5ea6f95075411d0afdc62f354fb8fba95 --- /dev/null +++ b/checkpoint-1000/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": 64, + "time_embedding_type": "positional", + "up_block_types": [ + "UpBlock2D", + "AttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upsample_type": "conv" +} diff --git a/checkpoint-1000/unet/diffusion_pytorch_model.safetensors b/checkpoint-1000/unet/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..967729fd4ee675eb80f2d838f8f1c66a4ef9866a --- /dev/null +++ b/checkpoint-1000/unet/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7d747f587bfc327b216933b5964e93b69ec0a2612d423cdae5a03a4c85adf7e3 +size 454741108 diff --git a/checkpoint-1000/unet_ema/config.json b/checkpoint-1000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..b646c63114142566cad20b95dd3bf67666a05591 --- /dev/null +++ b/checkpoint-1000/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": 1000, + "out_channels": 3, + "power": 0.75, + "resnet_time_scale_shift": "default", + "sample_size": 64, + "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-1000/unet_ema/diffusion_pytorch_model.safetensors b/checkpoint-1000/unet_ema/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..6c972b7ed1dc6e5556656384f2cd1ff82767c232 --- /dev/null +++ b/checkpoint-1000/unet_ema/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:155fce439932267063773b5ea250fddc5c328bb2e134fb867e38df9ec896cfff +size 454741108 diff --git a/checkpoint-1500/optimizer.bin b/checkpoint-1500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..027ac4cc4e90759e08557a0feb299fabc5568e7a --- /dev/null +++ b/checkpoint-1500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:f18020cbf4740c8d5fcf33c2d40403417d4adc2b393f4c0f28165d7fb7c4974e +size 909763736 diff --git a/checkpoint-1500/random_states_0.pkl b/checkpoint-1500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..992e57c2a39d865f331b433fe34a24d2bdeae4ed --- /dev/null +++ b/checkpoint-1500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:2007bfe3a4d92a3b81f9ed0b24f77c52130e9c68c1646a4f43a23aa981541e48 +size 14344 diff --git a/checkpoint-1500/scheduler.bin b/checkpoint-1500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..030fee87ad82be3daac05dca6c25ac779fca68d6 --- /dev/null +++ b/checkpoint-1500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:6fe145278f61f8087504f18ab7d101bb28cf2f51bf01a389ee45eee3225db42a +size 1000 diff --git a/checkpoint-1500/unet/config.json b/checkpoint-1500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..52d7f3f5ea6f95075411d0afdc62f354fb8fba95 --- /dev/null +++ b/checkpoint-1500/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": 64, + "time_embedding_type": "positional", + "up_block_types": [ + "UpBlock2D", + "AttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upsample_type": "conv" +} diff --git a/checkpoint-1500/unet/diffusion_pytorch_model.safetensors b/checkpoint-1500/unet/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..7743c230feb87842ba75c553ff6f85c721be1b48 --- /dev/null +++ b/checkpoint-1500/unet/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:347b90218b295fc86059e383396383ad46083e08e0e87f5e10698c84217c3d7f +size 454741108 diff --git a/checkpoint-1500/unet_ema/config.json b/checkpoint-1500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..fce5a714dfbcad73c7c2cbe0d64c6e95b2e51567 --- /dev/null +++ b/checkpoint-1500/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": 1500, + "out_channels": 3, + "power": 0.75, + "resnet_time_scale_shift": "default", + "sample_size": 64, + "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-1500/unet_ema/diffusion_pytorch_model.safetensors b/checkpoint-1500/unet_ema/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..a023be1d598deb9833a83c738fbeeefc8f675a60 --- /dev/null +++ b/checkpoint-1500/unet_ema/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:52e89d2ed4a37275bd5ecec0ff0ec5d828f0de384e7939f116b6a3701fc30386 +size 454741108 diff --git a/checkpoint-2000/optimizer.bin b/checkpoint-2000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..7b84b2213028a5a3345d6c0dc909b7e3c479b944 --- /dev/null +++ b/checkpoint-2000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b4139965a9d73d21ad0e9e49c99056110a728e9ceff761881047086ee3d0c85f +size 909763736 diff --git a/checkpoint-2000/random_states_0.pkl b/checkpoint-2000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..b03fbab42c568d17e965483075bd8ec14272dda1 --- /dev/null +++ b/checkpoint-2000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:54c691bf2b0fc69cd5bca8420172b3276121ee72ed480ad2f7471c8f32003287 +size 14344 diff --git a/checkpoint-2000/scheduler.bin b/checkpoint-2000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..54a4e77e88967b78e914cc1b8aa3147682e02cd6 --- /dev/null +++ b/checkpoint-2000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:9142071453d8567e9278fe7a2da8cc0825882fad37a3aee8b850972659736a7b +size 1000 diff --git a/checkpoint-2000/unet/config.json b/checkpoint-2000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..52d7f3f5ea6f95075411d0afdc62f354fb8fba95 --- /dev/null +++ b/checkpoint-2000/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": 64, + "time_embedding_type": "positional", + "up_block_types": [ + "UpBlock2D", + "AttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upsample_type": "conv" +} diff --git a/checkpoint-2000/unet/diffusion_pytorch_model.safetensors b/checkpoint-2000/unet/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..0b48ba2f5562cb2ab62d78eb8312accdf3f5110e --- /dev/null +++ b/checkpoint-2000/unet/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:96bc25d696d230bb377164e64a54e7c59e886ff77c2f747e3e59b9c415b043c9 +size 454741108 diff --git a/checkpoint-2000/unet_ema/config.json b/checkpoint-2000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..a5a21bf77068e284aa463d8050626568e704c7d6 --- /dev/null +++ b/checkpoint-2000/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": 2000, + "out_channels": 3, + "power": 0.75, + "resnet_time_scale_shift": "default", + "sample_size": 64, + "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-2000/unet_ema/diffusion_pytorch_model.safetensors b/checkpoint-2000/unet_ema/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..34b150c1eb28d6976714d2d64781a5b3e85db956 --- /dev/null +++ b/checkpoint-2000/unet_ema/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:113ce445dce1bb0336a83b588bfb988656d1b1a4ad1f211b29ffea272eb7a6ce +size 454741108 diff --git a/checkpoint-2500/optimizer.bin b/checkpoint-2500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..ed63725b419ac69b92efb6c8efe5525fa01b2395 --- /dev/null +++ b/checkpoint-2500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:39f629c2044d8dee41b48c51a0cf82b57c715ee98b018d64793ae1c98104a781 +size 909763736 diff --git a/checkpoint-2500/random_states_0.pkl b/checkpoint-2500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..c5990f1d1e9ce42c237dbe44722b4c28bc7cc107 --- /dev/null +++ b/checkpoint-2500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:09d1c4e56aa3d4bbe209382120b0ea2bfd86686a7e0b3ec234061c8130d740f9 +size 14344 diff --git a/checkpoint-2500/scheduler.bin b/checkpoint-2500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..982b0a5b1a6471826688b6cf92fffe6ae5a74154 --- /dev/null +++ b/checkpoint-2500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:568b741f2d469db8ce47c019deb5a58b56fec905179b64ca0b1c2025d0370f15 +size 1000 diff --git a/checkpoint-2500/unet/config.json b/checkpoint-2500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..52d7f3f5ea6f95075411d0afdc62f354fb8fba95 --- /dev/null +++ b/checkpoint-2500/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": 64, + "time_embedding_type": "positional", + "up_block_types": [ + "UpBlock2D", + "AttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upsample_type": "conv" +} diff --git a/checkpoint-2500/unet/diffusion_pytorch_model.safetensors b/checkpoint-2500/unet/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..59cb7d1360e12392f28d11a0a0f912b52e37fb56 --- /dev/null +++ b/checkpoint-2500/unet/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:acd2616f000513b9e3d2cd5808e70b34de11351e2a07c9b5260268f72c72222a +size 454741108 diff --git a/checkpoint-2500/unet_ema/config.json b/checkpoint-2500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..b76e1f2d9d9f4c01039ed43547982f33e9b821a9 --- /dev/null +++ b/checkpoint-2500/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": 2500, + "out_channels": 3, + "power": 0.75, + "resnet_time_scale_shift": "default", + "sample_size": 64, + "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-2500/unet_ema/diffusion_pytorch_model.safetensors b/checkpoint-2500/unet_ema/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..59b6ee3b44609c69a418e13e43900589400854b1 --- /dev/null +++ b/checkpoint-2500/unet_ema/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:d07492a652f62a5742cc952ea9cc30c18fbbb74bc02371b2cddbcf47a534bcb5 +size 454741108 diff --git a/checkpoint-3000/optimizer.bin b/checkpoint-3000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..e1f0a1bc078d21cdee2286875722588922eace79 --- /dev/null +++ b/checkpoint-3000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:5ca5a834b54df0dd1e22a1654a6991b11821c0bacfc896d26c4385bd221c3667 +size 909763736 diff --git a/checkpoint-3000/random_states_0.pkl b/checkpoint-3000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..e7b02810274ee7d7c5cd41bb7149ed8a389aad9a --- /dev/null +++ b/checkpoint-3000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c9ea24059e21f9f1b370b0925e952b21f4f8691a263672a37c1e2a7d1269802d +size 14344 diff --git a/checkpoint-3000/scheduler.bin b/checkpoint-3000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..ab2270323dfca880030813ed67b55b7ff008fac3 --- /dev/null +++ b/checkpoint-3000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:76855a8a73dc7883d5aa1a604a9ff5db53977b7c6c5bb8839c2d2af7f3b5ff8d +size 1000 diff --git a/checkpoint-3000/unet/config.json b/checkpoint-3000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..52d7f3f5ea6f95075411d0afdc62f354fb8fba95 --- /dev/null +++ b/checkpoint-3000/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": 64, + "time_embedding_type": "positional", + "up_block_types": [ + "UpBlock2D", + "AttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upsample_type": "conv" +} diff --git a/checkpoint-3000/unet/diffusion_pytorch_model.safetensors b/checkpoint-3000/unet/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..091989ef6a75de62b84616e155847fec8090f59c --- /dev/null +++ b/checkpoint-3000/unet/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:082688520d2585675856cf0d58c8710597cd4b8b482b4c658b8b67310d6925cd +size 454741108 diff --git a/checkpoint-3000/unet_ema/config.json b/checkpoint-3000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..83bfe35c8809873553f88aadc60400555ed2d531 --- /dev/null +++ b/checkpoint-3000/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": 3000, + "out_channels": 3, + "power": 0.75, + "resnet_time_scale_shift": "default", + "sample_size": 64, + "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-3000/unet_ema/diffusion_pytorch_model.safetensors b/checkpoint-3000/unet_ema/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..61a505a8bf6c0f83a3cff0ba9fb93e8773b90a38 --- /dev/null +++ b/checkpoint-3000/unet_ema/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:14d8e46f708ee1de395b08f621b521d94f73aa219f63e648f9a0a9b6681965f2 +size 454741108 diff --git a/checkpoint-3500/optimizer.bin b/checkpoint-3500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..5d6a05345121d4f49da603c8aa2cb79313d7204c --- /dev/null +++ b/checkpoint-3500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7940b107c79efe014416bdea65ea1f46f082d05d5b02634ebd85a18b6fa4f513 +size 909763736 diff --git a/checkpoint-3500/random_states_0.pkl b/checkpoint-3500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..face513c7adeb27175c06cfcb86d1214d9ad7f30 --- /dev/null +++ b/checkpoint-3500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:5b8b77fb89b5fb5e5695eb77a31db8dc975356bdbccd9f621005f45421f42815 +size 14344 diff --git a/checkpoint-3500/scheduler.bin b/checkpoint-3500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..08f8841787ec560b643398f006b05d30a91264fd --- /dev/null +++ b/checkpoint-3500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:895d68d19ec72c39ab5ecce3564c2de682ad32f302ffc1105ce4dafe9b8259a2 +size 1000 diff --git a/checkpoint-3500/unet/config.json b/checkpoint-3500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..52d7f3f5ea6f95075411d0afdc62f354fb8fba95 --- /dev/null +++ b/checkpoint-3500/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": 64, + "time_embedding_type": "positional", + "up_block_types": [ + "UpBlock2D", + "AttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upsample_type": "conv" +} diff --git a/checkpoint-3500/unet/diffusion_pytorch_model.safetensors b/checkpoint-3500/unet/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..e6ee54d32e86c820bd06b740e84b39b61b7d3f6e --- /dev/null +++ b/checkpoint-3500/unet/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:740610a93fda37e9334f59665a3fedd57a8ae96b778ee8bdc3bd2d48ab9e104b +size 454741108 diff --git a/checkpoint-3500/unet_ema/config.json b/checkpoint-3500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..cd0e6857fd858c19772c0ca6ff574e580752c42f --- /dev/null +++ b/checkpoint-3500/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": 3500, + "out_channels": 3, + "power": 0.75, + "resnet_time_scale_shift": "default", + "sample_size": 64, + "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-3500/unet_ema/diffusion_pytorch_model.safetensors b/checkpoint-3500/unet_ema/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..f09588e115e4ffd5b6015025b317b8d7bde714e4 --- /dev/null +++ b/checkpoint-3500/unet_ema/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c01a70ed3b4c0eec5e571d7edc768fa220a24ba42989e1cb07d7681d69438b61 +size 454741108 diff --git a/checkpoint-4000/optimizer.bin b/checkpoint-4000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..92c2389891b39346b71003d23f4de356b9aaf7bb --- /dev/null +++ b/checkpoint-4000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b36a2a55c673438c5e07e649918926492d220d0dcde92d4e244866ccc1a96ea3 +size 909763736 diff --git a/checkpoint-4000/random_states_0.pkl b/checkpoint-4000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..65d26d98e8b1dcc669caf557e0856beb20ea6afa --- /dev/null +++ b/checkpoint-4000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:628761d4a346348e546bac6fb54f54c63849265908fec32a5d6558ffef810d0c +size 14344 diff --git a/checkpoint-4000/scheduler.bin b/checkpoint-4000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..41a8949e07f2a641622f39d02ee6d26cce581ff5 --- /dev/null +++ b/checkpoint-4000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:fddbbc36287d334149a4c13f6b42eea000ec8207ec2fd4ae3d9aea6a948c68ad +size 1000 diff --git a/checkpoint-4000/unet/config.json b/checkpoint-4000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..52d7f3f5ea6f95075411d0afdc62f354fb8fba95 --- /dev/null +++ b/checkpoint-4000/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": 64, + "time_embedding_type": "positional", + "up_block_types": [ + "UpBlock2D", + "AttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upsample_type": "conv" +} diff --git a/checkpoint-4000/unet/diffusion_pytorch_model.safetensors b/checkpoint-4000/unet/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..77789d4e2f922ee9857c5c82d07820ca979e2fb6 --- /dev/null +++ b/checkpoint-4000/unet/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c8d1183e24096d7e1d5733ffa18c123a13111b1293c2ade3d88706cf16f2c82a +size 454741108 diff --git a/checkpoint-4000/unet_ema/config.json b/checkpoint-4000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..7215f1d4d95cb5d71242c59f24234b766f48e280 --- /dev/null +++ b/checkpoint-4000/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": 4000, + "out_channels": 3, + "power": 0.75, + "resnet_time_scale_shift": "default", + "sample_size": 64, + "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-4000/unet_ema/diffusion_pytorch_model.safetensors b/checkpoint-4000/unet_ema/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..7197e81458d501c405240c46cb4d4892b4326a25 --- /dev/null +++ b/checkpoint-4000/unet_ema/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b6d4191728a1e854234830e95d23632295afadac0baf2ae0a4eaa8973c13de79 +size 454741108 diff --git a/checkpoint-500/optimizer.bin b/checkpoint-500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..d78ac496c523557d7232594e935599f7ec47ca0e --- /dev/null +++ b/checkpoint-500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:111035e9431df97342fb8d8ef7ec4a8baf5f2d7c48d545c2d742f1f515ebf5ab +size 909763736 diff --git a/checkpoint-500/random_states_0.pkl b/checkpoint-500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..83500952db1b81d424ab066648d021ee17fed545 --- /dev/null +++ b/checkpoint-500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7b32c586a76d2f5afa74c44070cc753dc64d079b2c90eb8d81a845210e66796e +size 14344 diff --git a/checkpoint-500/scheduler.bin b/checkpoint-500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..5f0c87cfc1ffdc1fa8e66ebda086067a4ae7088b --- /dev/null +++ b/checkpoint-500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:6cf34ac8dd2addc454aeb1ee4eaba3267c2841857e117be40ac2ed02cf581c9c +size 1000 diff --git a/checkpoint-500/unet/config.json b/checkpoint-500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..52d7f3f5ea6f95075411d0afdc62f354fb8fba95 --- /dev/null +++ b/checkpoint-500/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": 64, + "time_embedding_type": "positional", + "up_block_types": [ + "UpBlock2D", + "AttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upsample_type": "conv" +} diff --git a/checkpoint-500/unet/diffusion_pytorch_model.safetensors b/checkpoint-500/unet/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..b01386af22a6ff9fecc3dd60a7a9d7836733172d --- /dev/null +++ b/checkpoint-500/unet/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:1ebb16530e80d845914cc2939962422bf6c5b10e6e03659c273735ae08787356 +size 454741108 diff --git a/checkpoint-500/unet_ema/config.json b/checkpoint-500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..a8113de28e2efbbc20e7d9162deb4e4c64fa6941 --- /dev/null +++ b/checkpoint-500/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": 500, + "out_channels": 3, + "power": 0.75, + "resnet_time_scale_shift": "default", + "sample_size": 64, + "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-500/unet_ema/diffusion_pytorch_model.safetensors b/checkpoint-500/unet_ema/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..2b61bbafbaf71430819be8eb225fc7b19810ec4e --- /dev/null +++ b/checkpoint-500/unet_ema/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:bccb2b49fd6f3488fb80b8410a5b3711a6f53fbab8df040eb27bc3e675160906 +size 454741108 diff --git a/logs/train_unconditional/events.out.tfevents.1698500688.b5d491bd28f5.34273.0 b/logs/train_unconditional/events.out.tfevents.1698500688.b5d491bd28f5.34273.0 new file mode 100644 index 0000000000000000000000000000000000000000..03e205939d6bf8e759ce19f1f614270b8158e733 --- /dev/null +++ b/logs/train_unconditional/events.out.tfevents.1698500688.b5d491bd28f5.34273.0 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:6f890e686843fa8ef6575e8d9622f306508a5ef5ff255590de74a94945460164 +size 900892 diff --git a/model_index.json b/model_index.json new file mode 100644 index 0000000000000000000000000000000000000000..bed2cc75f08178345406ca505681a83fed02fc98 --- /dev/null +++ b/model_index.json @@ -0,0 +1,12 @@ +{ + "_class_name": "DDPMPipeline", + "_diffusers_version": "0.22.0.dev0", + "scheduler": [ + "diffusers", + "DDPMScheduler" + ], + "unet": [ + "diffusers", + "UNet2DModel" + ] +} diff --git a/scheduler/scheduler_config.json b/scheduler/scheduler_config.json new file mode 100644 index 0000000000000000000000000000000000000000..05b73a21451c96ef3b3f9320cb4e2d7a6e086765 --- /dev/null +++ b/scheduler/scheduler_config.json @@ -0,0 +1,18 @@ +{ + "_class_name": "DDPMScheduler", + "_diffusers_version": "0.22.0.dev0", + "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, + "steps_offset": 0, + "thresholding": false, + "timestep_spacing": "leading", + "trained_betas": null, + "variance_type": "fixed_small" +} diff --git a/unet/config.json b/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..52d7f3f5ea6f95075411d0afdc62f354fb8fba95 --- /dev/null +++ b/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": 64, + "time_embedding_type": "positional", + "up_block_types": [ + "UpBlock2D", + "AttnUpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upsample_type": "conv" +} diff --git a/unet/diffusion_pytorch_model.safetensors b/unet/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..2099c2e020b770fb01e7c16b0112d1abd432f995 --- /dev/null +++ b/unet/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b7ebfa3d343506b8783d22c38b454022369963591dee7e496c08934002a79cd0 +size 454741108