diff --git a/checkpoint-1000/optimizer.bin b/checkpoint-1000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..b5a2fcd7311444ee4f2a0e12545bd78f4bb67d83 --- /dev/null +++ b/checkpoint-1000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:33a86ad6cdfeb281f3462c7e0ea94d2a122a4042e7fbc1677c44fe183901e0d8 +size 909763736 diff --git a/checkpoint-1000/random_states_0.pkl b/checkpoint-1000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..e3e1a5a4b7323e044fd92666c2b2469ccf3e5a1b --- /dev/null +++ b/checkpoint-1000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:0de04b78d9d1c1c4772b11c999f7e819d8cecbf535e489a9ce5062517c26004a +size 14408 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..6ce14893256850184859018f2ea199bd5875f61d --- /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": 32, + "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..bb4b742802357229b1914315d521023e856128bf --- /dev/null +++ b/checkpoint-1000/unet/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:927c9d736bf00c002dcf8d610e9f85abf77fc7dff0cde5c75e29fc995125aea4 +size 454741108 diff --git a/checkpoint-1000/unet_ema/config.json b/checkpoint-1000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..8954f31d9608da2a8b40376a5c07726853452e2c --- /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": 32, + "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..fb295a4b54e0b106e7de2cae986def409a53cefd --- /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:9ecd0c06667382662ddc9849d509c5d82dc4e9a434ed6058e652dc864e20a530 +size 454741108 diff --git a/checkpoint-1500/optimizer.bin b/checkpoint-1500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..7a38a9d64ffeb3b5f0303a9a62e289f278968fbd --- /dev/null +++ b/checkpoint-1500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:f09eb9dfcfc8b0247e559cd27a7fab92b93ac9d73dd3c8bd06f7f2f429a44f60 +size 909763736 diff --git a/checkpoint-1500/random_states_0.pkl b/checkpoint-1500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..e2ebd3fc87a47ec4361a23f5e1db7d37c1750589 --- /dev/null +++ b/checkpoint-1500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7eed35f2b15c33418d10e7b0021e80967aff16acaa8af8aa95539603507dae9a +size 14408 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..6ce14893256850184859018f2ea199bd5875f61d --- /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": 32, + "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..d43eb71c46a83ad65fefdb889228b2f473f5be30 --- /dev/null +++ b/checkpoint-1500/unet/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:19777e199f57930f0e2149525a7de761959e7df2dc18441d98ddcb200a323452 +size 454741108 diff --git a/checkpoint-1500/unet_ema/config.json b/checkpoint-1500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..271d7ad886fa6ffb569d61c3021e967420255f37 --- /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": 32, + "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..d613766b7a65a09b7b2842951c94b65e0ec93e05 --- /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:b8c1d1129dcbd57c5135172818f42aac101e88b08e1e5489feddb491a7ab8d3c +size 454741108 diff --git a/checkpoint-2000/optimizer.bin b/checkpoint-2000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..aea0cdd89cf71136f21864aec8a36098c359f041 --- /dev/null +++ b/checkpoint-2000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:2da599656dc36f7128e3015a98b025a5c074422b65bbeacadf265cd6cd5b972c +size 909763736 diff --git a/checkpoint-2000/random_states_0.pkl b/checkpoint-2000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..47f6c3645dd3b24b56e319b85726c0911db1cc3c --- /dev/null +++ b/checkpoint-2000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:0d53518edbf657588f093219dd64774cda54bb10dc0bef9ba983c0b48064afd4 +size 14408 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..6ce14893256850184859018f2ea199bd5875f61d --- /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": 32, + "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..af734b250f52a832019e18dc8357db1fe589fbed --- /dev/null +++ b/checkpoint-2000/unet/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:35e782c216388c872d332f85d9aef33623c678161460fc77778794db492bd386 +size 454741108 diff --git a/checkpoint-2000/unet_ema/config.json b/checkpoint-2000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..fb282961f576905fdd2704565438e96ff7cf19a6 --- /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": 32, + "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..2f63c49dc773c12259aeef9ac9b2e381760beee3 --- /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:5a8f4726ec2a99c3b91da15f3b9b727623b747524850c1da3d41424ffa618db1 +size 454741108 diff --git a/checkpoint-2500/optimizer.bin b/checkpoint-2500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..edefb9c6c2258b60983f6d67f1ba46685132d951 --- /dev/null +++ b/checkpoint-2500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:9050c3cc2c181f166218e78876ab004dbdd1b7e2b895d666159b01b56788d0ba +size 909763736 diff --git a/checkpoint-2500/random_states_0.pkl b/checkpoint-2500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..89cde9cfcc6252da20ccc1894c5062d1e070f45e --- /dev/null +++ b/checkpoint-2500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:5492287d69e1f456c23252c14ab7698efe416594d1dbc382fd5d6d388ff463d2 +size 14408 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..6ce14893256850184859018f2ea199bd5875f61d --- /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": 32, + "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..11b35686466be4fbdea6eacdf8a7b7fbe9af8cb7 --- /dev/null +++ b/checkpoint-2500/unet/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:dce0628e3e13354c653df2e592fa442b7610306cd13076368d8dc8ed8d696ec1 +size 454741108 diff --git a/checkpoint-2500/unet_ema/config.json b/checkpoint-2500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..6673f5ead3dbe77610b6cd4f0b3361fada9d2f3c --- /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": 32, + "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..e1646827da16ee6be8fcb8aebdf442cc5f6e2f1d --- /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:606d7ac250f9f5abece6cb074ca71de3109ddfbf4041afb2bc561f8367e5ea4f +size 454741108 diff --git a/checkpoint-3000/optimizer.bin b/checkpoint-3000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..33d1cda40006d74857d1b2186260db20166c5aab --- /dev/null +++ b/checkpoint-3000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:69f7872a6f539e43b1f1158cf232eb675776d5390d1f7b4a793eb15d9faa4162 +size 909763736 diff --git a/checkpoint-3000/random_states_0.pkl b/checkpoint-3000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..91196174bdd4c0611231203f4242146de171e30e --- /dev/null +++ b/checkpoint-3000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:df10159ed1049762233b4ef61a3dbdb4e5839a39a946189ae9d2fecc88d622d7 +size 14408 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..6ce14893256850184859018f2ea199bd5875f61d --- /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": 32, + "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..bb58fce921313d4df415490217d8d46f187ec3a9 --- /dev/null +++ b/checkpoint-3000/unet/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:bb57c1b359341bc6ddc72ab52ec155ded2fc174a93074e38910111d7db6ec742 +size 454741108 diff --git a/checkpoint-3000/unet_ema/config.json b/checkpoint-3000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..e2c5f58738b02947fb43ed9c3fa98c8954c38b2f --- /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": 32, + "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..29dfc2690cffd97e68564ff6d7c2f31fcdef003f --- /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:28b5a09b368d1d4a622d7f00d4d3168f549762a1935d7d8fed59c6d25cab82fd +size 454741108 diff --git a/checkpoint-3500/optimizer.bin b/checkpoint-3500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..ffeb8b6fe1090ed4340f47e9a4a84e289c3dd380 --- /dev/null +++ b/checkpoint-3500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:474c5b388c37883f945d3b129e8ccbb94e58a840b21bfe33865b4a270c85f7d8 +size 909763736 diff --git a/checkpoint-3500/random_states_0.pkl b/checkpoint-3500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..eb57808c4396a22db2ea4169726c11e56caa86b1 --- /dev/null +++ b/checkpoint-3500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:1c8157a34fa31604f32fa4b26f5f26554d6db2b75c9e471d63aa8578884d193f +size 14408 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..6ce14893256850184859018f2ea199bd5875f61d --- /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": 32, + "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..6f8f7fd25aa0a65f8110b2988ea3eaec93ece706 --- /dev/null +++ b/checkpoint-3500/unet/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:a8c62cf0a6911d62b647fa2e5e3ff11fc6f93325136f3a2faabec5e7684013f1 +size 454741108 diff --git a/checkpoint-3500/unet_ema/config.json b/checkpoint-3500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..f0563c02d13d40a1d58fc436c3c9b18951c174bf --- /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": 32, + "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..089b63646b07368b23505ab773260687c3cff933 --- /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:444c10f79ea69cef2af63e36f618ebf6a480c1a103eb880699069e7ddee45907 +size 454741108 diff --git a/checkpoint-4000/optimizer.bin b/checkpoint-4000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..859f3acae646bf24751d67936ac445a135ab40f3 --- /dev/null +++ b/checkpoint-4000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:1e36a7ead0917e7924941f3e5e01e595fe06c1510ccd3e99b315ad6dbb7dd41f +size 909763736 diff --git a/checkpoint-4000/random_states_0.pkl b/checkpoint-4000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..b7f724a2391093408584bc7c57fb64b03742678e --- /dev/null +++ b/checkpoint-4000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:216f9823b64b956a24448f1135f0f1671bb5db12adb15f43a6935e0edd6db3fa +size 14408 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..6ce14893256850184859018f2ea199bd5875f61d --- /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": 32, + "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..7949744132b05797b29b2c3066c48ace9fa560e6 --- /dev/null +++ b/checkpoint-4000/unet/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:0b3ff002afce73bbc2d30fdf32a81ef508cd6b02bbab2e44db4be967419c15ad +size 454741108 diff --git a/checkpoint-4000/unet_ema/config.json b/checkpoint-4000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..dd31a21bbb0a06d858a9833c93eeaa86e83a2129 --- /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": 32, + "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..cc0c4d2d9beb13abbfd2667f3b84f5d2bae02a42 --- /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:9e965f1d377b5d24161e32865ee57256dde9a4a84fa0b80f5cad2eded413b682 +size 454741108 diff --git a/checkpoint-500/optimizer.bin b/checkpoint-500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..8e9ec453d706e03fa4f1ffa65a8754c51a001a8a --- /dev/null +++ b/checkpoint-500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:ad589cdb4de08a451ff880a3fe06bfed1ba54c03278d67e11a3cb2ea87cb9e27 +size 909763736 diff --git a/checkpoint-500/random_states_0.pkl b/checkpoint-500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..78622b167008982b6ccd90725a8c64be5e27f717 --- /dev/null +++ b/checkpoint-500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:399d097fb3932a2ada5fbf0a1e65a1bc2339812f2912386796df85bb72d5fcd9 +size 14408 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..6ce14893256850184859018f2ea199bd5875f61d --- /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": 32, + "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..06ce2e72bb52efe4167ba298fee967f4062796fa --- /dev/null +++ b/checkpoint-500/unet/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e8602bc6823c25d2f0089067f7ab8b9b297362efa5ec37a21eafd9d4908c8e35 +size 454741108 diff --git a/checkpoint-500/unet_ema/config.json b/checkpoint-500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..11f78976c37f6f825564469046540cc5832d32e8 --- /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": 32, + "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..4b5a83456d9990db4def54730195d2f69ca4cf5e --- /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:872304e0767243563da026251d70853c77b72f3c51e4a3c4a8cbe58ce0407716 +size 454741108 diff --git a/logs/train_unconditional/events.out.tfevents.1698526105.6a163385c961.1406.0 b/logs/train_unconditional/events.out.tfevents.1698526105.6a163385c961.1406.0 new file mode 100644 index 0000000000000000000000000000000000000000..8490adbe12b4b94ccada64a8296338a231070c95 --- /dev/null +++ b/logs/train_unconditional/events.out.tfevents.1698526105.6a163385c961.1406.0 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:d6c1859fb60887690f67aea7baf5dbecdb7a31e0ad592a07b3234db274a124d7 +size 802814 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..6ce14893256850184859018f2ea199bd5875f61d --- /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": 32, + "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..fc9c7b1480dbbbc5b7bd6244a67389b848d024ed --- /dev/null +++ b/unet/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:973f5d6603017a62be4671c4de3c069b210f563597715ef653ec0c703d781e59 +size 454741108