diff --git a/checkpoint-114000/optimizer.bin b/checkpoint-114000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..3e144f8b0eff3dc27946ab469a36368fa0926e23 --- /dev/null +++ b/checkpoint-114000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:1800b27390804b0ec7191770bb6a27367207c7f7a3e027d40ce880f4b5e3af14 +size 73251413 diff --git a/checkpoint-114000/random_states_0.pkl b/checkpoint-114000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..44efa098221d985e4e1c68c817f8d22d4259b14c --- /dev/null +++ b/checkpoint-114000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:2c906250a7258576bec4b1a08a516cd3a7a9cc34b30bbe8bab25e7a16ae9d317 +size 14663 diff --git a/checkpoint-114000/scaler.pt b/checkpoint-114000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..4bbd9fca5bf9cf4e86e4e5b9790e0a38dbb2ea38 --- /dev/null +++ b/checkpoint-114000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:d69a8aa7959e49ee06e7803e137c57b3fa1573369a17f0278a637abe37d23f19 +size 557 diff --git a/checkpoint-114000/scheduler.bin b/checkpoint-114000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..ca7b4061e85e5ea1ce9bd6f52b1a47b8dcf2e98e --- /dev/null +++ b/checkpoint-114000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:8e3e8afd7d006cd624dd5f37cd3cd1316ea8d9422adffdf992488eb38ee6ed1f +size 563 diff --git a/checkpoint-114000/unet/config.json b/checkpoint-114000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..bb1cd664fc36d4dc821a4f0fdae4b94d7d99b356 --- /dev/null +++ b/checkpoint-114000/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-114000/unet/diffusion_pytorch_model.bin b/checkpoint-114000/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..51f217856cef4b2cfc0339c8bb41e4f7f7024ad5 --- /dev/null +++ b/checkpoint-114000/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:17f1cedf0d976a416b1a2c58ec16d1ac4274f2f0ebe471e0beca1db748ce896d +size 36621969 diff --git a/checkpoint-114000/unet_ema/config.json b/checkpoint-114000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..010b7e278ef7aeafd2d9f66287a80ecd78c6c42e --- /dev/null +++ b/checkpoint-114000/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": 114000, + "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-114000/unet_ema/diffusion_pytorch_model.bin b/checkpoint-114000/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..6668c6c87290d3acca72acb40b077b058c842c87 --- /dev/null +++ b/checkpoint-114000/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:70bf4d902978f224419f4d00a8c325adc91643e6a6148814aeb3eaede02a0b00 +size 36616081 diff --git a/checkpoint-114500/optimizer.bin b/checkpoint-114500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..7acaaef7ad17cb7bdce6a0d0f34c42e0a807d729 --- /dev/null +++ b/checkpoint-114500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:1972f2ce56407ec0e946fd2e8dc5f866320c17b40b30456dae978712eeaa068e +size 73251413 diff --git a/checkpoint-114500/random_states_0.pkl b/checkpoint-114500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..9a7ba766441727b95cb51760fcd7a07bd6ea7e6c --- /dev/null +++ b/checkpoint-114500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:af6a6af00895600bff9001ca96e4799691d119130b99dafc96204d0d6d8f8a16 +size 14663 diff --git a/checkpoint-114500/scaler.pt b/checkpoint-114500/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..6364084ebddaedc5ac1c88b6a13cb9465c870303 --- /dev/null +++ b/checkpoint-114500/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:426f292c93eb965351e12db4ca23d5beb3a986fc8345dc1224210a1367ab5988 +size 557 diff --git a/checkpoint-114500/scheduler.bin b/checkpoint-114500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..ef8f9e065119d709904a35232febbd4da8b31d1c --- /dev/null +++ b/checkpoint-114500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:4864e7d62842d88f705101b34bd32bea6fe93fbe20ebbc93c58d34ab01885959 +size 563 diff --git a/checkpoint-114500/unet/config.json b/checkpoint-114500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..bb1cd664fc36d4dc821a4f0fdae4b94d7d99b356 --- /dev/null +++ b/checkpoint-114500/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-114500/unet/diffusion_pytorch_model.bin b/checkpoint-114500/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..fe9745c2da20553b10988fc76ca2a3f9f0da8c52 --- /dev/null +++ b/checkpoint-114500/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:0d67aa58491e1fa0ecb769df88a7a1ffa3652357eae535281886e9996b644f4b +size 36621969 diff --git a/checkpoint-114500/unet_ema/config.json b/checkpoint-114500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..8b678196230d8e017139405e3c2e4c0836bff534 --- /dev/null +++ b/checkpoint-114500/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": 114500, + "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-114500/unet_ema/diffusion_pytorch_model.bin b/checkpoint-114500/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..75ccdb404fc830bda4be6527c96948994390dd99 --- /dev/null +++ b/checkpoint-114500/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:29cdeb4c200568aa40a93d51c0c5c3b809a6dfa5312d606c152ff9f22d724ce3 +size 36616081 diff --git a/checkpoint-115000/optimizer.bin b/checkpoint-115000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..3542a432a1e86569e16e1b30e2bb6208060c3528 --- /dev/null +++ b/checkpoint-115000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:5a6433c651ebfce07a9c030907d3da59c95d4d810034441855b28627bb7db19b +size 73251413 diff --git a/checkpoint-115000/random_states_0.pkl b/checkpoint-115000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..c2221b6125670c3edf557a3da8efe833707a8466 --- /dev/null +++ b/checkpoint-115000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7cae18f246e1badf3aebd91ded5955e6b5abb76f39839a4b4498adc87563b8f4 +size 14663 diff --git a/checkpoint-115000/scaler.pt b/checkpoint-115000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..fe2e0abe5d89c40eaeaf4e04a028626530e0956f --- /dev/null +++ b/checkpoint-115000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:dc5f4b9960dd730f875cac06f509d8c24f7cda1a596a905e73fd1cbe7a6cc733 +size 557 diff --git a/checkpoint-115000/scheduler.bin b/checkpoint-115000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..e53227cd56be1fefd368b2e02db4d21d89f053be --- /dev/null +++ b/checkpoint-115000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:9e0cea528ebf5bc0c30d5e9db69431edf20fee22eb085861a9b623205175ba08 +size 563 diff --git a/checkpoint-115000/unet/config.json b/checkpoint-115000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..bb1cd664fc36d4dc821a4f0fdae4b94d7d99b356 --- /dev/null +++ b/checkpoint-115000/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-115000/unet/diffusion_pytorch_model.bin b/checkpoint-115000/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..e5d0fa72b4cfc131d678bbdff28351110cbe7e3a --- /dev/null +++ b/checkpoint-115000/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:1759cfb82143cbc523033aee98d518902b7864760e259b34375fd10adb076453 +size 36621969 diff --git a/checkpoint-115000/unet_ema/config.json b/checkpoint-115000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..adcad1465e46008289ae157ee484ea4504abb5cf --- /dev/null +++ b/checkpoint-115000/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": 115000, + "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-115000/unet_ema/diffusion_pytorch_model.bin b/checkpoint-115000/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..93027462343f99796a1096030917cd973520e924 --- /dev/null +++ b/checkpoint-115000/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:6bf815b2b216595e389d0e99fa9f76a93d0f6d503fd52bb0293c6ea80cf054b3 +size 36616081 diff --git a/checkpoint-115500/optimizer.bin b/checkpoint-115500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..ca16bb65829b04c78040fe0f8b45e34e0af295ab --- /dev/null +++ b/checkpoint-115500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:73d2414c9d7a19a6a3c06d3cfda48dab2a685a73e7bb513e974f9ef84a4c75dc +size 73251413 diff --git a/checkpoint-115500/random_states_0.pkl b/checkpoint-115500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..7d309a3bf513cf635e5f2bff32d0625152fd7bf9 --- /dev/null +++ b/checkpoint-115500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c9b7705073d818a2a1e0586a987561e3b6ca70e8e7f82f781334b6c900aef5ea +size 14663 diff --git a/checkpoint-115500/scaler.pt b/checkpoint-115500/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..b1ac6632e8553364b730912a8b5ce1d21b49b74f --- /dev/null +++ b/checkpoint-115500/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:66bd0a4da8ca6a9b50c35cff73b6bf96778c069f1d7bf512eb12c33825a203e8 +size 557 diff --git a/checkpoint-115500/scheduler.bin b/checkpoint-115500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..a445e25ecd1cd31fc5fc17f0dcd60abf9b7e8ea3 --- /dev/null +++ b/checkpoint-115500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:dd677877b89c45a9878121c085cb92570a359993c3ebb045e70e023bd9bff694 +size 563 diff --git a/checkpoint-115500/unet/config.json b/checkpoint-115500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..bb1cd664fc36d4dc821a4f0fdae4b94d7d99b356 --- /dev/null +++ b/checkpoint-115500/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-115500/unet/diffusion_pytorch_model.bin b/checkpoint-115500/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..d483b7bd7de162a0c262b5420059726e01884738 --- /dev/null +++ b/checkpoint-115500/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:ccc5bf620bd06005190687a1257d3b248d2aacd54372e18da9538ea889734671 +size 36621969 diff --git a/checkpoint-115500/unet_ema/config.json b/checkpoint-115500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..b3b3c2da12b35a82106dadea24f429dbefe20107 --- /dev/null +++ b/checkpoint-115500/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": 115500, + "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-115500/unet_ema/diffusion_pytorch_model.bin b/checkpoint-115500/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..5ae8d378ad672f9ac7ba3e2ea9e2215bc59cdfb3 --- /dev/null +++ b/checkpoint-115500/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:680b92e011898a40ab271c78b97185bccf7ce5365859614ac7dfb2f3c9aeaafa +size 36616081 diff --git a/checkpoint-116000/optimizer.bin b/checkpoint-116000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..f24100a15672f1c044fee4b35b60ffdd818c0195 --- /dev/null +++ b/checkpoint-116000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:aebaaf727b5fedf97236be150a8cbe18e3116ce155fb0a5919477fb917756f18 +size 73251413 diff --git a/checkpoint-116000/random_states_0.pkl b/checkpoint-116000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..fa5213edf21cf8cac570805dfc6883b171631889 --- /dev/null +++ b/checkpoint-116000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:a6c8b591dd5a81398d780d464ab10700844730a37f3f154bbe2102a03e005e0a +size 14663 diff --git a/checkpoint-116000/scaler.pt b/checkpoint-116000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..94988ffb9b3bef1f05ec20aca5fed005dd675765 --- /dev/null +++ b/checkpoint-116000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:3f22f39e6a2deaf118267f6f14e2c39d4c15fbf122e97afd5013be0c1daf6c2c +size 557 diff --git a/checkpoint-116000/scheduler.bin b/checkpoint-116000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..c8ad87aa27cf8cbec04ba0c02e75ef8f7af2135d --- /dev/null +++ b/checkpoint-116000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:44ed95e57f6e9877ea0e63b1567162b7417831c9e8957532ff6428fa0a1fb4d4 +size 563 diff --git a/checkpoint-116000/unet/config.json b/checkpoint-116000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..bb1cd664fc36d4dc821a4f0fdae4b94d7d99b356 --- /dev/null +++ b/checkpoint-116000/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-116000/unet/diffusion_pytorch_model.bin b/checkpoint-116000/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..1d4a434f7f93aa51a94b87f051d2c36307780c8b --- /dev/null +++ b/checkpoint-116000/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:bbf957e13076022006b9a276f5e6e2c87d89da34e665c19d3925a24fa0092160 +size 36621969 diff --git a/checkpoint-116000/unet_ema/config.json b/checkpoint-116000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..dece33962b80545d8aebce030f1b1504e73ecea7 --- /dev/null +++ b/checkpoint-116000/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": 116000, + "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-116000/unet_ema/diffusion_pytorch_model.bin b/checkpoint-116000/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..9b033e39a97664dc3990754616caa06d3681da75 --- /dev/null +++ b/checkpoint-116000/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:d46d30cb227dcf6f62a4f008afcdaa3f2323524cd5ba988e471112074df50c84 +size 36616081 diff --git a/checkpoint-116500/optimizer.bin b/checkpoint-116500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..bc77b247992a72a7b2aa795548df20a9d5bf1746 --- /dev/null +++ b/checkpoint-116500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b6766bbf61049ee17bc82be7a4a7141db0c440e730f0347941c0446d1d35ba59 +size 73251413 diff --git a/checkpoint-116500/random_states_0.pkl b/checkpoint-116500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..c6c4429817de4ff33a4efc425682b0f4f43e7bf4 --- /dev/null +++ b/checkpoint-116500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:96193fd43acb03d27c847fccde51a08abd9ad43b64941d00b12b81b564904786 +size 14663 diff --git a/checkpoint-116500/scaler.pt b/checkpoint-116500/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..ae7d0de87b9ce73be066ba7c7c1f3d56ffbeeb3d --- /dev/null +++ b/checkpoint-116500/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:ca9457db2d25f585c3c5686184dfa36e321ea6b6d2fe61dddedd05ed1f05117b +size 557 diff --git a/checkpoint-116500/scheduler.bin b/checkpoint-116500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..b8190302b3c66f11106b9b302c016e5f01fc6734 --- /dev/null +++ b/checkpoint-116500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:a10d3efb8a0afe732c5f7a5dc697e4146a5c0e4718a2bc0ecec59a567283af87 +size 563 diff --git a/checkpoint-116500/unet/config.json b/checkpoint-116500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..bb1cd664fc36d4dc821a4f0fdae4b94d7d99b356 --- /dev/null +++ b/checkpoint-116500/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-116500/unet/diffusion_pytorch_model.bin b/checkpoint-116500/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..a188689ccf251c54f1505be0415b36eabfe987ac --- /dev/null +++ b/checkpoint-116500/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7dcfb67d8b72731cceff088a250d06d7948c175a1ff6a70a8e6b53458e0a3912 +size 36621969 diff --git a/checkpoint-116500/unet_ema/config.json b/checkpoint-116500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..2332de7ad81aa42397686997b17c82e247e8642b --- /dev/null +++ b/checkpoint-116500/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": 116500, + "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-116500/unet_ema/diffusion_pytorch_model.bin b/checkpoint-116500/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..8cf46807aef9ef3cf885637cefdaf5d771587322 --- /dev/null +++ b/checkpoint-116500/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:97c8382d7120ef8633eb626856d0d5ae6014edde1481cead28fa8adf080b67ad +size 36616081 diff --git a/checkpoint-117000/optimizer.bin b/checkpoint-117000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..b2ad5246e65a5aae739abde7c2c8426c4c87bb67 --- /dev/null +++ b/checkpoint-117000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:341c3ea89228117c6ed7f21c785e7af0c83fb43d2daef704dff59a024a78774a +size 73251413 diff --git a/checkpoint-117000/random_states_0.pkl b/checkpoint-117000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..fc89d76c5b96489fb2af11d5dc6c99a89c9a138a --- /dev/null +++ b/checkpoint-117000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:74e96b64289e5706f7e6f3e6a9b6ca9d920a2c180fd6fecd6d530a3e5f7a983a +size 14663 diff --git a/checkpoint-117000/scaler.pt b/checkpoint-117000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..00384287540fca922c5822ef6bedcb2cc7d40243 --- /dev/null +++ b/checkpoint-117000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:203181207dc30fc3fe79a3c76f18c03d94cd4010f18f7ffe7d03db4d625b4830 +size 557 diff --git a/checkpoint-117000/scheduler.bin b/checkpoint-117000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..0457857a83ad0f00430419baa592d1c85362a456 --- /dev/null +++ b/checkpoint-117000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:50fb3720d9d3104ac89ea5097e59e35f810610b6d279c1da866fd67192c23e48 +size 563 diff --git a/checkpoint-117000/unet/config.json b/checkpoint-117000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..bb1cd664fc36d4dc821a4f0fdae4b94d7d99b356 --- /dev/null +++ b/checkpoint-117000/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-117000/unet/diffusion_pytorch_model.bin b/checkpoint-117000/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..df43343f95f12963e9857f50f24b647b51b1917f --- /dev/null +++ b/checkpoint-117000/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:3e7009269582c8753a81bb9668b492c13e6c643d03dc67c9864e59126292e4ac +size 36621969 diff --git a/checkpoint-117000/unet_ema/config.json b/checkpoint-117000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..9fe05aed5b52eefe7e0e1106b0eee58c941490a1 --- /dev/null +++ b/checkpoint-117000/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": 117000, + "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-117000/unet_ema/diffusion_pytorch_model.bin b/checkpoint-117000/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..ba201eab457470506a6e6e99fdefef4257999a96 --- /dev/null +++ b/checkpoint-117000/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b91e7b34f021cdb383efe4b61d35fed21880618cf253fce39651843d9100c04f +size 36616081 diff --git a/unet/diffusion_pytorch_model.bin b/unet/diffusion_pytorch_model.bin index d24e6a17b5a3736604d0772c9fbafaae5b037568..f7959cb731304b2ae6ca8e6d934bd87f2fd4d46d 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:81e83a20d8724f28b61af7497a83fb3120748fbf48c76f6013715e3ee6cd5b42 +oid sha256:0a43e79ae2af725e756a7f75e1cd8e0caf60fc7fc2d310db3cd9f6b650da39f8 size 36621969