diff --git a/checkpoint-16000/optimizer.bin b/checkpoint-16000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..419e63900eada9e4b6b2ce50ed114664cd78c810 --- /dev/null +++ b/checkpoint-16000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:70fc20fc86c82879c0278de5d387fe17aede71e9b557fc680eb8a689ca9e0280 +size 909763736 diff --git a/checkpoint-16000/random_states_0.pkl b/checkpoint-16000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..b98eafaa71242842afe709fba46eb2df05573f82 --- /dev/null +++ b/checkpoint-16000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:cf3f68d5d8334f6f0ab55ed5fa14193d99cfcea71bbb193a95e8120dfe5b052e +size 14344 diff --git a/checkpoint-16000/scheduler.bin b/checkpoint-16000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..43470b7ad9d8b2aa2afd547596d6f02ec2988efa --- /dev/null +++ b/checkpoint-16000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:ef531c78c9f4a4409e89e760e5dfe9d5bcf11761aff5480c9eb9ab245f99bdf0 +size 1000 diff --git a/checkpoint-16000/unet/config.json b/checkpoint-16000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..ea03432831f52ebc1b35187f19dc4707b857d57d --- /dev/null +++ b/checkpoint-16000/unet/config.json @@ -0,0 +1,51 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.28.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, + "num_train_timesteps": 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-16000/unet/diffusion_pytorch_model.safetensors b/checkpoint-16000/unet/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..3abcf189816c05fd0bd7eef2b2aa479dcdad7fd5 --- /dev/null +++ b/checkpoint-16000/unet/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:5a6af9b75fe7cb7a18c0effdc51bf98c9501f992d9a8e86e97338c4ddf4bfae9 +size 454741108 diff --git a/checkpoint-16000/unet_ema/config.json b/checkpoint-16000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..860a2d35ec956ab48755348dda0a233211735f6e --- /dev/null +++ b/checkpoint-16000/unet_ema/config.json @@ -0,0 +1,58 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.28.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, + "num_train_timesteps": null, + "optimization_step": 16000, + "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-16000/unet_ema/diffusion_pytorch_model.safetensors b/checkpoint-16000/unet_ema/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..94dc2756d2aae912b3e0cd062829b7e252f9db76 --- /dev/null +++ b/checkpoint-16000/unet_ema/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:0567e4b2ac2fcb6ef2a8d640f0c2540d53d58ce4e1302419e6117f2cbbc1d8fa +size 454741108 diff --git a/checkpoint-16500/optimizer.bin b/checkpoint-16500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..6f754c04a16685c66b8b72f11e228d14e9d927c7 --- /dev/null +++ b/checkpoint-16500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b468afd64f9e6fcaeff0feb9475c2d59872e1747d5e65ae7bc3511855208ae55 +size 909763736 diff --git a/checkpoint-16500/random_states_0.pkl b/checkpoint-16500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..b6815ddf7ac0314e8da6dc948478210c66e54e6b --- /dev/null +++ b/checkpoint-16500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:a68bf4c1f0ca9f13b1f8d1dda39a8a5b5272782f947693b1e4324406cf168151 +size 14344 diff --git a/checkpoint-16500/scheduler.bin b/checkpoint-16500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..6a60963b1836c86ad77f2b0e12e364ab869ccbdb --- /dev/null +++ b/checkpoint-16500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:1203daaba352f67582930eb8212a3a92a6f84157377d790ce8a4401551843802 +size 1000 diff --git a/checkpoint-16500/unet/config.json b/checkpoint-16500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..ea03432831f52ebc1b35187f19dc4707b857d57d --- /dev/null +++ b/checkpoint-16500/unet/config.json @@ -0,0 +1,51 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.28.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, + "num_train_timesteps": 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-16500/unet/diffusion_pytorch_model.safetensors b/checkpoint-16500/unet/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..90afacdd3bd13510d4bf56a53455a3a2f7c45f34 --- /dev/null +++ b/checkpoint-16500/unet/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:9fecc8d75caf6797bd24cb662cdabf5591c1371871eee1d2351bc2e6c1064f28 +size 454741108 diff --git a/checkpoint-16500/unet_ema/config.json b/checkpoint-16500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..195a3d1b00f5998844d62bafe591e441db6c04c7 --- /dev/null +++ b/checkpoint-16500/unet_ema/config.json @@ -0,0 +1,58 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.28.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, + "num_train_timesteps": null, + "optimization_step": 16500, + "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-16500/unet_ema/diffusion_pytorch_model.safetensors b/checkpoint-16500/unet_ema/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..638bdbfc902768d066c0bf48073afeff5d3d1178 --- /dev/null +++ b/checkpoint-16500/unet_ema/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:d06ae8255b50b6b2fb4c73d33996cdec72d513446812763d41394faa4267d293 +size 454741108 diff --git a/checkpoint-17000/optimizer.bin b/checkpoint-17000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..1d1f5069f17846fdbe4dae07e08686f8b021033d --- /dev/null +++ b/checkpoint-17000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:01bd289f12ea8b24ff02f2fa8ebe2e858294d28a376a3dcf75604b30c4c2b539 +size 909763736 diff --git a/checkpoint-17000/random_states_0.pkl b/checkpoint-17000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..d74ea4e2ddb733a2f86f193adc797cbc163899cb --- /dev/null +++ b/checkpoint-17000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:d14fa6d8e56b922c6144103ef75543334433e4049c06cefe5d934a5a32c498e5 +size 14344 diff --git a/checkpoint-17000/scheduler.bin b/checkpoint-17000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..2b4eb4e312fb52167433f93b27693a1a0463324d --- /dev/null +++ b/checkpoint-17000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:f0ce4b22204edf749adba59113d4aadc28281fa5a9e9e6fadd51d5b986375f2d +size 1000 diff --git a/checkpoint-17000/unet/config.json b/checkpoint-17000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..ea03432831f52ebc1b35187f19dc4707b857d57d --- /dev/null +++ b/checkpoint-17000/unet/config.json @@ -0,0 +1,51 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.28.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, + "num_train_timesteps": 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-17000/unet/diffusion_pytorch_model.safetensors b/checkpoint-17000/unet/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..777d73bd2dc765153b53b709744d817d6bfea92b --- /dev/null +++ b/checkpoint-17000/unet/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:a3776edaf56ab7f29f3b3014e1670e3f2bcf2b93f77cf367f1e1c3a62732c98c +size 454741108 diff --git a/checkpoint-17000/unet_ema/config.json b/checkpoint-17000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..f36f3a1a9a5cf81cc1c1045a374e1305fe2c738b --- /dev/null +++ b/checkpoint-17000/unet_ema/config.json @@ -0,0 +1,58 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.28.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, + "num_train_timesteps": null, + "optimization_step": 17000, + "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-17000/unet_ema/diffusion_pytorch_model.safetensors b/checkpoint-17000/unet_ema/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..d2e98d0f6ff5b7c646bec77e43882f7b9d81f0bd --- /dev/null +++ b/checkpoint-17000/unet_ema/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:77dda003140446579084c7db90295f9903f34ef24bc127bd10f5513e70240d5c +size 454741108 diff --git a/checkpoint-17500/optimizer.bin b/checkpoint-17500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..7bac40afa4f29fbbdf96e1b5de3031449143d073 --- /dev/null +++ b/checkpoint-17500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:a11c3ea4f7a0b01b8152e1412e55b00e4decb4f9475aaed4233930c9e9934670 +size 909763736 diff --git a/checkpoint-17500/random_states_0.pkl b/checkpoint-17500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..061b3284c9068dc8be07382c457554be7dbbd2de --- /dev/null +++ b/checkpoint-17500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:72e25a3438ca510d37fa49d2f98041fa386b4d5ee13897075cdd3aa45276af8f +size 14344 diff --git a/checkpoint-17500/scheduler.bin b/checkpoint-17500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..79450498891455fa3b642c00be172b2a96bb59e8 --- /dev/null +++ b/checkpoint-17500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c2c600e247146fca3c3c75091ec4de22b8f6022f326c7471a494b25e0b12545e +size 1000 diff --git a/checkpoint-17500/unet/config.json b/checkpoint-17500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..ea03432831f52ebc1b35187f19dc4707b857d57d --- /dev/null +++ b/checkpoint-17500/unet/config.json @@ -0,0 +1,51 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.28.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, + "num_train_timesteps": 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-17500/unet/diffusion_pytorch_model.safetensors b/checkpoint-17500/unet/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..c05c37c1036526067885d9bd4854206539c2756d --- /dev/null +++ b/checkpoint-17500/unet/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:d25057cc54fec849ec009f145609709b5d13efa1b22eecfac230f68c9434039d +size 454741108 diff --git a/checkpoint-17500/unet_ema/config.json b/checkpoint-17500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..bf5e440f9231731c229fe963c99febfb32a55539 --- /dev/null +++ b/checkpoint-17500/unet_ema/config.json @@ -0,0 +1,58 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.28.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, + "num_train_timesteps": null, + "optimization_step": 17500, + "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-17500/unet_ema/diffusion_pytorch_model.safetensors b/checkpoint-17500/unet_ema/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..d6766932ec05b940ec5be10a53eeb300cfbb62e3 --- /dev/null +++ b/checkpoint-17500/unet_ema/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:67fe7cb6d03412cbf97765d886aeb0cdda4b7af0799a3a566cfb676c87ab896d +size 454741108 diff --git a/checkpoint-18000/optimizer.bin b/checkpoint-18000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..7167e7786f64620c00c72fddd251f151ca6739e9 --- /dev/null +++ b/checkpoint-18000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:386bf4faecd223c746c24c360dd8e948ed6e67d1829ce443e1242b191ae84339 +size 909763736 diff --git a/checkpoint-18000/random_states_0.pkl b/checkpoint-18000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..93ba11280b29b120ce56457da6b4988e70bc3483 --- /dev/null +++ b/checkpoint-18000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:341ba8c723bae6887e8e69951f81a75db475a2e95a2c30941b58b77081a23852 +size 14344 diff --git a/checkpoint-18000/scheduler.bin b/checkpoint-18000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..54844fddb2e0d1bbf323ba16c438fd1cb6559305 --- /dev/null +++ b/checkpoint-18000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c5a1d2afc5c094712aa25387e1af98707df86e80330c33cee8acf713721585dc +size 1000 diff --git a/checkpoint-18000/unet/config.json b/checkpoint-18000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..ea03432831f52ebc1b35187f19dc4707b857d57d --- /dev/null +++ b/checkpoint-18000/unet/config.json @@ -0,0 +1,51 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.28.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, + "num_train_timesteps": 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-18000/unet/diffusion_pytorch_model.safetensors b/checkpoint-18000/unet/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..1a3fb0f17770b402e195884d9ac70dd604e392c0 --- /dev/null +++ b/checkpoint-18000/unet/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:288f991365c0a77ee3e0dda91bc5a2c3dff52ee466aa6697333344101159e677 +size 454741108 diff --git a/checkpoint-18000/unet_ema/config.json b/checkpoint-18000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..3fa061638619695b90be8abf548245cb553321c4 --- /dev/null +++ b/checkpoint-18000/unet_ema/config.json @@ -0,0 +1,58 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.28.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, + "num_train_timesteps": null, + "optimization_step": 18000, + "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-18000/unet_ema/diffusion_pytorch_model.safetensors b/checkpoint-18000/unet_ema/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..976b34fc21ee200211297f0a34244de44c8081dc --- /dev/null +++ b/checkpoint-18000/unet_ema/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:544820b84b1f88ec385e813d765b65a669c28652e4ab88b5edbf97e9a2d54921 +size 454741108 diff --git a/checkpoint-18500/optimizer.bin b/checkpoint-18500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..0b1efc31494dcb2e283fcee46742a73b58c4bf61 --- /dev/null +++ b/checkpoint-18500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:706bed3eef0aa31e12f2adcd611d18079e1932c144f02568e4b5d6b5be0c475b +size 909763736 diff --git a/checkpoint-18500/random_states_0.pkl b/checkpoint-18500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..21e589de9e9f5619cc630c7f8c3e4c2c723d70f9 --- /dev/null +++ b/checkpoint-18500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:95a74ed8bc72c7c7fe3f29c1339bf252305925ad1b4c7c64932522035cee1d2a +size 14344 diff --git a/checkpoint-18500/scheduler.bin b/checkpoint-18500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..109c0841916d446fd15998ef4a9aee71931a9cd7 --- /dev/null +++ b/checkpoint-18500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:25addfde7a0cd7f7fd68623744cf5b21845e352cb60e682e9046bd88cc411305 +size 1000 diff --git a/checkpoint-18500/unet/config.json b/checkpoint-18500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..ea03432831f52ebc1b35187f19dc4707b857d57d --- /dev/null +++ b/checkpoint-18500/unet/config.json @@ -0,0 +1,51 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.28.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, + "num_train_timesteps": 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-18500/unet/diffusion_pytorch_model.safetensors b/checkpoint-18500/unet/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..5b0da8da758cc1509a09afd025c4b68c7849fb5f --- /dev/null +++ b/checkpoint-18500/unet/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:4d26bccda6a9f1135f6b740a1986b05f8ea41ee94fe658ab33262803ead66889 +size 454741108 diff --git a/checkpoint-18500/unet_ema/config.json b/checkpoint-18500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..b84ff88ab8963963b25d98d44b7f12666dc22e07 --- /dev/null +++ b/checkpoint-18500/unet_ema/config.json @@ -0,0 +1,58 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.28.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, + "num_train_timesteps": null, + "optimization_step": 18500, + "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-18500/unet_ema/diffusion_pytorch_model.safetensors b/checkpoint-18500/unet_ema/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..83bb2a48f91d815c9f6e0d223c2493be01dd783f --- /dev/null +++ b/checkpoint-18500/unet_ema/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:bc44071e0fbcc21824b9f61bdf0b8cbc59bf582e70fe02ee85dce2021b8dc379 +size 454741108 diff --git a/checkpoint-19000/optimizer.bin b/checkpoint-19000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..c1d22b4c965c4008bcadcd5afb543a439f18a045 --- /dev/null +++ b/checkpoint-19000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:225ffc6ca9429a508ff7b22c46a995b5cbdd0b4bd910cb6b10c1868ae78f31a5 +size 909763736 diff --git a/checkpoint-19000/random_states_0.pkl b/checkpoint-19000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..958025ffedc7879ecfc7cd2b98d9f71f4a190fdc --- /dev/null +++ b/checkpoint-19000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:3e860f9b61f1ffbecfadb92bea3ef7d1834bedbbbcc7dbfbb2ede5e611026b95 +size 14344 diff --git a/checkpoint-19000/scheduler.bin b/checkpoint-19000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..def9d3b62d32751c1def37494cb20d089ae42895 --- /dev/null +++ b/checkpoint-19000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7f7e1dcc6ffa53ae817e64c2fd47604e942c44ae4c5818a302fcb306f1092134 +size 1000 diff --git a/checkpoint-19000/unet/config.json b/checkpoint-19000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..ea03432831f52ebc1b35187f19dc4707b857d57d --- /dev/null +++ b/checkpoint-19000/unet/config.json @@ -0,0 +1,51 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.28.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, + "num_train_timesteps": 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-19000/unet/diffusion_pytorch_model.safetensors b/checkpoint-19000/unet/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..2920016ff08ea2fc365c39f54f01d9f4fab5760c --- /dev/null +++ b/checkpoint-19000/unet/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:55eb16d620c0d168378200baf3ea82b95dac11cb414ee5ae5e14a3a03f2b7c1f +size 454741108 diff --git a/checkpoint-19000/unet_ema/config.json b/checkpoint-19000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..805596c3f2e1ccc6d609b80b40c0e1655c0575dd --- /dev/null +++ b/checkpoint-19000/unet_ema/config.json @@ -0,0 +1,58 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.28.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, + "num_train_timesteps": null, + "optimization_step": 19000, + "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-19000/unet_ema/diffusion_pytorch_model.safetensors b/checkpoint-19000/unet_ema/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..021795a1e6c2371576a851e72e2068b7cfa06e0c --- /dev/null +++ b/checkpoint-19000/unet_ema/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:f6fcb458c2e4ad480356a46a4304b1540af6ce5b5ccdc7d4e039bfdf08d9903f +size 454741108 diff --git a/checkpoint-19500/optimizer.bin b/checkpoint-19500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..5522c5c9326f1ecfbb68465e85caf03e373bfbf1 --- /dev/null +++ b/checkpoint-19500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7ca28d4c15d39f4da582d3818bf0067374c055b30bb243cacbeaeb19aad3f76a +size 909763736 diff --git a/checkpoint-19500/random_states_0.pkl b/checkpoint-19500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..3cab5a8a7686f2e90805ef8dd7aa5fbcae9b2de9 --- /dev/null +++ b/checkpoint-19500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:2e8bc58dfaf87d890d4f49f78782438319dd4b8f34d93cc4132297a033e443f7 +size 14344 diff --git a/checkpoint-19500/scheduler.bin b/checkpoint-19500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..333014a30a639616bfae294893f5f7be46ebdfa7 --- /dev/null +++ b/checkpoint-19500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e23d483adab6139f29b6cd9eedda08419920305b4ba5b0e4dbf06e78043c3464 +size 1000 diff --git a/checkpoint-19500/unet/config.json b/checkpoint-19500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..ea03432831f52ebc1b35187f19dc4707b857d57d --- /dev/null +++ b/checkpoint-19500/unet/config.json @@ -0,0 +1,51 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.28.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, + "num_train_timesteps": 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-19500/unet/diffusion_pytorch_model.safetensors b/checkpoint-19500/unet/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..d5bda4ecf723409daa21601473b3ff3f61cc8aae --- /dev/null +++ b/checkpoint-19500/unet/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:ad43fdd8318d2de95f5a1bdbdd68bbc8f5201994f15e451197964f99f849c406 +size 454741108 diff --git a/checkpoint-19500/unet_ema/config.json b/checkpoint-19500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..78b4b7af98cfac34e47c8d5e8b9c7f92924ea999 --- /dev/null +++ b/checkpoint-19500/unet_ema/config.json @@ -0,0 +1,58 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.28.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, + "num_train_timesteps": null, + "optimization_step": 19500, + "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-19500/unet_ema/diffusion_pytorch_model.safetensors b/checkpoint-19500/unet_ema/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..d3dfaf547cc2e932056f72279a558301ed938ba1 --- /dev/null +++ b/checkpoint-19500/unet_ema/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:1df7ed04573672092d799094fde5c1b0f3375742ecd10c3990eef0ab42224557 +size 454741108 diff --git a/checkpoint-20000/optimizer.bin b/checkpoint-20000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..693e735cc6abd575221393dea4e27c74c0bcfe1b --- /dev/null +++ b/checkpoint-20000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:aa8e19b650d56181d4e250c7ebf19706235773a51302803d86ca65fc4189a0e1 +size 909763736 diff --git a/checkpoint-20000/random_states_0.pkl b/checkpoint-20000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..aa8e8f30ae0db2a99680e96b19f4271cb5bc38f6 --- /dev/null +++ b/checkpoint-20000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c9e75660d937683aa4559e2b2e3d33550fb0d8fd1331c96ae86d139951b9ab43 +size 14344 diff --git a/checkpoint-20000/scheduler.bin b/checkpoint-20000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..842c9a5d998fa218bd9450eecc7eda475712a6e9 --- /dev/null +++ b/checkpoint-20000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:fbc6468503ff86b756cb6107a7afeccf1d2bdad49be2d3b4da8ca758ccf17288 +size 1000 diff --git a/checkpoint-20000/unet/config.json b/checkpoint-20000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..ea03432831f52ebc1b35187f19dc4707b857d57d --- /dev/null +++ b/checkpoint-20000/unet/config.json @@ -0,0 +1,51 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.28.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, + "num_train_timesteps": 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-20000/unet/diffusion_pytorch_model.safetensors b/checkpoint-20000/unet/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..85a1a5d6b16ffd996c20d9f09cee8178a900bea5 --- /dev/null +++ b/checkpoint-20000/unet/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:49babea5ded794ee8ccf0a959eb53584a2176417266199887830a8d0951ab31a +size 454741108 diff --git a/checkpoint-20000/unet_ema/config.json b/checkpoint-20000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..98aba2d4faa9727049c551d450fc4140b3527b42 --- /dev/null +++ b/checkpoint-20000/unet_ema/config.json @@ -0,0 +1,58 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.28.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, + "num_train_timesteps": null, + "optimization_step": 20000, + "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-20000/unet_ema/diffusion_pytorch_model.safetensors b/checkpoint-20000/unet_ema/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..6b45f92fe9c6afb717fca3e89c603e82361294e0 --- /dev/null +++ b/checkpoint-20000/unet_ema/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:cc7ba1c7a1f4d6a0795462f10cab0a03e11779dab926634b725870339ab824fd +size 454741108 diff --git a/checkpoint-20500/optimizer.bin b/checkpoint-20500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..5f7798849a99826941fffddd29f875967366a1e4 --- /dev/null +++ b/checkpoint-20500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:af3397a2c9e62db35b1774fcd1ff5675a328366c0ba4512ba9c6388990ed791b +size 909763736 diff --git a/checkpoint-20500/random_states_0.pkl b/checkpoint-20500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..b9fd49187eed4c07e37101f6ebddba5b84fc546f --- /dev/null +++ b/checkpoint-20500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:968bd2cb55fed20298e43f52a0f941677fa6a0bd0db7215ece8bd550aa6af79d +size 14344 diff --git a/checkpoint-20500/scheduler.bin b/checkpoint-20500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..0ad5a3bf6b0902c91222726f640c4dea7164c83a --- /dev/null +++ b/checkpoint-20500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:f1635e0121d8061539e323f06d8a956e9ee10c442d92400350c057a2a0be5a68 +size 1000 diff --git a/checkpoint-20500/unet/config.json b/checkpoint-20500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..ea03432831f52ebc1b35187f19dc4707b857d57d --- /dev/null +++ b/checkpoint-20500/unet/config.json @@ -0,0 +1,51 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.28.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, + "num_train_timesteps": 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-20500/unet/diffusion_pytorch_model.safetensors b/checkpoint-20500/unet/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..9ef8391851b4b5dad72219e5d89b1db0de877723 --- /dev/null +++ b/checkpoint-20500/unet/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:0f1d155460e688624bc2aabe285c5eff920f91f91a4c1ac94d95344daf23db40 +size 454741108 diff --git a/checkpoint-20500/unet_ema/config.json b/checkpoint-20500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..243619fbe2d84cfb32a64db4cafa0c3ed62d95d9 --- /dev/null +++ b/checkpoint-20500/unet_ema/config.json @@ -0,0 +1,58 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.28.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, + "num_train_timesteps": null, + "optimization_step": 20500, + "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-20500/unet_ema/diffusion_pytorch_model.safetensors b/checkpoint-20500/unet_ema/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..5b8db1e40fded216e95ec0343f9e400e0861677a --- /dev/null +++ b/checkpoint-20500/unet_ema/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:d414a2d394ec65007edaa7715a23fc2e9a8e80a03aadbf30d8ae7b8b731d9b46 +size 454741108 diff --git a/logs/train_unconditional/events.out.tfevents.1715806000.jobtest-6wkxj-dk9zr.141.0 b/logs/train_unconditional/events.out.tfevents.1715806000.jobtest-6wkxj-dk9zr.141.0 index 68819fe66f7ce8e8ebbd29532e408c5bcea00fcd..d8307c72c3e0e8a66f6160e5be2fbc1bb63e988d 100644 --- a/logs/train_unconditional/events.out.tfevents.1715806000.jobtest-6wkxj-dk9zr.141.0 +++ b/logs/train_unconditional/events.out.tfevents.1715806000.jobtest-6wkxj-dk9zr.141.0 @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:20726d393b318b4c912f39257f4284e1bbb579c97c4d6635fb164ad29568d7fc -size 3303344 +oid sha256:9feab0c3dea3741f6cff8e5e3c4a473ec9a1484a3904ea9dc4d8f6db22c28610 +size 4341524 diff --git a/unet/diffusion_pytorch_model.safetensors b/unet/diffusion_pytorch_model.safetensors index 34d102bae838702fe03c69b25092176b53c73c54..9d19d0a70e18b39790108462d4c4377e5320a6e4 100644 --- a/unet/diffusion_pytorch_model.safetensors +++ b/unet/diffusion_pytorch_model.safetensors @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:d780b690dbb3846a425daccff115fa705b95654e3796d168b6860b802d8241fd +oid sha256:3f01afef722db03966bdc33e3662a9e5c419388d71e29ad056ab472db79871b8 size 454741108