diff --git a/checkpoint-126500/optimizer.bin b/checkpoint-126500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..9904caf2e7eb4ed5553cbe6890ea01ee9104f152 --- /dev/null +++ b/checkpoint-126500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e036cff0cf69b922be36ee3a8601d30d7fa20d259a395bb449433280b8279771 +size 73251413 diff --git a/checkpoint-126500/random_states_0.pkl b/checkpoint-126500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..81e4c0999ce53f6eb30488d3d11c5d1201ddb133 --- /dev/null +++ b/checkpoint-126500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:35c0d442dae95369232746c59580cd1731dac4cd678876def7c5e6afa5c541aa +size 14727 diff --git a/checkpoint-126500/scaler.pt b/checkpoint-126500/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..eae8d2ba27243596f47631e3ddbbf6242a3a4851 --- /dev/null +++ b/checkpoint-126500/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:ff317ccfd09d7365103f8d6418ff93166dcb25dffae2ef23911619d9a6697a38 +size 557 diff --git a/checkpoint-126500/scheduler.bin b/checkpoint-126500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..01339bff709806ba67fc5907342fb3580447b219 --- /dev/null +++ b/checkpoint-126500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b7f1749e960aae9ecb84d05b8e66220bc9d33f36955bb2e811ce9d501599f6f4 +size 563 diff --git a/checkpoint-126500/unet/config.json b/checkpoint-126500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..c2d7b8c0d490531013583af57a806d21893e494c --- /dev/null +++ b/checkpoint-126500/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", + "CrossAttnDownBlock2D", + "DownBlock2D" + ], + "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": [ + "UpBlock2D", + "CrossAttnUpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "use_linear_projection": false +} diff --git a/checkpoint-126500/unet/diffusion_pytorch_model.bin b/checkpoint-126500/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..b4124b38eb4c5921c7af70e21687a36f41ac90e9 --- /dev/null +++ b/checkpoint-126500/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:8c700c6bbd8a97a0d6b5060fd638873408befa73156fb8ee63dd00fb2d822575 +size 36621969 diff --git a/checkpoint-126500/unet_ema/config.json b/checkpoint-126500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..2c719819978a4739cbaab55da5e68ac28dc392bb --- /dev/null +++ b/checkpoint-126500/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", + "CrossAttnDownBlock2D", + "DownBlock2D" + ], + "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": 126500, + "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": [ + "UpBlock2D", + "CrossAttnUpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "update_after_step": 0, + "use_ema_warmup": true, + "use_linear_projection": false +} diff --git a/checkpoint-126500/unet_ema/diffusion_pytorch_model.bin b/checkpoint-126500/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..5eed0e70bbe79b775c892abcc8016e1939885c1a --- /dev/null +++ b/checkpoint-126500/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:04b717c958df4c82293e374fe363ea2f8b1dc04609b27526d67dd9a157af969e +size 36616081 diff --git a/checkpoint-127000/optimizer.bin b/checkpoint-127000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..6cc8444145709a5703c7b2dcb05a0e8ce6e3950b --- /dev/null +++ b/checkpoint-127000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:3944ca44d65c8adff59fc23fcefecf380381504d8607f7e7118f6882749a322e +size 73251413 diff --git a/checkpoint-127000/random_states_0.pkl b/checkpoint-127000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..a313a06b26d19140c1e96d815e3c4a57a76f0270 --- /dev/null +++ b/checkpoint-127000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:8c6fb54d8e70f35db2a2cb0a72d973aa36d213464f7a5f0dedc95502a0b9db79 +size 14727 diff --git a/checkpoint-127000/scaler.pt b/checkpoint-127000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..c7ab7b69668530320f0def789e294f32be807130 --- /dev/null +++ b/checkpoint-127000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:8022534461c8912d6de2326e8118bc91a5d5e2af338d7652610691256c75ff3e +size 557 diff --git a/checkpoint-127000/scheduler.bin b/checkpoint-127000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..dce62a7e02ca8af57b2569d11a4c63f2a294b9ec --- /dev/null +++ b/checkpoint-127000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:07578a2cfc3bcaba34a44d0111e1cf088ed0955ac1efdb56cef8f6e0c1dcf8d8 +size 563 diff --git a/checkpoint-127000/unet/config.json b/checkpoint-127000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..c2d7b8c0d490531013583af57a806d21893e494c --- /dev/null +++ b/checkpoint-127000/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", + "CrossAttnDownBlock2D", + "DownBlock2D" + ], + "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": [ + "UpBlock2D", + "CrossAttnUpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "use_linear_projection": false +} diff --git a/checkpoint-127000/unet/diffusion_pytorch_model.bin b/checkpoint-127000/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..efdd51424af460638381970b9e054ede70249857 --- /dev/null +++ b/checkpoint-127000/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:87a143eb277023b8a9aaf82b96f41ad148d9a4f68ff6ab6525142f7f6cf61272 +size 36621969 diff --git a/checkpoint-127000/unet_ema/config.json b/checkpoint-127000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..0502a6208d7af04ec6239e49b2b9a21791625668 --- /dev/null +++ b/checkpoint-127000/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", + "CrossAttnDownBlock2D", + "DownBlock2D" + ], + "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": 127000, + "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": [ + "UpBlock2D", + "CrossAttnUpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "update_after_step": 0, + "use_ema_warmup": true, + "use_linear_projection": false +} diff --git a/checkpoint-127000/unet_ema/diffusion_pytorch_model.bin b/checkpoint-127000/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..a6f918ad6c36173023def58bd70ce468a94e9de4 --- /dev/null +++ b/checkpoint-127000/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:28f5794b3e6f6edc5fc00b1bc4ea784c247801593b7a66bcd7f27b74b232327f +size 36616081 diff --git a/checkpoint-127500/optimizer.bin b/checkpoint-127500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..412ebb720dc65dc36463db0bc9d91f4e41a57370 --- /dev/null +++ b/checkpoint-127500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:4e4ca9fab49338df43e595fccc937741a404b0a6a1eec442bc601687ede091ea +size 73251413 diff --git a/checkpoint-127500/random_states_0.pkl b/checkpoint-127500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..8c46877b7d361d93ed2432552e0cc6bca555aef8 --- /dev/null +++ b/checkpoint-127500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:93a4ffc4e14d398b2ea757b772e092009e2039c3b89d022524c6c9becc4d9d4d +size 14727 diff --git a/checkpoint-127500/scaler.pt b/checkpoint-127500/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..ec6cd8e57314d8cfdbb68d44e5f0f849a032c58a --- /dev/null +++ b/checkpoint-127500/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:5eca7d82b8fae51b52d912d8b7d76c40a301473427a5526a04297966a19eb668 +size 557 diff --git a/checkpoint-127500/scheduler.bin b/checkpoint-127500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..7e5c80b21d47e46b72cc7e958d8f161bf6b21226 --- /dev/null +++ b/checkpoint-127500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:320dec1b0b6e860be9470de84a26285a5aebb55c199b1b9beaf5a482c72f7107 +size 563 diff --git a/checkpoint-127500/unet/config.json b/checkpoint-127500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..c2d7b8c0d490531013583af57a806d21893e494c --- /dev/null +++ b/checkpoint-127500/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", + "CrossAttnDownBlock2D", + "DownBlock2D" + ], + "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": [ + "UpBlock2D", + "CrossAttnUpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "use_linear_projection": false +} diff --git a/checkpoint-127500/unet/diffusion_pytorch_model.bin b/checkpoint-127500/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..9232929bf74f7a69abff27a9a4aa1ad8c2f34e20 --- /dev/null +++ b/checkpoint-127500/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7c9c759e0d08e466bc9ca0dd544c464f191b2b72aa90b266b65941c87e666d41 +size 36621969 diff --git a/checkpoint-127500/unet_ema/config.json b/checkpoint-127500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..5997f6750dfd1d7c31e648ece94a22652f00aff6 --- /dev/null +++ b/checkpoint-127500/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", + "CrossAttnDownBlock2D", + "DownBlock2D" + ], + "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": 127500, + "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": [ + "UpBlock2D", + "CrossAttnUpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "update_after_step": 0, + "use_ema_warmup": true, + "use_linear_projection": false +} diff --git a/checkpoint-127500/unet_ema/diffusion_pytorch_model.bin b/checkpoint-127500/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..a26b7b3723349a52d8a9c49f8fae52ec6753d0ef --- /dev/null +++ b/checkpoint-127500/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:50d543e57202ffc45351856563a173c5afc1425c9aa7546e48b0a8efbff2356f +size 36616081 diff --git a/checkpoint-128000/optimizer.bin b/checkpoint-128000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..07095229405882c7decb7b6db6e9c098b601598d --- /dev/null +++ b/checkpoint-128000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:eeb20d4181da85c560b69dfaa29ffc54adaa649b09834d3e24d74a78532f4d16 +size 73251413 diff --git a/checkpoint-128000/random_states_0.pkl b/checkpoint-128000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..2fb79e93366c138d837fae84276f665554af8443 --- /dev/null +++ b/checkpoint-128000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:474eee4750044286088c0780b7521d86a83d403aa24616cef0d66791c9a0c2b9 +size 14727 diff --git a/checkpoint-128000/scaler.pt b/checkpoint-128000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..341a1152a50e0d4efad74e3214f4f45d789a0ae1 --- /dev/null +++ b/checkpoint-128000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:197f664bc492b81230fb4e324e3a52bf7b1614c7b24fe943b5d4d13ff7a88bbd +size 557 diff --git a/checkpoint-128000/scheduler.bin b/checkpoint-128000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..5df474b0f8ecb844da3da1cb8316eebe94a13e1e --- /dev/null +++ b/checkpoint-128000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:406c23571832d12a09320b6d064561428004bf8f314acf204081e1f3f2561bf9 +size 563 diff --git a/checkpoint-128000/unet/config.json b/checkpoint-128000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..c2d7b8c0d490531013583af57a806d21893e494c --- /dev/null +++ b/checkpoint-128000/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", + "CrossAttnDownBlock2D", + "DownBlock2D" + ], + "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": [ + "UpBlock2D", + "CrossAttnUpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "use_linear_projection": false +} diff --git a/checkpoint-128000/unet/diffusion_pytorch_model.bin b/checkpoint-128000/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..9a154b9032d407b5a467d9de5ed51c2f43da2fa1 --- /dev/null +++ b/checkpoint-128000/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:2c50df967512a16cc11d422082a60bd3f472689380f2efd4e28a17728529832b +size 36621969 diff --git a/checkpoint-128000/unet_ema/config.json b/checkpoint-128000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..babde0ffb6b51b90ecf161c8df88f2c88fbeaf26 --- /dev/null +++ b/checkpoint-128000/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", + "CrossAttnDownBlock2D", + "DownBlock2D" + ], + "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": 128000, + "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": [ + "UpBlock2D", + "CrossAttnUpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "update_after_step": 0, + "use_ema_warmup": true, + "use_linear_projection": false +} diff --git a/checkpoint-128000/unet_ema/diffusion_pytorch_model.bin b/checkpoint-128000/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..e0b6ea09ce2684914685902e59970de2c3a81ede --- /dev/null +++ b/checkpoint-128000/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:aa8b702117bacdede4a26cce53d41302325d1e1ded2c06c9d9711b3fa4d6983a +size 36616081 diff --git a/checkpoint-128500/optimizer.bin b/checkpoint-128500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..8b69cf498b6fb6592573cab4bce40412ed420545 --- /dev/null +++ b/checkpoint-128500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:73cd4eb5d57aa3144a3cd7fc2d6e6f14a05f44473e1a0709a9f87640d7b0c8af +size 73251413 diff --git a/checkpoint-128500/random_states_0.pkl b/checkpoint-128500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..82b83d753e01602ee63bb754f406adee928ebe82 --- /dev/null +++ b/checkpoint-128500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:667f116b7b4858a11ed44bb044f8b281244860a0d33b81d8354d71faa5b8841f +size 14727 diff --git a/checkpoint-128500/scaler.pt b/checkpoint-128500/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..d7870031cb897905d5b27a1643f80ab7ada5629b --- /dev/null +++ b/checkpoint-128500/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7b73b6c85382a42c7e81a273fbbb4a381eaafccd6697ac58ccad8dea811a903b +size 557 diff --git a/checkpoint-128500/scheduler.bin b/checkpoint-128500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..1b45b112047fc29e72d6c40d1b8477bfed911d81 --- /dev/null +++ b/checkpoint-128500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:24fe0804c21e641afa8b76c4e63c90eb95f1206db6fed784fb79fd21b374c622 +size 563 diff --git a/checkpoint-128500/unet/config.json b/checkpoint-128500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..c2d7b8c0d490531013583af57a806d21893e494c --- /dev/null +++ b/checkpoint-128500/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", + "CrossAttnDownBlock2D", + "DownBlock2D" + ], + "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": [ + "UpBlock2D", + "CrossAttnUpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "use_linear_projection": false +} diff --git a/checkpoint-128500/unet/diffusion_pytorch_model.bin b/checkpoint-128500/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..abbde64266ad90a5363d41fcaa1038cc90fc2b3a --- /dev/null +++ b/checkpoint-128500/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:56be74c592ff360841e6e050e17552b82a397abbf7684e7d9b1db20718022549 +size 36621969 diff --git a/checkpoint-128500/unet_ema/config.json b/checkpoint-128500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..f1c2ab9ad5fdcfd2c661c7462c5350c6ef4f66e3 --- /dev/null +++ b/checkpoint-128500/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", + "CrossAttnDownBlock2D", + "DownBlock2D" + ], + "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": 128500, + "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": [ + "UpBlock2D", + "CrossAttnUpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "update_after_step": 0, + "use_ema_warmup": true, + "use_linear_projection": false +} diff --git a/checkpoint-128500/unet_ema/diffusion_pytorch_model.bin b/checkpoint-128500/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..bf1b0c48b786a80b3a0fb4a762eb919f4b7cf76f --- /dev/null +++ b/checkpoint-128500/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:5260a02112bf8f14d74c3aaa6bef2a6f376bd15c7391b831585ef51910aafb9b +size 36616081 diff --git a/checkpoint-129000/optimizer.bin b/checkpoint-129000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..f78a62f9f766f74dc6c54238143145c88c9fac03 --- /dev/null +++ b/checkpoint-129000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:371a95c44852e1361ad6b4833cd9baefb76ba78dc30c0d8634973dd83d60d176 +size 73251413 diff --git a/checkpoint-129000/random_states_0.pkl b/checkpoint-129000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..fe4b7711a6e0bb75711156403d8536e869164b7d --- /dev/null +++ b/checkpoint-129000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b9ebb80a178157b12a2e9be7fab99c746dcd2e7e448d5aad4119ca2891f31ff1 +size 14727 diff --git a/checkpoint-129000/scaler.pt b/checkpoint-129000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..4ad9aec074e58cdc1e73861194b9fbdd24cd1059 --- /dev/null +++ b/checkpoint-129000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:491ab07a53cf7b44ea8b147d0eeddf4a8f9b2d2e3a95032d4b155ba8ec9b7e0b +size 557 diff --git a/checkpoint-129000/scheduler.bin b/checkpoint-129000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..966db9665c24ddab219788f24e2f686a267aa01f --- /dev/null +++ b/checkpoint-129000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:22465b5265d6adf7cbef542b40123d246e74cbd0035bf5085708f687cde86498 +size 563 diff --git a/checkpoint-129000/unet/config.json b/checkpoint-129000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..c2d7b8c0d490531013583af57a806d21893e494c --- /dev/null +++ b/checkpoint-129000/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", + "CrossAttnDownBlock2D", + "DownBlock2D" + ], + "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": [ + "UpBlock2D", + "CrossAttnUpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "use_linear_projection": false +} diff --git a/checkpoint-129000/unet/diffusion_pytorch_model.bin b/checkpoint-129000/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..4d0a25f2e43e107ee1d54f45e4718c37fe686061 --- /dev/null +++ b/checkpoint-129000/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c136fed540210d7393dc75085f082343c969839d6a038cf6f92f753004b480e5 +size 36621969 diff --git a/checkpoint-129000/unet_ema/config.json b/checkpoint-129000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..af423901bb27b3eb2507cfe9daa6f71b36a42d62 --- /dev/null +++ b/checkpoint-129000/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", + "CrossAttnDownBlock2D", + "DownBlock2D" + ], + "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": 129000, + "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": [ + "UpBlock2D", + "CrossAttnUpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "update_after_step": 0, + "use_ema_warmup": true, + "use_linear_projection": false +} diff --git a/checkpoint-129000/unet_ema/diffusion_pytorch_model.bin b/checkpoint-129000/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..bc8895389cd2ad08c7d9079472e8d4003dcf2b9d --- /dev/null +++ b/checkpoint-129000/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:621e829df96c9dd08bf56a3940634e3bdfa21cfc2c979da9c10a4a10aced1417 +size 36616081 diff --git a/checkpoint-129500/optimizer.bin b/checkpoint-129500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..786a27e5f46f41b1b17b2c4f82ac6bb65cde7478 --- /dev/null +++ b/checkpoint-129500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:126272408bc23118dad5a7a94e13f7c69c04c93adcc048351f45e6d6082bf87e +size 73251413 diff --git a/checkpoint-129500/random_states_0.pkl b/checkpoint-129500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..931bda16050ddf25fdda794a67490569eb57e8be --- /dev/null +++ b/checkpoint-129500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:8f23393dbc402b6e9a01b12ba85718a870c4f5da444aecf5e8d26c8518545721 +size 14727 diff --git a/checkpoint-129500/scaler.pt b/checkpoint-129500/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..b41925ddd3f7723dd8ac3acce7de0aeb82f30ed0 --- /dev/null +++ b/checkpoint-129500/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:754dd04f098d5159f252a886b384f67f8bd9c5ce1824f272feff7abaab52197a +size 557 diff --git a/checkpoint-129500/scheduler.bin b/checkpoint-129500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..f09bf6a7c108df1541d0595ccedefa7ba761f3d3 --- /dev/null +++ b/checkpoint-129500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:a0d7ae7ead6708e7232305740efe4419a8fde6473fda9d6e9adc829a83a40f18 +size 563 diff --git a/checkpoint-129500/unet/config.json b/checkpoint-129500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..c2d7b8c0d490531013583af57a806d21893e494c --- /dev/null +++ b/checkpoint-129500/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", + "CrossAttnDownBlock2D", + "DownBlock2D" + ], + "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": [ + "UpBlock2D", + "CrossAttnUpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "use_linear_projection": false +} diff --git a/checkpoint-129500/unet/diffusion_pytorch_model.bin b/checkpoint-129500/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..e74a92905e3457dbde1e31f4b72955e3cab28fc5 --- /dev/null +++ b/checkpoint-129500/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:3d7b99ff91bbb0658ed7b0fdbbf00387740a84d8308df725ccbd31300a8f4d57 +size 36621969 diff --git a/checkpoint-129500/unet_ema/config.json b/checkpoint-129500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..681d7cb73b4efdb451ef300a93aebbe013b01e49 --- /dev/null +++ b/checkpoint-129500/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", + "CrossAttnDownBlock2D", + "DownBlock2D" + ], + "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": 129500, + "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": [ + "UpBlock2D", + "CrossAttnUpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "update_after_step": 0, + "use_ema_warmup": true, + "use_linear_projection": false +} diff --git a/checkpoint-129500/unet_ema/diffusion_pytorch_model.bin b/checkpoint-129500/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..3befb7e23f4a8f11fd37d52e149aa35b15dfe1bb --- /dev/null +++ b/checkpoint-129500/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:fa8f16d062ba98350af895559838d862f93f00129abe872c5f47a376ed67def0 +size 36616081 diff --git a/checkpoint-130000/optimizer.bin b/checkpoint-130000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..ee109ae5fafee3a8a536052e5f9b4cf22b771d04 --- /dev/null +++ b/checkpoint-130000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:081f33af64dc67c029b56f9ffe8dac10383bc3f259b85c367428fcf28dd43c06 +size 73251413 diff --git a/checkpoint-130000/random_states_0.pkl b/checkpoint-130000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..9da45228f83b0e351d07902429799d8f27893e29 --- /dev/null +++ b/checkpoint-130000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:6d261e22de728f1d108b6afc2145bbaea02835b8c98e3c11db771a44b459e48a +size 14727 diff --git a/checkpoint-130000/scaler.pt b/checkpoint-130000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..0d40f8a28a13ab5083a9490ac6c47a6fe5c89919 --- /dev/null +++ b/checkpoint-130000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:dd43ab50ac47bf6056662eb1a362a330b69f9194d051c94c1b44e7ba78a769b3 +size 557 diff --git a/checkpoint-130000/scheduler.bin b/checkpoint-130000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..d5420dc8927c23e69eaa012439ba3a38dcb9d110 --- /dev/null +++ b/checkpoint-130000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:715b31197ef38b6e1c78d74557492d506166ca34b6ea98fae56d3f53bb099e4c +size 563 diff --git a/checkpoint-130000/unet/config.json b/checkpoint-130000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..c2d7b8c0d490531013583af57a806d21893e494c --- /dev/null +++ b/checkpoint-130000/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", + "CrossAttnDownBlock2D", + "DownBlock2D" + ], + "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": [ + "UpBlock2D", + "CrossAttnUpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "use_linear_projection": false +} diff --git a/checkpoint-130000/unet/diffusion_pytorch_model.bin b/checkpoint-130000/unet/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..263c9c999e90bb2939d21bef9fc21df4261b54aa --- /dev/null +++ b/checkpoint-130000/unet/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:4f212fcc0820ffe2efa31996be225b36019bdd71c2aa595a14b486bd5cc45693 +size 36621969 diff --git a/checkpoint-130000/unet_ema/config.json b/checkpoint-130000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..6a007612a86dceeb4693bd42421c5c14c2c75d2c --- /dev/null +++ b/checkpoint-130000/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", + "CrossAttnDownBlock2D", + "DownBlock2D" + ], + "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": 130000, + "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": [ + "UpBlock2D", + "CrossAttnUpBlock2D", + "UpBlock2D", + "UpBlock2D" + ], + "upcast_attention": false, + "update_after_step": 0, + "use_ema_warmup": true, + "use_linear_projection": false +} diff --git a/checkpoint-130000/unet_ema/diffusion_pytorch_model.bin b/checkpoint-130000/unet_ema/diffusion_pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..29f7f4725cafe6a3a076f77841e4f4bd9db537bb --- /dev/null +++ b/checkpoint-130000/unet_ema/diffusion_pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:ae6a1d218b29f62a3004be5d311cc0851a0fc2d58c2b6000bd68daa702b22901 +size 36616081 diff --git a/unet/diffusion_pytorch_model.bin b/unet/diffusion_pytorch_model.bin index 4745bb92451e1b0d475dcf9a7c9d58c85c0b89f3..e307f9e9dd49931b96cd1ec697be8482eddd8e86 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:aef3b603bbc9dd238a0a1d88e1ef28479093e0f1d7d98291724dd001acdcf6d8 +oid sha256:6344b8906475a63e0459683e6459f9fec1633c1b67c4a3d9bc180afea384c0d1 size 36621969