diff --git a/checkpoint-1000/optimizer.bin b/checkpoint-1000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..911b4d71adb8d0010a41cccfb493a9b0fd881d82 --- /dev/null +++ b/checkpoint-1000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c61dc06f5fce098d332c70160bb5d443ef84dd2171b2b68212076288de80e6f6 +size 909763736 diff --git a/checkpoint-1000/random_states_0.pkl b/checkpoint-1000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..cc2b57bee07f2ce7d0fab9bc77bd213a675b1303 --- /dev/null +++ b/checkpoint-1000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:dad3930b68e400b563d0e697b3052e2e1f3ca65e5f21938a79429bd1019dd58b +size 14344 diff --git a/checkpoint-1000/scheduler.bin b/checkpoint-1000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..375ec3c3e2a295b7f8da7296209d33ac2d23bf16 --- /dev/null +++ b/checkpoint-1000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:6b86031261a937c99a5fe445f078b6e4fe5e48eae3857c3810ef5de9d8d3ab6b +size 1000 diff --git a/checkpoint-1000/unet/config.json b/checkpoint-1000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..723efd5f8ddc77a53ace12ef7f4e89dccfd9f220 --- /dev/null +++ b/checkpoint-1000/unet/config.json @@ -0,0 +1,51 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.23.1", + "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-1000/unet/diffusion_pytorch_model.safetensors b/checkpoint-1000/unet/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..a479ccbddd457179e028d8c42ca7d8db60ca0b76 --- /dev/null +++ b/checkpoint-1000/unet/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:28f87be0fbcf6a62cca506259820d56ffed0a90900f96bea5f09d2c419bd6ab4 +size 454741108 diff --git a/checkpoint-1000/unet_ema/config.json b/checkpoint-1000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..84fb94d73b1d288e8f0ff265745e583c918991f0 --- /dev/null +++ b/checkpoint-1000/unet_ema/config.json @@ -0,0 +1,58 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.23.1", + "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": 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..f58acfd7bc7d33564177ecc5f6942fbd74507cc3 --- /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:8f54dee0c990b630cfd12c92de30623c1dd69cc10c95190afce97a30513ab46f +size 454741108 diff --git a/checkpoint-1500/optimizer.bin b/checkpoint-1500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..372cd8c8ae85f917cb991abc00550110928c7945 --- /dev/null +++ b/checkpoint-1500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:d1f279a1d0844c1800d2ea37e088fd70bc512c78f8a6e13ccbac1bc53aa91df8 +size 909763736 diff --git a/checkpoint-1500/random_states_0.pkl b/checkpoint-1500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..8850ff2e5fa54184f6e0d1e768473f936370f5c7 --- /dev/null +++ b/checkpoint-1500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e447aea30ce6e471bcc30635dd42f5b46544393987e61e1100170a1c243e18e0 +size 14344 diff --git a/checkpoint-1500/scheduler.bin b/checkpoint-1500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..bb0f8035ad5e504d93d38c2574b74fac703cb3b5 --- /dev/null +++ b/checkpoint-1500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7d3ef90278bb3f5fda9049255540ecd16f299890c459ec808534c82d1001ef55 +size 1000 diff --git a/checkpoint-1500/unet/config.json b/checkpoint-1500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..723efd5f8ddc77a53ace12ef7f4e89dccfd9f220 --- /dev/null +++ b/checkpoint-1500/unet/config.json @@ -0,0 +1,51 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.23.1", + "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-1500/unet/diffusion_pytorch_model.safetensors b/checkpoint-1500/unet/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..e49031d5da0d156846fc80496591c542e551dc6b --- /dev/null +++ b/checkpoint-1500/unet/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:818792b32c5dc200e37708c90db4b946afb11fda7015e934cc66e8c199e321f6 +size 454741108 diff --git a/checkpoint-1500/unet_ema/config.json b/checkpoint-1500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..73453ad7eac64a03ddd2dfe4796be3d7826f9a44 --- /dev/null +++ b/checkpoint-1500/unet_ema/config.json @@ -0,0 +1,58 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.23.1", + "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": 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..838c2ba68cf6e85ef4c5050d617d4cd85b0f956c --- /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:15b9f533923ab0a780b27db3a918603943dda2e50911c7bbabe6f2cbe9ba17ed +size 454741108 diff --git a/checkpoint-2000/optimizer.bin b/checkpoint-2000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..045b7560c055388660d423066e24f62f83f10761 --- /dev/null +++ b/checkpoint-2000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7c236bb35e96f8dc007cc51614f7d21482586283d89c7063aba520d9f6c6974a +size 909763736 diff --git a/checkpoint-2000/random_states_0.pkl b/checkpoint-2000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..46efcfade946f97bb7702f8dc0ce622a0e15dff3 --- /dev/null +++ b/checkpoint-2000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:80b296177f5432e508e912da8c51df6a01ba930e1234a96e1a27230bc9cf635d +size 14344 diff --git a/checkpoint-2000/scheduler.bin b/checkpoint-2000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..931640af338726694b186d6d85a18609946e7b7a --- /dev/null +++ b/checkpoint-2000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:78170adbf3cf6ac2bef73c72c0aa3e794a9892b0516e0b98365d4f537c03f26a +size 1000 diff --git a/checkpoint-2000/unet/config.json b/checkpoint-2000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..723efd5f8ddc77a53ace12ef7f4e89dccfd9f220 --- /dev/null +++ b/checkpoint-2000/unet/config.json @@ -0,0 +1,51 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.23.1", + "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-2000/unet/diffusion_pytorch_model.safetensors b/checkpoint-2000/unet/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..0b9c20bcf036c0240ee180f58ac8744770dd2bc8 --- /dev/null +++ b/checkpoint-2000/unet/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b009694ad2268725baaac5d8770610f3baec723d28c2d87e5b20e3d6913e7eb1 +size 454741108 diff --git a/checkpoint-2000/unet_ema/config.json b/checkpoint-2000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..00de2a0e18e12dc6277c875d30d83d2324b3471b --- /dev/null +++ b/checkpoint-2000/unet_ema/config.json @@ -0,0 +1,58 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.23.1", + "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": 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..9b880c1881872ed1e7f88a0e5e89b59a7113005c --- /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:117e3f34ebaa58c06ed9adeb10b08d388944f835ef1da316dce8a8ff0d5b747c +size 454741108 diff --git a/checkpoint-2500/optimizer.bin b/checkpoint-2500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..fab09a4e06c76aa1163a7030905ca9fe5dc22e88 --- /dev/null +++ b/checkpoint-2500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:4803c11d81905e3cdd3429a49fb727ddb91b18cf491ba01b4b91c42c40b93bd0 +size 909763736 diff --git a/checkpoint-2500/random_states_0.pkl b/checkpoint-2500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..a8bf65d65a3b69f128b68f5eac9d41a4399aba51 --- /dev/null +++ b/checkpoint-2500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c4c7fd7c03f2f12f4682a161785daecca6f66bb3f8a4123f3fc27dec7559412c +size 14344 diff --git a/checkpoint-2500/scheduler.bin b/checkpoint-2500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..6f46a6c2886f013239107275eccfee213f6e2cd6 --- /dev/null +++ b/checkpoint-2500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:6548ae0a212ef2edeb47da37fc1297ddbc8ef26dece8738ca263ed55912c1c9d +size 1000 diff --git a/checkpoint-2500/unet/config.json b/checkpoint-2500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..723efd5f8ddc77a53ace12ef7f4e89dccfd9f220 --- /dev/null +++ b/checkpoint-2500/unet/config.json @@ -0,0 +1,51 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.23.1", + "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-2500/unet/diffusion_pytorch_model.safetensors b/checkpoint-2500/unet/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..99a204e1c3a2acb9bbac6e5ab0b27db99ed1de1c --- /dev/null +++ b/checkpoint-2500/unet/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e21f77fb83667b1c10df532d617a06614da6f0e75301af5a41434a534c83408f +size 454741108 diff --git a/checkpoint-2500/unet_ema/config.json b/checkpoint-2500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..02fce2076d67ee326aeca801f94215c547a3f943 --- /dev/null +++ b/checkpoint-2500/unet_ema/config.json @@ -0,0 +1,58 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.23.1", + "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": 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..9f34c0c97f66bd219cde2dfb35bbdb23ec9434f5 --- /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:6ab60ab2d9cf2944f2b1e83330553408b7303b195b599b2a8dcb936a52242dc6 +size 454741108 diff --git a/checkpoint-3000/optimizer.bin b/checkpoint-3000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..8bce9042e177284f1de6eca1f695204a0ef90ebe --- /dev/null +++ b/checkpoint-3000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:9d5176047d6149027b5d618da59c3f7a384f5775dad79b2d85f40a942eb64311 +size 909763736 diff --git a/checkpoint-3000/random_states_0.pkl b/checkpoint-3000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..efa53f3071375b37c58e53425d4cc3ec6939ae45 --- /dev/null +++ b/checkpoint-3000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:24db5b67933d802d298064be181aa394091709e693537b9da79028a404e17a58 +size 14344 diff --git a/checkpoint-3000/scheduler.bin b/checkpoint-3000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..6f116a252c8ae7c3b51be6369f56ad44e29b1bb1 --- /dev/null +++ b/checkpoint-3000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:562ea14859e2a40b54204f3a0bee2fa8913b65cbe472be88e229e992bfcdb892 +size 1000 diff --git a/checkpoint-3000/unet/config.json b/checkpoint-3000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..723efd5f8ddc77a53ace12ef7f4e89dccfd9f220 --- /dev/null +++ b/checkpoint-3000/unet/config.json @@ -0,0 +1,51 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.23.1", + "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-3000/unet/diffusion_pytorch_model.safetensors b/checkpoint-3000/unet/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..e4f31b6b0d7bc0be966928563dc4d31a1b4fb189 --- /dev/null +++ b/checkpoint-3000/unet/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:25b049a16ce43546b4cfd363c0d5c9051e8580396c5e0ae0dc931870c30f9058 +size 454741108 diff --git a/checkpoint-3000/unet_ema/config.json b/checkpoint-3000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..58881e53d71d815542e5e29fb0fcd504a29d896b --- /dev/null +++ b/checkpoint-3000/unet_ema/config.json @@ -0,0 +1,58 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.23.1", + "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": 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..7e0b105187b5b2f76591834554ce4fb8e6d1bae0 --- /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:85332facadf168158b309c1061a032b1ba8520521baaa091816c629db535e456 +size 454741108 diff --git a/checkpoint-3500/optimizer.bin b/checkpoint-3500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..bef66e01574e7dfd8e5ce291c7ec87f2635e78fd --- /dev/null +++ b/checkpoint-3500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c60432641c942027502cea2342ae94b63c715280a48c75865373220736ae96dd +size 909763736 diff --git a/checkpoint-3500/random_states_0.pkl b/checkpoint-3500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..efeee93d24c34d94638e94095eaaeb8aab32b781 --- /dev/null +++ b/checkpoint-3500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:aa2ef459ec5e1063b92ac1f2a7eba445bdfd187efa894fe51c9b0fe578f126d9 +size 14344 diff --git a/checkpoint-3500/scheduler.bin b/checkpoint-3500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..89b98a0ba16a948494ca280d4417893a5a337be7 --- /dev/null +++ b/checkpoint-3500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:4da964b0be7528b075baf1572a6f8b80d83109342bac8dcc19fe076000d766b9 +size 1000 diff --git a/checkpoint-3500/unet/config.json b/checkpoint-3500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..723efd5f8ddc77a53ace12ef7f4e89dccfd9f220 --- /dev/null +++ b/checkpoint-3500/unet/config.json @@ -0,0 +1,51 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.23.1", + "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-3500/unet/diffusion_pytorch_model.safetensors b/checkpoint-3500/unet/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..1a25ddb75eb1d0098251b8e5405648b79e497d4f --- /dev/null +++ b/checkpoint-3500/unet/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:d2040f9752b2215d749026a8f8ded1dcff5eb1f08ad984ecbd3f9925414339a3 +size 454741108 diff --git a/checkpoint-3500/unet_ema/config.json b/checkpoint-3500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..0d561f4d804e8896e3e2d8fed99e4b24b6a11f33 --- /dev/null +++ b/checkpoint-3500/unet_ema/config.json @@ -0,0 +1,58 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.23.1", + "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": 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..60ddebf8f96d33d1db9234d7d23d0b9eb6a8425b --- /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:625ccd47928cab554ee9b1aaf70d2cae0dacf7142a1d6182b6acfed0f9d431d5 +size 454741108 diff --git a/checkpoint-4000/optimizer.bin b/checkpoint-4000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..4eab31fc9c6dac2c188b3b19b3823c4023a5a1ee --- /dev/null +++ b/checkpoint-4000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:ff94d279685daf1f18c6d04444cf73f23e69bb993a074e613240c55e38254ee4 +size 909763736 diff --git a/checkpoint-4000/random_states_0.pkl b/checkpoint-4000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..4ec5d68e9da8e7f5dedd7045c8157ce58ddf2367 --- /dev/null +++ b/checkpoint-4000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7ae64a48717a0cc9edcb87a64cdc1bf18211bf04c2cccc90404956274205e4bd +size 14344 diff --git a/checkpoint-4000/scheduler.bin b/checkpoint-4000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..9b6b22ae12bc95babf0e48530bddfac5da0b1c2c --- /dev/null +++ b/checkpoint-4000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:cb8754cf27b300c6d71a400a01531c6b3681404cba3a03ec8cf21ddfc15d9b92 +size 1000 diff --git a/checkpoint-4000/unet/config.json b/checkpoint-4000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..723efd5f8ddc77a53ace12ef7f4e89dccfd9f220 --- /dev/null +++ b/checkpoint-4000/unet/config.json @@ -0,0 +1,51 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.23.1", + "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-4000/unet/diffusion_pytorch_model.safetensors b/checkpoint-4000/unet/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..0979520a330ab341af2ec61ae99b758f8b98f275 --- /dev/null +++ b/checkpoint-4000/unet/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:a12c899b837f2663da8820b4c2c058efeb7ff343902e0558398810379e7e1de8 +size 454741108 diff --git a/checkpoint-4000/unet_ema/config.json b/checkpoint-4000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..1678f56320e028d686f87b1ff259172286628d01 --- /dev/null +++ b/checkpoint-4000/unet_ema/config.json @@ -0,0 +1,58 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.23.1", + "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": 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..1faab786242a0d08a14a75332e781baeb49eaf10 --- /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:f196ecedf76a4325054e5dd50ad181cf39e7a9097eafc56276b2d9295f538b11 +size 454741108 diff --git a/checkpoint-4500/optimizer.bin b/checkpoint-4500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..8860a62567200b1db19bfcceeead77d70837f579 --- /dev/null +++ b/checkpoint-4500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:602f30f877026f13f2f29b4dd2ff8b5d316bd44117219279449caaa69c550cca +size 909763736 diff --git a/checkpoint-4500/random_states_0.pkl b/checkpoint-4500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..a71fdc9c0ce4abb9d4013a999fbaae1de902effe --- /dev/null +++ b/checkpoint-4500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:4eef6b8c6b64f0a079b92c1ae057836ad5f3d965c2d6c7d951ff7ee84a521208 +size 14344 diff --git a/checkpoint-4500/scheduler.bin b/checkpoint-4500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..acd3e30a12b2741da3a15be8536fdedf0a263006 --- /dev/null +++ b/checkpoint-4500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c5f141fc3a421991c83fa96c576340e8618ed422989e41d19af9e43cc643bbcf +size 1000 diff --git a/checkpoint-4500/unet/config.json b/checkpoint-4500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..723efd5f8ddc77a53ace12ef7f4e89dccfd9f220 --- /dev/null +++ b/checkpoint-4500/unet/config.json @@ -0,0 +1,51 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.23.1", + "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-4500/unet/diffusion_pytorch_model.safetensors b/checkpoint-4500/unet/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..d8893e39b9a81504c4cb8d367d10d3029810c268 --- /dev/null +++ b/checkpoint-4500/unet/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:f12445e59dc28000f93468f2158aae8f00db79fffdd310f7c6ed60e358eaf314 +size 454741108 diff --git a/checkpoint-4500/unet_ema/config.json b/checkpoint-4500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..cb07793a2ac87deeb001253b570292f31f5872ae --- /dev/null +++ b/checkpoint-4500/unet_ema/config.json @@ -0,0 +1,58 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.23.1", + "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": 4500, + "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-4500/unet_ema/diffusion_pytorch_model.safetensors b/checkpoint-4500/unet_ema/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..aedc62e1b7d7948d7295620705246fc900b90512 --- /dev/null +++ b/checkpoint-4500/unet_ema/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:9fde1c720d18fdb486360b12f599b716544bc1a0b395e105762a0a95d6fad2f1 +size 454741108 diff --git a/checkpoint-500/optimizer.bin b/checkpoint-500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..702c1c054966652a48ef52f31fe698e94c381c51 --- /dev/null +++ b/checkpoint-500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e26cbcf4a5b84d281996a97f6c4af39ee9804795984dac17777795a3ff9edfea +size 909763736 diff --git a/checkpoint-500/random_states_0.pkl b/checkpoint-500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..91fe77df9c8f97fa369d7bd5e3a1a1af5c98e59b --- /dev/null +++ b/checkpoint-500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:6911ba52d44e963101624af53fc8b847d7b97a755a5426b16e8cba7d82c4e10b +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..723efd5f8ddc77a53ace12ef7f4e89dccfd9f220 --- /dev/null +++ b/checkpoint-500/unet/config.json @@ -0,0 +1,51 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.23.1", + "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-500/unet/diffusion_pytorch_model.safetensors b/checkpoint-500/unet/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..1c039f31b0529ca0fc5d28bd52cbf7267353ad7d --- /dev/null +++ b/checkpoint-500/unet/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:5a3882f4037fac347c55587cf900f1a27b1cc2efe0e960b40e2d04b9b842488d +size 454741108 diff --git a/checkpoint-500/unet_ema/config.json b/checkpoint-500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..738f4af56bbb6a2c48340cd79136bbf53981cb46 --- /dev/null +++ b/checkpoint-500/unet_ema/config.json @@ -0,0 +1,58 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.23.1", + "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": 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..2ce0fe0129ddbe5110da35f8f437c5512b4e4781 --- /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:cfd8683ae02fcd717fda4c837bd82da6e80ecc5e5444095b11d15d0eee82fd7d +size 454741108 diff --git a/checkpoint-5000/optimizer.bin b/checkpoint-5000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..2247105299bc99e91cdda198b6d6c7570d2806c5 --- /dev/null +++ b/checkpoint-5000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:997792334ed17d4f77ff9f570fba7f7edad2a1644d979700044fbf6d42f2c8f9 +size 909763736 diff --git a/checkpoint-5000/random_states_0.pkl b/checkpoint-5000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..a9108d917bd33e987262efa7d3b0a359a2d03cc9 --- /dev/null +++ b/checkpoint-5000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:6351c09dbf8fcd45a649467e5a3b60b2c30b6a13f43a8ff7dde6fe22dd7dbac9 +size 14344 diff --git a/checkpoint-5000/scheduler.bin b/checkpoint-5000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..2ee27e507b345bfa3faceb08f802d4406197d57e --- /dev/null +++ b/checkpoint-5000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:276f174665d2ef52f42c8c161ffe24ad75b89c23f90ee6a9d14fae61e62b385f +size 1000 diff --git a/checkpoint-5000/unet/config.json b/checkpoint-5000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..723efd5f8ddc77a53ace12ef7f4e89dccfd9f220 --- /dev/null +++ b/checkpoint-5000/unet/config.json @@ -0,0 +1,51 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.23.1", + "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-5000/unet/diffusion_pytorch_model.safetensors b/checkpoint-5000/unet/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..261ca8085f7979467b0bd6f6bb6ef755dbc2be99 --- /dev/null +++ b/checkpoint-5000/unet/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:720bee455e7472a702b13f84466bb99f1165fb882cb02fed8ae4023e365d26fa +size 454741108 diff --git a/checkpoint-5000/unet_ema/config.json b/checkpoint-5000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..79d6a196a0864cb433ebe8b2ef3bb82a2bbb6d52 --- /dev/null +++ b/checkpoint-5000/unet_ema/config.json @@ -0,0 +1,58 @@ +{ + "_class_name": "UNet2DModel", + "_diffusers_version": "0.23.1", + "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": 5000, + "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-5000/unet_ema/diffusion_pytorch_model.safetensors b/checkpoint-5000/unet_ema/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..0e51ed57e65b986453c6bfa75702a8fab91f19c2 --- /dev/null +++ b/checkpoint-5000/unet_ema/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:056746c833bdd84f7c7b7eb3ac59b19015dbc773a9aa611e1cddd86f91a2ceaf +size 454741108 diff --git a/unet/diffusion_pytorch_model.safetensors b/unet/diffusion_pytorch_model.safetensors index 93a2d447273aa2baff8c23edc633a85d625d9603..e45ddbcf985ee24015d9acc412eff84160d9169c 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:ae3c861e5d4a64c1ac2860f967608662765e53e0f890808a143ea5a899a7540f +oid sha256:66cba5f29cfa82f84e79aaec9336cb2b46cdef8774b45f8adff4f266e98f9092 size 454741108