diff --git a/checkpoint-117500/optimizer.bin b/checkpoint-117500/optimizer.bin
new file mode 100644
index 0000000000000000000000000000000000000000..eefb939b9947c3f708f94f6e6e8d74e0c2338040
--- /dev/null
+++ b/checkpoint-117500/optimizer.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2e0fcd45e80ba2877fc28361a4d30bf2343e2db1a23d7ab43c413476a07f1097
+size 73251413
diff --git a/checkpoint-117500/random_states_0.pkl b/checkpoint-117500/random_states_0.pkl
new file mode 100644
index 0000000000000000000000000000000000000000..84f2634e1adcf08859c0e152f941678d41109f20
--- /dev/null
+++ b/checkpoint-117500/random_states_0.pkl
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:01395318db1c9d4c19bec667947a3aea1b05149a2445f11ae1d87f56c5d0751f
+size 14663
diff --git a/checkpoint-117500/scaler.pt b/checkpoint-117500/scaler.pt
new file mode 100644
index 0000000000000000000000000000000000000000..3ea1e14f1400a2e55265ecc8a42deb1dbe1b9968
--- /dev/null
+++ b/checkpoint-117500/scaler.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a0cf1a3634d9c158c6f877b49d0e46e34a1bc97dee2ec735919212253f45f2ba
+size 557
diff --git a/checkpoint-117500/scheduler.bin b/checkpoint-117500/scheduler.bin
new file mode 100644
index 0000000000000000000000000000000000000000..646f051fe88325fda9a4051732bbafec9bda9ba8
--- /dev/null
+++ b/checkpoint-117500/scheduler.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:005fcbd1174cfd135cd78068333586fc1c9574d872673c9411dc0d57adbd530c
+size 563
diff --git a/checkpoint-117500/unet/config.json b/checkpoint-117500/unet/config.json
new file mode 100644
index 0000000000000000000000000000000000000000..bb1cd664fc36d4dc821a4f0fdae4b94d7d99b356
--- /dev/null
+++ b/checkpoint-117500/unet/config.json
@@ -0,0 +1,60 @@
+{
+ "_class_name": "UNet2DConditionModel",
+ "_diffusers_version": "0.16.1",
+ "act_fn": "silu",
+ "addition_embed_type": null,
+ "addition_embed_type_num_heads": 64,
+ "attention_head_dim": 8,
+ "block_out_channels": [
+ 64,
+ 64,
+ 128,
+ 128
+ ],
+ "center_input_sample": false,
+ "class_embed_type": null,
+ "class_embeddings_concat": false,
+ "conv_in_kernel": 3,
+ "conv_out_kernel": 3,
+ "cross_attention_dim": 256,
+ "cross_attention_norm": null,
+ "down_block_types": [
+ "DownBlock2D",
+ "DownBlock2D",
+ "DownBlock2D",
+ "CrossAttnDownBlock2D"
+ ],
+ "downsample_padding": 1,
+ "dual_cross_attention": false,
+ "encoder_hid_dim": null,
+ "flip_sin_to_cos": true,
+ "freq_shift": 0,
+ "in_channels": 3,
+ "layers_per_block": 2,
+ "mid_block_only_cross_attention": null,
+ "mid_block_scale_factor": 1,
+ "mid_block_type": "UNetMidBlock2DCrossAttn",
+ "norm_eps": 1e-05,
+ "norm_num_groups": 32,
+ "num_class_embeds": null,
+ "only_cross_attention": false,
+ "out_channels": 3,
+ "projection_class_embeddings_input_dim": null,
+ "resnet_out_scale_factor": 1.0,
+ "resnet_skip_time_act": false,
+ "resnet_time_scale_shift": "default",
+ "sample_size": 256,
+ "time_cond_proj_dim": null,
+ "time_embedding_act_fn": null,
+ "time_embedding_dim": null,
+ "time_embedding_type": "positional",
+ "timestep_post_act": null,
+ "up_block_types": [
+ "CrossAttnUpBlock2D",
+ "UpBlock2D",
+ "UpBlock2D",
+ "UpBlock2D"
+ ],
+ "upcast_attention": false,
+ "use_linear_projection": false
+}
diff --git a/checkpoint-117500/unet/diffusion_pytorch_model.bin b/checkpoint-117500/unet/diffusion_pytorch_model.bin
new file mode 100644
index 0000000000000000000000000000000000000000..c3f12062811c851bb046b9ad5a1e7850a22896e5
--- /dev/null
+++ b/checkpoint-117500/unet/diffusion_pytorch_model.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:44f75daf6cc709b785a6d599ec61e9141ec2cd23d6e9a288bf505aa137b99582
+size 36621969
diff --git a/checkpoint-117500/unet_ema/config.json b/checkpoint-117500/unet_ema/config.json
new file mode 100644
index 0000000000000000000000000000000000000000..9464a0056c85035de8f7cd88a5e76b75ee87ed2c
--- /dev/null
+++ b/checkpoint-117500/unet_ema/config.json
@@ -0,0 +1,67 @@
+{
+ "_class_name": "UNet2DConditionModel",
+ "_diffusers_version": "0.16.1",
+ "act_fn": "silu",
+ "addition_embed_type": null,
+ "addition_embed_type_num_heads": 64,
+ "attention_head_dim": 8,
+ "block_out_channels": [
+ 64,
+ 64,
+ 128,
+ 128
+ ],
+ "center_input_sample": false,
+ "class_embed_type": null,
+ "class_embeddings_concat": false,
+ "conv_in_kernel": 3,
+ "conv_out_kernel": 3,
+ "cross_attention_dim": 256,
+ "cross_attention_norm": null,
+ "decay": 0.9999,
+ "down_block_types": [
+ "DownBlock2D",
+ "DownBlock2D",
+ "DownBlock2D",
+ "CrossAttnDownBlock2D"
+ ],
+ "downsample_padding": 1,
+ "dual_cross_attention": false,
+ "encoder_hid_dim": null,
+ "flip_sin_to_cos": true,
+ "freq_shift": 0,
+ "in_channels": 3,
+ "inv_gamma": 1.0,
+ "layers_per_block": 2,
+ "mid_block_only_cross_attention": null,
+ "mid_block_scale_factor": 1,
+ "mid_block_type": "UNetMidBlock2DCrossAttn",
+ "min_decay": 0.0,
+ "norm_eps": 1e-05,
+ "norm_num_groups": 32,
+ "num_class_embeds": null,
+ "only_cross_attention": false,
+ "optimization_step": 117500,
+ "out_channels": 3,
+ "power": 0.75,
+ "projection_class_embeddings_input_dim": null,
+ "resnet_out_scale_factor": 1.0,
+ "resnet_skip_time_act": false,
+ "resnet_time_scale_shift": "default",
+ "sample_size": 256,
+ "time_cond_proj_dim": null,
+ "time_embedding_act_fn": null,
+ "time_embedding_dim": null,
+ "time_embedding_type": "positional",
+ "timestep_post_act": null,
+ "up_block_types": [
+ "CrossAttnUpBlock2D",
+ "UpBlock2D",
+ "UpBlock2D",
+ "UpBlock2D"
+ ],
+ "upcast_attention": false,
+ "update_after_step": 0,
+ "use_ema_warmup": true,
+ "use_linear_projection": false
+}
diff --git a/checkpoint-117500/unet_ema/diffusion_pytorch_model.bin b/checkpoint-117500/unet_ema/diffusion_pytorch_model.bin
new file mode 100644
index 0000000000000000000000000000000000000000..af8827f12f6f534217eff6dd324ede3ddfb2b746
--- /dev/null
+++ b/checkpoint-117500/unet_ema/diffusion_pytorch_model.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:79411464e1fa4561a2cb607ffe4e9b9135c3a4f53c21dc94dbc9cfeeee1c6056
+size 36616081
diff --git a/checkpoint-118000/optimizer.bin b/checkpoint-118000/optimizer.bin
new file mode 100644
index 0000000000000000000000000000000000000000..6b8cb580d2c64cd1ab56f782c1c242c59df3a95e
--- /dev/null
+++ b/checkpoint-118000/optimizer.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1e4d0d3b43163c50f51d7ca1844ac6abedba8c43a18f7483b281e36f0cc562d6
+size 73251413
diff --git a/checkpoint-118000/random_states_0.pkl b/checkpoint-118000/random_states_0.pkl
new file mode 100644
index 0000000000000000000000000000000000000000..3758affd5b64bcec628e67b046df0d05d9ed9ec9
--- /dev/null
+++ b/checkpoint-118000/random_states_0.pkl
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:38deba8a5368bcdc3483ff03dd4621b581aae48bd9bbe4ddbec37e5da9d4ed92
+size 14663
diff --git a/checkpoint-118000/scaler.pt b/checkpoint-118000/scaler.pt
new file mode 100644
index 0000000000000000000000000000000000000000..cb2b39384991cfd057a6c155255680b1e6b91123
--- /dev/null
+++ b/checkpoint-118000/scaler.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:236299ba8f8e8c1220158889471deb193e15dcc451eb38e86eb159ce562fd0c7
+size 557
diff --git a/checkpoint-118000/scheduler.bin b/checkpoint-118000/scheduler.bin
new file mode 100644
index 0000000000000000000000000000000000000000..7b91cd8a2215d7f28f5aedf4eab1aba2af7b7587
--- /dev/null
+++ b/checkpoint-118000/scheduler.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e4c8944f45365884962b599473e22730e34469402a00f1eaae8dbfd8faef5098
+size 563
diff --git a/checkpoint-118000/unet/config.json b/checkpoint-118000/unet/config.json
new file mode 100644
index 0000000000000000000000000000000000000000..bb1cd664fc36d4dc821a4f0fdae4b94d7d99b356
--- /dev/null
+++ b/checkpoint-118000/unet/config.json
@@ -0,0 +1,60 @@
+{
+ "_class_name": "UNet2DConditionModel",
+ "_diffusers_version": "0.16.1",
+ "act_fn": "silu",
+ "addition_embed_type": null,
+ "addition_embed_type_num_heads": 64,
+ "attention_head_dim": 8,
+ "block_out_channels": [
+ 64,
+ 64,
+ 128,
+ 128
+ ],
+ "center_input_sample": false,
+ "class_embed_type": null,
+ "class_embeddings_concat": false,
+ "conv_in_kernel": 3,
+ "conv_out_kernel": 3,
+ "cross_attention_dim": 256,
+ "cross_attention_norm": null,
+ "down_block_types": [
+ "DownBlock2D",
+ "DownBlock2D",
+ "DownBlock2D",
+ "CrossAttnDownBlock2D"
+ ],
+ "downsample_padding": 1,
+ "dual_cross_attention": false,
+ "encoder_hid_dim": null,
+ "flip_sin_to_cos": true,
+ "freq_shift": 0,
+ "in_channels": 3,
+ "layers_per_block": 2,
+ "mid_block_only_cross_attention": null,
+ "mid_block_scale_factor": 1,
+ "mid_block_type": "UNetMidBlock2DCrossAttn",
+ "norm_eps": 1e-05,
+ "norm_num_groups": 32,
+ "num_class_embeds": null,
+ "only_cross_attention": false,
+ "out_channels": 3,
+ "projection_class_embeddings_input_dim": null,
+ "resnet_out_scale_factor": 1.0,
+ "resnet_skip_time_act": false,
+ "resnet_time_scale_shift": "default",
+ "sample_size": 256,
+ "time_cond_proj_dim": null,
+ "time_embedding_act_fn": null,
+ "time_embedding_dim": null,
+ "time_embedding_type": "positional",
+ "timestep_post_act": null,
+ "up_block_types": [
+ "CrossAttnUpBlock2D",
+ "UpBlock2D",
+ "UpBlock2D",
+ "UpBlock2D"
+ ],
+ "upcast_attention": false,
+ "use_linear_projection": false
+}
diff --git a/checkpoint-118000/unet/diffusion_pytorch_model.bin b/checkpoint-118000/unet/diffusion_pytorch_model.bin
new file mode 100644
index 0000000000000000000000000000000000000000..4b38d0039af03b938fb54e761e2f47fb527410af
--- /dev/null
+++ b/checkpoint-118000/unet/diffusion_pytorch_model.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:21ea103f0f7364eb4f7914857bc91709ae0343971c81190f6fe1f0b8d62c0127
+size 36621969
diff --git a/checkpoint-118000/unet_ema/config.json b/checkpoint-118000/unet_ema/config.json
new file mode 100644
index 0000000000000000000000000000000000000000..47b4eb3dbcff4d520a2687b0414fc4edf88b8949
--- /dev/null
+++ b/checkpoint-118000/unet_ema/config.json
@@ -0,0 +1,67 @@
+{
+ "_class_name": "UNet2DConditionModel",
+ "_diffusers_version": "0.16.1",
+ "act_fn": "silu",
+ "addition_embed_type": null,
+ "addition_embed_type_num_heads": 64,
+ "attention_head_dim": 8,
+ "block_out_channels": [
+ 64,
+ 64,
+ 128,
+ 128
+ ],
+ "center_input_sample": false,
+ "class_embed_type": null,
+ "class_embeddings_concat": false,
+ "conv_in_kernel": 3,
+ "conv_out_kernel": 3,
+ "cross_attention_dim": 256,
+ "cross_attention_norm": null,
+ "decay": 0.9999,
+ "down_block_types": [
+ "DownBlock2D",
+ "DownBlock2D",
+ "DownBlock2D",
+ "CrossAttnDownBlock2D"
+ ],
+ "downsample_padding": 1,
+ "dual_cross_attention": false,
+ "encoder_hid_dim": null,
+ "flip_sin_to_cos": true,
+ "freq_shift": 0,
+ "in_channels": 3,
+ "inv_gamma": 1.0,
+ "layers_per_block": 2,
+ "mid_block_only_cross_attention": null,
+ "mid_block_scale_factor": 1,
+ "mid_block_type": "UNetMidBlock2DCrossAttn",
+ "min_decay": 0.0,
+ "norm_eps": 1e-05,
+ "norm_num_groups": 32,
+ "num_class_embeds": null,
+ "only_cross_attention": false,
+ "optimization_step": 118000,
+ "out_channels": 3,
+ "power": 0.75,
+ "projection_class_embeddings_input_dim": null,
+ "resnet_out_scale_factor": 1.0,
+ "resnet_skip_time_act": false,
+ "resnet_time_scale_shift": "default",
+ "sample_size": 256,
+ "time_cond_proj_dim": null,
+ "time_embedding_act_fn": null,
+ "time_embedding_dim": null,
+ "time_embedding_type": "positional",
+ "timestep_post_act": null,
+ "up_block_types": [
+ "CrossAttnUpBlock2D",
+ "UpBlock2D",
+ "UpBlock2D",
+ "UpBlock2D"
+ ],
+ "upcast_attention": false,
+ "update_after_step": 0,
+ "use_ema_warmup": true,
+ "use_linear_projection": false
+}
diff --git a/checkpoint-118000/unet_ema/diffusion_pytorch_model.bin b/checkpoint-118000/unet_ema/diffusion_pytorch_model.bin
new file mode 100644
index 0000000000000000000000000000000000000000..0f0f93ed6f4c2b2ebab16d6ca6b915289cfd152b
--- /dev/null
+++ b/checkpoint-118000/unet_ema/diffusion_pytorch_model.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:53300312bb2ec0679ab718a6ca532e3a3bf13e9b16ff4783e22b494835f1ca26
+size 36616081
diff --git a/checkpoint-118500/optimizer.bin b/checkpoint-118500/optimizer.bin
new file mode 100644
index 0000000000000000000000000000000000000000..a299c1507b43beae663bd7776d141dd039f9b30d
--- /dev/null
+++ b/checkpoint-118500/optimizer.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:16051c79cd27d6c694cecad09e7d2c5e2fb5d08ecd74759de98d73b530441458
+size 73251413
diff --git a/checkpoint-118500/random_states_0.pkl b/checkpoint-118500/random_states_0.pkl
new file mode 100644
index 0000000000000000000000000000000000000000..9bc75cbe9127fec771ab79e461c6a551ba08a5a5
--- /dev/null
+++ b/checkpoint-118500/random_states_0.pkl
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b9531ab4e1175f4bd57dff933350166ccc3347eb3fb5cb35ef916d031ba9067e
+size 14663
diff --git a/checkpoint-118500/scaler.pt b/checkpoint-118500/scaler.pt
new file mode 100644
index 0000000000000000000000000000000000000000..3e0b1836b25be3620bcb0219929b355572e7019e
--- /dev/null
+++ b/checkpoint-118500/scaler.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:305b6681fdbc1b6edd6d69d2ead1fa2dd69eea1abe2421ad7c3c082b759f4a3d
+size 557
diff --git a/checkpoint-118500/scheduler.bin b/checkpoint-118500/scheduler.bin
new file mode 100644
index 0000000000000000000000000000000000000000..bddf7e858bf70439e9e4fc45b54d1dd359c30a51
--- /dev/null
+++ b/checkpoint-118500/scheduler.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:25517a33ca37d91a0c5ba8d1f77a09f6bb365336c1c943153f358fba1eb66672
+size 563
diff --git a/checkpoint-118500/unet/config.json b/checkpoint-118500/unet/config.json
new file mode 100644
index 0000000000000000000000000000000000000000..bb1cd664fc36d4dc821a4f0fdae4b94d7d99b356
--- /dev/null
+++ b/checkpoint-118500/unet/config.json
@@ -0,0 +1,60 @@
+{
+ "_class_name": "UNet2DConditionModel",
+ "_diffusers_version": "0.16.1",
+ "act_fn": "silu",
+ "addition_embed_type": null,
+ "addition_embed_type_num_heads": 64,
+ "attention_head_dim": 8,
+ "block_out_channels": [
+ 64,
+ 64,
+ 128,
+ 128
+ ],
+ "center_input_sample": false,
+ "class_embed_type": null,
+ "class_embeddings_concat": false,
+ "conv_in_kernel": 3,
+ "conv_out_kernel": 3,
+ "cross_attention_dim": 256,
+ "cross_attention_norm": null,
+ "down_block_types": [
+ "DownBlock2D",
+ "DownBlock2D",
+ "DownBlock2D",
+ "CrossAttnDownBlock2D"
+ ],
+ "downsample_padding": 1,
+ "dual_cross_attention": false,
+ "encoder_hid_dim": null,
+ "flip_sin_to_cos": true,
+ "freq_shift": 0,
+ "in_channels": 3,
+ "layers_per_block": 2,
+ "mid_block_only_cross_attention": null,
+ "mid_block_scale_factor": 1,
+ "mid_block_type": "UNetMidBlock2DCrossAttn",
+ "norm_eps": 1e-05,
+ "norm_num_groups": 32,
+ "num_class_embeds": null,
+ "only_cross_attention": false,
+ "out_channels": 3,
+ "projection_class_embeddings_input_dim": null,
+ "resnet_out_scale_factor": 1.0,
+ "resnet_skip_time_act": false,
+ "resnet_time_scale_shift": "default",
+ "sample_size": 256,
+ "time_cond_proj_dim": null,
+ "time_embedding_act_fn": null,
+ "time_embedding_dim": null,
+ "time_embedding_type": "positional",
+ "timestep_post_act": null,
+ "up_block_types": [
+ "CrossAttnUpBlock2D",
+ "UpBlock2D",
+ "UpBlock2D",
+ "UpBlock2D"
+ ],
+ "upcast_attention": false,
+ "use_linear_projection": false
+}
diff --git a/checkpoint-118500/unet/diffusion_pytorch_model.bin b/checkpoint-118500/unet/diffusion_pytorch_model.bin
new file mode 100644
index 0000000000000000000000000000000000000000..74f7b1a876894425a5ae27cac4c60491349629b9
--- /dev/null
+++ b/checkpoint-118500/unet/diffusion_pytorch_model.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0ef0802c572de91e8be6c5c7e8a211319703a466578def4a1551879af232b352
+size 36621969
diff --git a/checkpoint-118500/unet_ema/config.json b/checkpoint-118500/unet_ema/config.json
new file mode 100644
index 0000000000000000000000000000000000000000..d9689b2e236e9b154ecf4f58f770f9ce3e348d56
--- /dev/null
+++ b/checkpoint-118500/unet_ema/config.json
@@ -0,0 +1,67 @@
+{
+ "_class_name": "UNet2DConditionModel",
+ "_diffusers_version": "0.16.1",
+ "act_fn": "silu",
+ "addition_embed_type": null,
+ "addition_embed_type_num_heads": 64,
+ "attention_head_dim": 8,
+ "block_out_channels": [
+ 64,
+ 64,
+ 128,
+ 128
+ ],
+ "center_input_sample": false,
+ "class_embed_type": null,
+ "class_embeddings_concat": false,
+ "conv_in_kernel": 3,
+ "conv_out_kernel": 3,
+ "cross_attention_dim": 256,
+ "cross_attention_norm": null,
+ "decay": 0.9999,
+ "down_block_types": [
+ "DownBlock2D",
+ "DownBlock2D",
+ "DownBlock2D",
+ "CrossAttnDownBlock2D"
+ ],
+ "downsample_padding": 1,
+ "dual_cross_attention": false,
+ "encoder_hid_dim": null,
+ "flip_sin_to_cos": true,
+ "freq_shift": 0,
+ "in_channels": 3,
+ "inv_gamma": 1.0,
+ "layers_per_block": 2,
+ "mid_block_only_cross_attention": null,
+ "mid_block_scale_factor": 1,
+ "mid_block_type": "UNetMidBlock2DCrossAttn",
+ "min_decay": 0.0,
+ "norm_eps": 1e-05,
+ "norm_num_groups": 32,
+ "num_class_embeds": null,
+ "only_cross_attention": false,
+ "optimization_step": 118500,
+ "out_channels": 3,
+ "power": 0.75,
+ "projection_class_embeddings_input_dim": null,
+ "resnet_out_scale_factor": 1.0,
+ "resnet_skip_time_act": false,
+ "resnet_time_scale_shift": "default",
+ "sample_size": 256,
+ "time_cond_proj_dim": null,
+ "time_embedding_act_fn": null,
+ "time_embedding_dim": null,
+ "time_embedding_type": "positional",
+ "timestep_post_act": null,
+ "up_block_types": [
+ "CrossAttnUpBlock2D",
+ "UpBlock2D",
+ "UpBlock2D",
+ "UpBlock2D"
+ ],
+ "upcast_attention": false,
+ "update_after_step": 0,
+ "use_ema_warmup": true,
+ "use_linear_projection": false
+}
diff --git a/checkpoint-118500/unet_ema/diffusion_pytorch_model.bin b/checkpoint-118500/unet_ema/diffusion_pytorch_model.bin
new file mode 100644
index 0000000000000000000000000000000000000000..fd09e04d126bbc7ffc22694756497316b2f8c81f
--- /dev/null
+++ b/checkpoint-118500/unet_ema/diffusion_pytorch_model.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:51f06851a258e2def8f3358b8b495ad3ceb94c617308bd02722d2fae850c9d57
+size 36616081
diff --git a/checkpoint-119000/optimizer.bin b/checkpoint-119000/optimizer.bin
new file mode 100644
index 0000000000000000000000000000000000000000..c22520625fb74470122da7b9784fcd9b7ee1834b
--- /dev/null
+++ b/checkpoint-119000/optimizer.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e3b61e4304780fac1337e967f248470e292bc8d63ba383274aa374428e7f1b4f
+size 73251413
diff --git a/checkpoint-119000/random_states_0.pkl b/checkpoint-119000/random_states_0.pkl
new file mode 100644
index 0000000000000000000000000000000000000000..d23a2bb7a6df32d9569b0315ac5b93a75276eff6
--- /dev/null
+++ b/checkpoint-119000/random_states_0.pkl
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0e59137566e657f8bc648e220583eb26da257ebcd83cbd2de89e25356344154c
+size 14663
diff --git a/checkpoint-119000/scaler.pt b/checkpoint-119000/scaler.pt
new file mode 100644
index 0000000000000000000000000000000000000000..f2e97b92a1949062804a4a82ab0fb05af095da8d
--- /dev/null
+++ b/checkpoint-119000/scaler.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ae9b77663a2c7ae261324cf4554f07218861423fb7d718e1ac876fbf1bb436ea
+size 557
diff --git a/checkpoint-119000/scheduler.bin b/checkpoint-119000/scheduler.bin
new file mode 100644
index 0000000000000000000000000000000000000000..0defa4e48b4d7bccea10215717a217c85c1e9ee8
--- /dev/null
+++ b/checkpoint-119000/scheduler.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7a22e78e4ef79f09d56f647fa4dfacac114f10905fd97ccd2579dac15ea1c981
+size 563
diff --git a/checkpoint-119000/unet/config.json b/checkpoint-119000/unet/config.json
new file mode 100644
index 0000000000000000000000000000000000000000..bb1cd664fc36d4dc821a4f0fdae4b94d7d99b356
--- /dev/null
+++ b/checkpoint-119000/unet/config.json
@@ -0,0 +1,60 @@
+{
+ "_class_name": "UNet2DConditionModel",
+ "_diffusers_version": "0.16.1",
+ "act_fn": "silu",
+ "addition_embed_type": null,
+ "addition_embed_type_num_heads": 64,
+ "attention_head_dim": 8,
+ "block_out_channels": [
+ 64,
+ 64,
+ 128,
+ 128
+ ],
+ "center_input_sample": false,
+ "class_embed_type": null,
+ "class_embeddings_concat": false,
+ "conv_in_kernel": 3,
+ "conv_out_kernel": 3,
+ "cross_attention_dim": 256,
+ "cross_attention_norm": null,
+ "down_block_types": [
+ "DownBlock2D",
+ "DownBlock2D",
+ "DownBlock2D",
+ "CrossAttnDownBlock2D"
+ ],
+ "downsample_padding": 1,
+ "dual_cross_attention": false,
+ "encoder_hid_dim": null,
+ "flip_sin_to_cos": true,
+ "freq_shift": 0,
+ "in_channels": 3,
+ "layers_per_block": 2,
+ "mid_block_only_cross_attention": null,
+ "mid_block_scale_factor": 1,
+ "mid_block_type": "UNetMidBlock2DCrossAttn",
+ "norm_eps": 1e-05,
+ "norm_num_groups": 32,
+ "num_class_embeds": null,
+ "only_cross_attention": false,
+ "out_channels": 3,
+ "projection_class_embeddings_input_dim": null,
+ "resnet_out_scale_factor": 1.0,
+ "resnet_skip_time_act": false,
+ "resnet_time_scale_shift": "default",
+ "sample_size": 256,
+ "time_cond_proj_dim": null,
+ "time_embedding_act_fn": null,
+ "time_embedding_dim": null,
+ "time_embedding_type": "positional",
+ "timestep_post_act": null,
+ "up_block_types": [
+ "CrossAttnUpBlock2D",
+ "UpBlock2D",
+ "UpBlock2D",
+ "UpBlock2D"
+ ],
+ "upcast_attention": false,
+ "use_linear_projection": false
+}
diff --git a/checkpoint-119000/unet/diffusion_pytorch_model.bin b/checkpoint-119000/unet/diffusion_pytorch_model.bin
new file mode 100644
index 0000000000000000000000000000000000000000..1d8b9e98a3e62be55f9e7b1a66a81bb96bf6cf91
--- /dev/null
+++ b/checkpoint-119000/unet/diffusion_pytorch_model.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:315ef57d04a611519a7420b04573eae2fbfe9a7b3cac7b4557ee231fdaa69e6b
+size 36621969
diff --git a/checkpoint-119000/unet_ema/config.json b/checkpoint-119000/unet_ema/config.json
new file mode 100644
index 0000000000000000000000000000000000000000..549acd94963e061c47a3e093319dadaba5e29e2d
--- /dev/null
+++ b/checkpoint-119000/unet_ema/config.json
@@ -0,0 +1,67 @@
+{
+ "_class_name": "UNet2DConditionModel",
+ "_diffusers_version": "0.16.1",
+ "act_fn": "silu",
+ "addition_embed_type": null,
+ "addition_embed_type_num_heads": 64,
+ "attention_head_dim": 8,
+ "block_out_channels": [
+ 64,
+ 64,
+ 128,
+ 128
+ ],
+ "center_input_sample": false,
+ "class_embed_type": null,
+ "class_embeddings_concat": false,
+ "conv_in_kernel": 3,
+ "conv_out_kernel": 3,
+ "cross_attention_dim": 256,
+ "cross_attention_norm": null,
+ "decay": 0.9999,
+ "down_block_types": [
+ "DownBlock2D",
+ "DownBlock2D",
+ "DownBlock2D",
+ "CrossAttnDownBlock2D"
+ ],
+ "downsample_padding": 1,
+ "dual_cross_attention": false,
+ "encoder_hid_dim": null,
+ "flip_sin_to_cos": true,
+ "freq_shift": 0,
+ "in_channels": 3,
+ "inv_gamma": 1.0,
+ "layers_per_block": 2,
+ "mid_block_only_cross_attention": null,
+ "mid_block_scale_factor": 1,
+ "mid_block_type": "UNetMidBlock2DCrossAttn",
+ "min_decay": 0.0,
+ "norm_eps": 1e-05,
+ "norm_num_groups": 32,
+ "num_class_embeds": null,
+ "only_cross_attention": false,
+ "optimization_step": 119000,
+ "out_channels": 3,
+ "power": 0.75,
+ "projection_class_embeddings_input_dim": null,
+ "resnet_out_scale_factor": 1.0,
+ "resnet_skip_time_act": false,
+ "resnet_time_scale_shift": "default",
+ "sample_size": 256,
+ "time_cond_proj_dim": null,
+ "time_embedding_act_fn": null,
+ "time_embedding_dim": null,
+ "time_embedding_type": "positional",
+ "timestep_post_act": null,
+ "up_block_types": [
+ "CrossAttnUpBlock2D",
+ "UpBlock2D",
+ "UpBlock2D",
+ "UpBlock2D"
+ ],
+ "upcast_attention": false,
+ "update_after_step": 0,
+ "use_ema_warmup": true,
+ "use_linear_projection": false
+}
diff --git a/checkpoint-119000/unet_ema/diffusion_pytorch_model.bin b/checkpoint-119000/unet_ema/diffusion_pytorch_model.bin
new file mode 100644
index 0000000000000000000000000000000000000000..fee883fe9a9e1b9ebbb3f1b03d9c1192f7915888
--- /dev/null
+++ b/checkpoint-119000/unet_ema/diffusion_pytorch_model.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:32b97bc09299d1331a54e9a49b2f286dbc446200c6515458f2f26bb3c254a558
+size 36616081
diff --git a/checkpoint-119500/optimizer.bin b/checkpoint-119500/optimizer.bin
new file mode 100644
index 0000000000000000000000000000000000000000..d216086f6d0d53d7ba134f884a4231ee5cc586fb
--- /dev/null
+++ b/checkpoint-119500/optimizer.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a8c3e0f1c9a399fbcea74fa0e36fc14c92b9dc5a58cd5acfe4e30697efcf5aad
+size 73251413
diff --git a/checkpoint-119500/random_states_0.pkl b/checkpoint-119500/random_states_0.pkl
new file mode 100644
index 0000000000000000000000000000000000000000..85d018ad0f77c0b65a1e277a60b7327277e49a29
--- /dev/null
+++ b/checkpoint-119500/random_states_0.pkl
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:baf28430f42ae1db6356cb062d27cb512e731c08edc40239e1e1bc7b0f6ccf0e
+size 14663
diff --git a/checkpoint-119500/scaler.pt b/checkpoint-119500/scaler.pt
new file mode 100644
index 0000000000000000000000000000000000000000..106faaa9d96c8859c5b7bd6a53778d06e1ac2b86
--- /dev/null
+++ b/checkpoint-119500/scaler.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ed1fff6313ee93c2b0f3390a1f6e0786a2a4fe59912a7dab7de9bfd991f0b4c4
+size 557
diff --git a/checkpoint-119500/scheduler.bin b/checkpoint-119500/scheduler.bin
new file mode 100644
index 0000000000000000000000000000000000000000..5ca6ab2a443cbc48484eb3906eb427fdf4ecf432
--- /dev/null
+++ b/checkpoint-119500/scheduler.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ef41881c15d1879277b0716b035abcf819b6ddfe968214244bca5b34ee125850
+size 563
diff --git a/checkpoint-119500/unet/config.json b/checkpoint-119500/unet/config.json
new file mode 100644
index 0000000000000000000000000000000000000000..bb1cd664fc36d4dc821a4f0fdae4b94d7d99b356
--- /dev/null
+++ b/checkpoint-119500/unet/config.json
@@ -0,0 +1,60 @@
+{
+ "_class_name": "UNet2DConditionModel",
+ "_diffusers_version": "0.16.1",
+ "act_fn": "silu",
+ "addition_embed_type": null,
+ "addition_embed_type_num_heads": 64,
+ "attention_head_dim": 8,
+ "block_out_channels": [
+ 64,
+ 64,
+ 128,
+ 128
+ ],
+ "center_input_sample": false,
+ "class_embed_type": null,
+ "class_embeddings_concat": false,
+ "conv_in_kernel": 3,
+ "conv_out_kernel": 3,
+ "cross_attention_dim": 256,
+ "cross_attention_norm": null,
+ "down_block_types": [
+ "DownBlock2D",
+ "DownBlock2D",
+ "DownBlock2D",
+ "CrossAttnDownBlock2D"
+ ],
+ "downsample_padding": 1,
+ "dual_cross_attention": false,
+ "encoder_hid_dim": null,
+ "flip_sin_to_cos": true,
+ "freq_shift": 0,
+ "in_channels": 3,
+ "layers_per_block": 2,
+ "mid_block_only_cross_attention": null,
+ "mid_block_scale_factor": 1,
+ "mid_block_type": "UNetMidBlock2DCrossAttn",
+ "norm_eps": 1e-05,
+ "norm_num_groups": 32,
+ "num_class_embeds": null,
+ "only_cross_attention": false,
+ "out_channels": 3,
+ "projection_class_embeddings_input_dim": null,
+ "resnet_out_scale_factor": 1.0,
+ "resnet_skip_time_act": false,
+ "resnet_time_scale_shift": "default",
+ "sample_size": 256,
+ "time_cond_proj_dim": null,
+ "time_embedding_act_fn": null,
+ "time_embedding_dim": null,
+ "time_embedding_type": "positional",
+ "timestep_post_act": null,
+ "up_block_types": [
+ "CrossAttnUpBlock2D",
+ "UpBlock2D",
+ "UpBlock2D",
+ "UpBlock2D"
+ ],
+ "upcast_attention": false,
+ "use_linear_projection": false
+}
diff --git a/checkpoint-119500/unet/diffusion_pytorch_model.bin b/checkpoint-119500/unet/diffusion_pytorch_model.bin
new file mode 100644
index 0000000000000000000000000000000000000000..5af8d691b33f68abc2787bf31e1d004b9b461099
--- /dev/null
+++ b/checkpoint-119500/unet/diffusion_pytorch_model.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c42cd1f1eea4718edac95903a3e2d9c88a92f7e7cf7119217b48a483d441f4e5
+size 36621969
diff --git a/checkpoint-119500/unet_ema/config.json b/checkpoint-119500/unet_ema/config.json
new file mode 100644
index 0000000000000000000000000000000000000000..be3db0383ac73630e103d1e8912795949ed9eca6
--- /dev/null
+++ b/checkpoint-119500/unet_ema/config.json
@@ -0,0 +1,67 @@
+{
+ "_class_name": "UNet2DConditionModel",
+ "_diffusers_version": "0.16.1",
+ "act_fn": "silu",
+ "addition_embed_type": null,
+ "addition_embed_type_num_heads": 64,
+ "attention_head_dim": 8,
+ "block_out_channels": [
+ 64,
+ 64,
+ 128,
+ 128
+ ],
+ "center_input_sample": false,
+ "class_embed_type": null,
+ "class_embeddings_concat": false,
+ "conv_in_kernel": 3,
+ "conv_out_kernel": 3,
+ "cross_attention_dim": 256,
+ "cross_attention_norm": null,
+ "decay": 0.9999,
+ "down_block_types": [
+ "DownBlock2D",
+ "DownBlock2D",
+ "DownBlock2D",
+ "CrossAttnDownBlock2D"
+ ],
+ "downsample_padding": 1,
+ "dual_cross_attention": false,
+ "encoder_hid_dim": null,
+ "flip_sin_to_cos": true,
+ "freq_shift": 0,
+ "in_channels": 3,
+ "inv_gamma": 1.0,
+ "layers_per_block": 2,
+ "mid_block_only_cross_attention": null,
+ "mid_block_scale_factor": 1,
+ "mid_block_type": "UNetMidBlock2DCrossAttn",
+ "min_decay": 0.0,
+ "norm_eps": 1e-05,
+ "norm_num_groups": 32,
+ "num_class_embeds": null,
+ "only_cross_attention": false,
+ "optimization_step": 119500,
+ "out_channels": 3,
+ "power": 0.75,
+ "projection_class_embeddings_input_dim": null,
+ "resnet_out_scale_factor": 1.0,
+ "resnet_skip_time_act": false,
+ "resnet_time_scale_shift": "default",
+ "sample_size": 256,
+ "time_cond_proj_dim": null,
+ "time_embedding_act_fn": null,
+ "time_embedding_dim": null,
+ "time_embedding_type": "positional",
+ "timestep_post_act": null,
+ "up_block_types": [
+ "CrossAttnUpBlock2D",
+ "UpBlock2D",
+ "UpBlock2D",
+ "UpBlock2D"
+ ],
+ "upcast_attention": false,
+ "update_after_step": 0,
+ "use_ema_warmup": true,
+ "use_linear_projection": false
+}
diff --git a/checkpoint-119500/unet_ema/diffusion_pytorch_model.bin b/checkpoint-119500/unet_ema/diffusion_pytorch_model.bin
new file mode 100644
index 0000000000000000000000000000000000000000..f3635e469bf1db542237283d2b5db59ec1346ee4
--- /dev/null
+++ b/checkpoint-119500/unet_ema/diffusion_pytorch_model.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b67833a070508c08cb66abd925b14c115fec88879fce67562cd1b4e2e0ea018d
+size 36616081
diff --git a/checkpoint-120000/optimizer.bin b/checkpoint-120000/optimizer.bin
new file mode 100644
index 0000000000000000000000000000000000000000..befdcd814ca66d6bbb4ac5a76ee739bbdbbd1463
--- /dev/null
+++ b/checkpoint-120000/optimizer.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7bf96d05519e30e123dc0024b711fc42d63e2c763921ce89108bcc5652c49fe2
+size 73251413
diff --git a/checkpoint-120000/random_states_0.pkl b/checkpoint-120000/random_states_0.pkl
new file mode 100644
index 0000000000000000000000000000000000000000..41d8eb8a5959dcab24988881f453c5f903302b02
--- /dev/null
+++ b/checkpoint-120000/random_states_0.pkl
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f9b3eeb7dbb5df12e7f4b87ce71c8d91aeaff406fc432ae1f4089140df8cf027
+size 14663
diff --git a/checkpoint-120000/scaler.pt b/checkpoint-120000/scaler.pt
new file mode 100644
index 0000000000000000000000000000000000000000..ecec5c25c16e1b87869434164ab2f3223e626b6c
--- /dev/null
+++ b/checkpoint-120000/scaler.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ba6b35e7bfa7d052f7e49fc4461f6ba9e8b4f18a188ac468dbb55882c7940787
+size 557
diff --git a/checkpoint-120000/scheduler.bin b/checkpoint-120000/scheduler.bin
new file mode 100644
index 0000000000000000000000000000000000000000..cbda0d71e010acedc7d611ddeab9868379803e6a
--- /dev/null
+++ b/checkpoint-120000/scheduler.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:66b8420e6b804161ff637c619535e729c9b0def11eeab7b5689ddb8bc5b246f1
+size 563
diff --git a/checkpoint-120000/unet/config.json b/checkpoint-120000/unet/config.json
new file mode 100644
index 0000000000000000000000000000000000000000..bb1cd664fc36d4dc821a4f0fdae4b94d7d99b356
--- /dev/null
+++ b/checkpoint-120000/unet/config.json
@@ -0,0 +1,60 @@
+{
+ "_class_name": "UNet2DConditionModel",
+ "_diffusers_version": "0.16.1",
+ "act_fn": "silu",
+ "addition_embed_type": null,
+ "addition_embed_type_num_heads": 64,
+ "attention_head_dim": 8,
+ "block_out_channels": [
+ 64,
+ 64,
+ 128,
+ 128
+ ],
+ "center_input_sample": false,
+ "class_embed_type": null,
+ "class_embeddings_concat": false,
+ "conv_in_kernel": 3,
+ "conv_out_kernel": 3,
+ "cross_attention_dim": 256,
+ "cross_attention_norm": null,
+ "down_block_types": [
+ "DownBlock2D",
+ "DownBlock2D",
+ "DownBlock2D",
+ "CrossAttnDownBlock2D"
+ ],
+ "downsample_padding": 1,
+ "dual_cross_attention": false,
+ "encoder_hid_dim": null,
+ "flip_sin_to_cos": true,
+ "freq_shift": 0,
+ "in_channels": 3,
+ "layers_per_block": 2,
+ "mid_block_only_cross_attention": null,
+ "mid_block_scale_factor": 1,
+ "mid_block_type": "UNetMidBlock2DCrossAttn",
+ "norm_eps": 1e-05,
+ "norm_num_groups": 32,
+ "num_class_embeds": null,
+ "only_cross_attention": false,
+ "out_channels": 3,
+ "projection_class_embeddings_input_dim": null,
+ "resnet_out_scale_factor": 1.0,
+ "resnet_skip_time_act": false,
+ "resnet_time_scale_shift": "default",
+ "sample_size": 256,
+ "time_cond_proj_dim": null,
+ "time_embedding_act_fn": null,
+ "time_embedding_dim": null,
+ "time_embedding_type": "positional",
+ "timestep_post_act": null,
+ "up_block_types": [
+ "CrossAttnUpBlock2D",
+ "UpBlock2D",
+ "UpBlock2D",
+ "UpBlock2D"
+ ],
+ "upcast_attention": false,
+ "use_linear_projection": false
+}
diff --git a/checkpoint-120000/unet/diffusion_pytorch_model.bin b/checkpoint-120000/unet/diffusion_pytorch_model.bin
new file mode 100644
index 0000000000000000000000000000000000000000..251ac891bcc6cb8422c46afa5fd155e911e55131
--- /dev/null
+++ b/checkpoint-120000/unet/diffusion_pytorch_model.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:837c11d2542e62b33adea354f1fad8f45a0c9289d6b4b793d7b2cfa033f01b6d
+size 36621969
diff --git a/checkpoint-120000/unet_ema/config.json b/checkpoint-120000/unet_ema/config.json
new file mode 100644
index 0000000000000000000000000000000000000000..a50c7d5307bdf50ffa1fdb5b94dd7b4f1044ccce
--- /dev/null
+++ b/checkpoint-120000/unet_ema/config.json
@@ -0,0 +1,67 @@
+{
+ "_class_name": "UNet2DConditionModel",
+ "_diffusers_version": "0.16.1",
+ "act_fn": "silu",
+ "addition_embed_type": null,
+ "addition_embed_type_num_heads": 64,
+ "attention_head_dim": 8,
+ "block_out_channels": [
+ 64,
+ 64,
+ 128,
+ 128
+ ],
+ "center_input_sample": false,
+ "class_embed_type": null,
+ "class_embeddings_concat": false,
+ "conv_in_kernel": 3,
+ "conv_out_kernel": 3,
+ "cross_attention_dim": 256,
+ "cross_attention_norm": null,
+ "decay": 0.9999,
+ "down_block_types": [
+ "DownBlock2D",
+ "DownBlock2D",
+ "DownBlock2D",
+ "CrossAttnDownBlock2D"
+ ],
+ "downsample_padding": 1,
+ "dual_cross_attention": false,
+ "encoder_hid_dim": null,
+ "flip_sin_to_cos": true,
+ "freq_shift": 0,
+ "in_channels": 3,
+ "inv_gamma": 1.0,
+ "layers_per_block": 2,
+ "mid_block_only_cross_attention": null,
+ "mid_block_scale_factor": 1,
+ "mid_block_type": "UNetMidBlock2DCrossAttn",
+ "min_decay": 0.0,
+ "norm_eps": 1e-05,
+ "norm_num_groups": 32,
+ "num_class_embeds": null,
+ "only_cross_attention": false,
+ "optimization_step": 120000,
+ "out_channels": 3,
+ "power": 0.75,
+ "projection_class_embeddings_input_dim": null,
+ "resnet_out_scale_factor": 1.0,
+ "resnet_skip_time_act": false,
+ "resnet_time_scale_shift": "default",
+ "sample_size": 256,
+ "time_cond_proj_dim": null,
+ "time_embedding_act_fn": null,
+ "time_embedding_dim": null,
+ "time_embedding_type": "positional",
+ "timestep_post_act": null,
+ "up_block_types": [
+ "CrossAttnUpBlock2D",
+ "UpBlock2D",
+ "UpBlock2D",
+ "UpBlock2D"
+ ],
+ "upcast_attention": false,
+ "update_after_step": 0,
+ "use_ema_warmup": true,
+ "use_linear_projection": false
+}
diff --git a/checkpoint-120000/unet_ema/diffusion_pytorch_model.bin b/checkpoint-120000/unet_ema/diffusion_pytorch_model.bin
new file mode 100644
index 0000000000000000000000000000000000000000..782e0aa3f3f12f492c6eb921d353125fefe81bfc
--- /dev/null
+++ b/checkpoint-120000/unet_ema/diffusion_pytorch_model.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2dd37ddc0bbd255659f6d78e9584d38628375be0d5b517cb31ab5c470e367986
+size 36616081
diff --git a/checkpoint-120500/optimizer.bin b/checkpoint-120500/optimizer.bin
new file mode 100644
index 0000000000000000000000000000000000000000..69b4306dcb1a15a9379b293a61117fdd432c23b5
--- /dev/null
+++ b/checkpoint-120500/optimizer.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c3a4771b6b614fb61471f1fcb9a6a08e22bd63e00993e3aee75e801749a31808
+size 73251413
diff --git a/checkpoint-120500/random_states_0.pkl b/checkpoint-120500/random_states_0.pkl
new file mode 100644
index 0000000000000000000000000000000000000000..848ad09dc210128536c1030b8ae64f87fe747a55
--- /dev/null
+++ b/checkpoint-120500/random_states_0.pkl
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:010b221e776ab846f4b9b8563b0ac33e4a4c096335e09107caf6e192437072e5
+size 14663
diff --git a/checkpoint-120500/scaler.pt b/checkpoint-120500/scaler.pt
new file mode 100644
index 0000000000000000000000000000000000000000..d8d063c82d71aa40d3b42d0513321e245cbeb158
--- /dev/null
+++ b/checkpoint-120500/scaler.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5524c626bb1e14d4ebfdf68e8a45796bd8b8d4bdad258be6e56507a3589ecc27
+size 557
diff --git a/checkpoint-120500/scheduler.bin b/checkpoint-120500/scheduler.bin
new file mode 100644
index 0000000000000000000000000000000000000000..c6b4054c49e17e484a364fe1daec83abc5b68ab2
--- /dev/null
+++ b/checkpoint-120500/scheduler.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:19eb4bc4044021d4f1448991c672003d40a632a0bfb4b4f57b5519559b703a55
+size 563
diff --git a/checkpoint-120500/unet/config.json b/checkpoint-120500/unet/config.json
new file mode 100644
index 0000000000000000000000000000000000000000..bb1cd664fc36d4dc821a4f0fdae4b94d7d99b356
--- /dev/null
+++ b/checkpoint-120500/unet/config.json
@@ -0,0 +1,60 @@
+{
+ "_class_name": "UNet2DConditionModel",
+ "_diffusers_version": "0.16.1",
+ "act_fn": "silu",
+ "addition_embed_type": null,
+ "addition_embed_type_num_heads": 64,
+ "attention_head_dim": 8,
+ "block_out_channels": [
+ 64,
+ 64,
+ 128,
+ 128
+ ],
+ "center_input_sample": false,
+ "class_embed_type": null,
+ "class_embeddings_concat": false,
+ "conv_in_kernel": 3,
+ "conv_out_kernel": 3,
+ "cross_attention_dim": 256,
+ "cross_attention_norm": null,
+ "down_block_types": [
+ "DownBlock2D",
+ "DownBlock2D",
+ "DownBlock2D",
+ "CrossAttnDownBlock2D"
+ ],
+ "downsample_padding": 1,
+ "dual_cross_attention": false,
+ "encoder_hid_dim": null,
+ "flip_sin_to_cos": true,
+ "freq_shift": 0,
+ "in_channels": 3,
+ "layers_per_block": 2,
+ "mid_block_only_cross_attention": null,
+ "mid_block_scale_factor": 1,
+ "mid_block_type": "UNetMidBlock2DCrossAttn",
+ "norm_eps": 1e-05,
+ "norm_num_groups": 32,
+ "num_class_embeds": null,
+ "only_cross_attention": false,
+ "out_channels": 3,
+ "projection_class_embeddings_input_dim": null,
+ "resnet_out_scale_factor": 1.0,
+ "resnet_skip_time_act": false,
+ "resnet_time_scale_shift": "default",
+ "sample_size": 256,
+ "time_cond_proj_dim": null,
+ "time_embedding_act_fn": null,
+ "time_embedding_dim": null,
+ "time_embedding_type": "positional",
+ "timestep_post_act": null,
+ "up_block_types": [
+ "CrossAttnUpBlock2D",
+ "UpBlock2D",
+ "UpBlock2D",
+ "UpBlock2D"
+ ],
+ "upcast_attention": false,
+ "use_linear_projection": false
+}
diff --git a/checkpoint-120500/unet/diffusion_pytorch_model.bin b/checkpoint-120500/unet/diffusion_pytorch_model.bin
new file mode 100644
index 0000000000000000000000000000000000000000..f414038b87dff51cf7940e35154c4a415cdf94bb
--- /dev/null
+++ b/checkpoint-120500/unet/diffusion_pytorch_model.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e4d41aaab34a9fed5a5dce23834cbe284c5aa807e5cbf7877b6b0eb70f6a0e66
+size 36621969
diff --git a/checkpoint-120500/unet_ema/config.json b/checkpoint-120500/unet_ema/config.json
new file mode 100644
index 0000000000000000000000000000000000000000..cbf17bb5340ab6494f88a2b0568a2593e1d6a2c1
--- /dev/null
+++ b/checkpoint-120500/unet_ema/config.json
@@ -0,0 +1,67 @@
+{
+ "_class_name": "UNet2DConditionModel",
+ "_diffusers_version": "0.16.1",
+ "act_fn": "silu",
+ "addition_embed_type": null,
+ "addition_embed_type_num_heads": 64,
+ "attention_head_dim": 8,
+ "block_out_channels": [
+ 64,
+ 64,
+ 128,
+ 128
+ ],
+ "center_input_sample": false,
+ "class_embed_type": null,
+ "class_embeddings_concat": false,
+ "conv_in_kernel": 3,
+ "conv_out_kernel": 3,
+ "cross_attention_dim": 256,
+ "cross_attention_norm": null,
+ "decay": 0.9999,
+ "down_block_types": [
+ "DownBlock2D",
+ "DownBlock2D",
+ "DownBlock2D",
+ "CrossAttnDownBlock2D"
+ ],
+ "downsample_padding": 1,
+ "dual_cross_attention": false,
+ "encoder_hid_dim": null,
+ "flip_sin_to_cos": true,
+ "freq_shift": 0,
+ "in_channels": 3,
+ "inv_gamma": 1.0,
+ "layers_per_block": 2,
+ "mid_block_only_cross_attention": null,
+ "mid_block_scale_factor": 1,
+ "mid_block_type": "UNetMidBlock2DCrossAttn",
+ "min_decay": 0.0,
+ "norm_eps": 1e-05,
+ "norm_num_groups": 32,
+ "num_class_embeds": null,
+ "only_cross_attention": false,
+ "optimization_step": 120500,
+ "out_channels": 3,
+ "power": 0.75,
+ "projection_class_embeddings_input_dim": null,
+ "resnet_out_scale_factor": 1.0,
+ "resnet_skip_time_act": false,
+ "resnet_time_scale_shift": "default",
+ "sample_size": 256,
+ "time_cond_proj_dim": null,
+ "time_embedding_act_fn": null,
+ "time_embedding_dim": null,
+ "time_embedding_type": "positional",
+ "timestep_post_act": null,
+ "up_block_types": [
+ "CrossAttnUpBlock2D",
+ "UpBlock2D",
+ "UpBlock2D",
+ "UpBlock2D"
+ ],
+ "upcast_attention": false,
+ "update_after_step": 0,
+ "use_ema_warmup": true,
+ "use_linear_projection": false
+}
diff --git a/checkpoint-120500/unet_ema/diffusion_pytorch_model.bin b/checkpoint-120500/unet_ema/diffusion_pytorch_model.bin
new file mode 100644
index 0000000000000000000000000000000000000000..d5a15d098d0ce8037ab2544f13eb735c77d05ec5
--- /dev/null
+++ b/checkpoint-120500/unet_ema/diffusion_pytorch_model.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4298c3dbf1149d55458a3eea245fe0ba95c5436c7c550c957a16874de0f1cc01
+size 36616081
diff --git a/unet/diffusion_pytorch_model.bin b/unet/diffusion_pytorch_model.bin
index f7959cb731304b2ae6ca8e6d934bd87f2fd4d46d..dba1e07a7127f21a2721402a9d2c2dd07d7df135 100644
--- a/unet/diffusion_pytorch_model.bin
+++ b/unet/diffusion_pytorch_model.bin
@@ -1,3 +1,3 @@
version https://git-lfs.github.com/spec/v1
-oid sha256:0a43e79ae2af725e756a7f75e1cd8e0caf60fc7fc2d310db3cd9f6b650da39f8
+oid sha256:1a233d43dd957ce2a35563fed3b73b5788348301ac01c66636478664f157cf73
size 36621969