diff --git a/checkpoint-1000/optimizer.bin b/checkpoint-1000/optimizer.bin index ffbb9fa69ccf75c666fcf961efe56e192259863f..252140a8ab6776cad940c5db2762b5f51ddf9019 100644 --- a/checkpoint-1000/optimizer.bin +++ b/checkpoint-1000/optimizer.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:4826aa5ca40c2789c037dae97cffa5d2b25f899370c8b1c9af47a9fc7a34da73 +oid sha256:b12d4a02ce433600feed6f41730270b6a88a787437622d36edc36d6217714d80 size 6876750164 diff --git a/checkpoint-1000/random_states_0.pkl b/checkpoint-1000/random_states_0.pkl index 3cb4cd86db9e08f44623df2a192d1bebfe6f12f0..5f2ed8903ae7b5683f61e044a4ba0baa09ba343c 100644 --- a/checkpoint-1000/random_states_0.pkl +++ b/checkpoint-1000/random_states_0.pkl @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:aa25e20d199053580be9ce57a05318947768ffa19292c5a4504cbaf110e28472 -size 14408 +oid sha256:4bb22af01c54aa70d042c828fcaef6625168573b887b5cce9df0ffa9eed753af +size 14344 diff --git a/checkpoint-1000/scheduler.bin b/checkpoint-1000/scheduler.bin index 71ab870e37a6f9f50db110ef4af812b6e9eaab82..045db640fcd7f1a71a90de0c73ce177a22e601ad 100644 --- a/checkpoint-1000/scheduler.bin +++ b/checkpoint-1000/scheduler.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:cbc3eb564e692e49cdba05c1c2426f4156e46ad00a354ee02b4f09c1ea777047 +oid sha256:ccd5c09464e63cb83bab5fc29289c9f1b45e08f0292df765a199932e5d555e36 size 1000 diff --git a/checkpoint-1000/unet/config.json b/checkpoint-1000/unet/config.json index 4872afd6ff2f1ca9a40ed0bca31b5449b1760578..7635fba9f6130f3988be75419c7700d9db25e562 100644 --- a/checkpoint-1000/unet/config.json +++ b/checkpoint-1000/unet/config.json @@ -1,7 +1,7 @@ { "_class_name": "UNet2DConditionModel", "_diffusers_version": "0.32.0.dev0", - "_name_or_path": "CompVis/stable-diffusion-v1-4", + "_name_or_path": "runwayml/stable-diffusion-v1-5", "act_fn": "silu", "addition_embed_type": null, "addition_embed_type_num_heads": 64, diff --git a/checkpoint-1000/unet/diffusion_pytorch_model.safetensors b/checkpoint-1000/unet/diffusion_pytorch_model.safetensors index 496cdaa2e4168fac76d4730443088e2dc293d621..a5c8d94cd7907edf986486f4913b73fb4e849d75 100644 --- a/checkpoint-1000/unet/diffusion_pytorch_model.safetensors +++ b/checkpoint-1000/unet/diffusion_pytorch_model.safetensors @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:57e4c03070667d18d9d121fb215801a480f2b00f32449c1f085c792dc47002eb +oid sha256:2ec573a50bf1d4c3a55ef748ad2b49e138e017186320b9c3022982b3164e03d2 size 3438167536 diff --git a/checkpoint-1000/unet_ema/config.json b/checkpoint-1000/unet_ema/config.json index 6750dc8e34cd0e3fa9e163b829d52733b11520ad..6bd7c2191f8f198cd400d52a3ab8f8d8568e733b 100644 --- a/checkpoint-1000/unet_ema/config.json +++ b/checkpoint-1000/unet_ema/config.json @@ -1,7 +1,7 @@ { "_class_name": "UNet2DConditionModel", "_diffusers_version": "0.32.0.dev0", - "_name_or_path": "CompVis/stable-diffusion-v1-4", + "_name_or_path": "runwayml/stable-diffusion-v1-5", "act_fn": "silu", "addition_embed_type": null, "addition_embed_type_num_heads": 64, diff --git a/checkpoint-1000/unet_ema/diffusion_pytorch_model.safetensors b/checkpoint-1000/unet_ema/diffusion_pytorch_model.safetensors index 4cb18fdb1116970d1ed012243aee5e89a6ccaf48..41276af9f679946ee0477064400e0467f84097ee 100644 --- a/checkpoint-1000/unet_ema/diffusion_pytorch_model.safetensors +++ b/checkpoint-1000/unet_ema/diffusion_pytorch_model.safetensors @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:9c030a2febb0b5391d1224c2a5ef8dd8cc521f5a555e947cf6194e62078490a5 +oid sha256:adbe68dfb63ffaaef8aa3a7bb92a526109b7cd28b14ba716ccea92fd60a63b1d size 3438167536 diff --git a/checkpoint-1500/optimizer.bin b/checkpoint-1500/optimizer.bin index e36588b0872c9fc0bbb611ba52c0e97afb3ee972..e4afa272b644c70ff97134c2a6a13c9768e0aff5 100644 --- a/checkpoint-1500/optimizer.bin +++ b/checkpoint-1500/optimizer.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:d2a8e3896fd36c1a7a1f59dd8ac582baab9b2febb2696a65b65baa7ee3aabef8 +oid sha256:b28eacfe2877f270749286971bb620a7fbf51eeaf1b19447bd9e89a9a150e540 size 6876750164 diff --git a/checkpoint-1500/random_states_0.pkl b/checkpoint-1500/random_states_0.pkl index f780ee0a326790cfc95b3942216b7f5ef45689db..251dbfbcaec345a705bb9614d59e4f8bcfbbce7e 100644 --- a/checkpoint-1500/random_states_0.pkl +++ b/checkpoint-1500/random_states_0.pkl @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:c9b6acd93bf6d61397913a26a0c0ebcc054d7388517ee87c0dd5d4db3703233b +oid sha256:15005307875474319f88102e1f08cbe93f4409250da4911456dfda7d5a036297 size 14344 diff --git a/checkpoint-1500/scheduler.bin b/checkpoint-1500/scheduler.bin index 2f3ae46bc11d0db235417c677f8879819454515f..b9bb84550c0526165ee6696e07dbbb662331bf01 100644 --- a/checkpoint-1500/scheduler.bin +++ b/checkpoint-1500/scheduler.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:ffb777bcad70bcb56fe14b21849a7bb094914b2b9da9f419d1a5aa250c5375e3 +oid sha256:621302e0322d6be7b05bdf1f8ccb91bf42e0fa11304f5b8a4a17a4dcc08092a6 size 1000 diff --git a/checkpoint-1500/unet/config.json b/checkpoint-1500/unet/config.json index 4872afd6ff2f1ca9a40ed0bca31b5449b1760578..7635fba9f6130f3988be75419c7700d9db25e562 100644 --- a/checkpoint-1500/unet/config.json +++ b/checkpoint-1500/unet/config.json @@ -1,7 +1,7 @@ { "_class_name": "UNet2DConditionModel", "_diffusers_version": "0.32.0.dev0", - "_name_or_path": "CompVis/stable-diffusion-v1-4", + "_name_or_path": "runwayml/stable-diffusion-v1-5", "act_fn": "silu", "addition_embed_type": null, "addition_embed_type_num_heads": 64, diff --git a/checkpoint-1500/unet/diffusion_pytorch_model.safetensors b/checkpoint-1500/unet/diffusion_pytorch_model.safetensors index b483e17644b5b178b2ce92c430505dd78dce25e5..22b2004a1dede2498ce32871b7848349c4620da1 100644 --- a/checkpoint-1500/unet/diffusion_pytorch_model.safetensors +++ b/checkpoint-1500/unet/diffusion_pytorch_model.safetensors @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:590bf8ee2dee6349a3f348ef389ffc278468609ee95f7af2f879833a3f23d4c9 +oid sha256:7695d38b5566064e2feebb526e2ea2e818df64be9d05ddbb51d7389b568e444e size 3438167536 diff --git a/checkpoint-1500/unet_ema/config.json b/checkpoint-1500/unet_ema/config.json index 7acd74840e153b8c64049d0cb2d815ef7a524e63..d4ae1e48c0225ed56a0055f0fdc675fb61b484d9 100644 --- a/checkpoint-1500/unet_ema/config.json +++ b/checkpoint-1500/unet_ema/config.json @@ -1,7 +1,7 @@ { "_class_name": "UNet2DConditionModel", "_diffusers_version": "0.32.0.dev0", - "_name_or_path": "CompVis/stable-diffusion-v1-4", + "_name_or_path": "runwayml/stable-diffusion-v1-5", "act_fn": "silu", "addition_embed_type": null, "addition_embed_type_num_heads": 64, diff --git a/checkpoint-1500/unet_ema/diffusion_pytorch_model.safetensors b/checkpoint-1500/unet_ema/diffusion_pytorch_model.safetensors index 954f12eb8e3d3571a7fc1b3cd39a478cb6f764a6..0ba0e8d3091d1c1aad1ee76b8d897bad32c2cf65 100644 --- a/checkpoint-1500/unet_ema/diffusion_pytorch_model.safetensors +++ b/checkpoint-1500/unet_ema/diffusion_pytorch_model.safetensors @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:d6f0ed9d8f70206585f5c5fe4b7471e32db4cb22cb8c9cf5fe3334d291a7e786 +oid sha256:ed9ea5b64cb017b2ab8a55823377d93e21b950c6d33e1fc9ca6ba056212d360f size 3438167536 diff --git a/checkpoint-2000/optimizer.bin b/checkpoint-2000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..7f4c1057de4d61c866817905d9a8e48d52b45615 --- /dev/null +++ b/checkpoint-2000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:033394cc25b1c704a187a05136300ad3031845d1f7d3c2a850edc6467d0b164d +size 6876750164 diff --git a/checkpoint-2000/random_states_0.pkl b/checkpoint-2000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..ad44d6d7df360b157d3b15ef89e7396fd3bd2fe4 --- /dev/null +++ b/checkpoint-2000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:6d499d0bdcc295d74cad932da112510f9612edf05be18111150a0e2941df6859 +size 14344 diff --git a/checkpoint-2000/scaler.pt b/checkpoint-2000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..a658d3ff25850e9f57ede2c9b36e1c4c72cca194 --- /dev/null +++ b/checkpoint-2000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:58a796e4298dbc8650597cf1d417313553aeb1d22b0b27d883f12c57d40e6d50 +size 988 diff --git a/checkpoint-2000/scheduler.bin b/checkpoint-2000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..db91997f56dba04377b7ba0deadb44bdccdef348 --- /dev/null +++ b/checkpoint-2000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:25148513ab53ebd027ce20940c594cb30e72a7a32c4a07ded384ef6992c03b83 +size 1000 diff --git a/checkpoint-2000/unet/config.json b/checkpoint-2000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..7635fba9f6130f3988be75419c7700d9db25e562 --- /dev/null +++ b/checkpoint-2000/unet/config.json @@ -0,0 +1,68 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.32.0.dev0", + "_name_or_path": "runwayml/stable-diffusion-v1-5", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "addition_time_embed_dim": null, + "attention_head_dim": 8, + "attention_type": "default", + "block_out_channels": [ + 320, + 640, + 1280, + 1280 + ], + "center_input_sample": false, + "class_embed_type": null, + "class_embeddings_concat": false, + "conv_in_kernel": 3, + "conv_out_kernel": 3, + "cross_attention_dim": 768, + "cross_attention_norm": null, + "down_block_types": [ + "CrossAttnDownBlock2D", + "CrossAttnDownBlock2D", + "CrossAttnDownBlock2D", + "DownBlock2D" + ], + "downsample_padding": 1, + "dropout": 0.0, + "dual_cross_attention": false, + "encoder_hid_dim": null, + "encoder_hid_dim_type": null, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 4, + "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_attention_heads": null, + "num_class_embeds": null, + "only_cross_attention": false, + "out_channels": 4, + "projection_class_embeddings_input_dim": null, + "resnet_out_scale_factor": 1.0, + "resnet_skip_time_act": false, + "resnet_time_scale_shift": "default", + "reverse_transformer_layers_per_block": null, + "sample_size": 64, + "time_cond_proj_dim": null, + "time_embedding_act_fn": null, + "time_embedding_dim": null, + "time_embedding_type": "positional", + "timestep_post_act": null, + "transformer_layers_per_block": 1, + "up_block_types": [ + "UpBlock2D", + "CrossAttnUpBlock2D", + "CrossAttnUpBlock2D", + "CrossAttnUpBlock2D" + ], + "upcast_attention": false, + "use_linear_projection": false +} diff --git a/checkpoint-2000/unet/diffusion_pytorch_model.safetensors b/checkpoint-2000/unet/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..f308029eb00530762c0b848ebb9020cebe28a75f --- /dev/null +++ b/checkpoint-2000/unet/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:3790a8362f682cbd65c47f6d2543d1bd281791863373f81f39cfa5508faefa12 +size 3438167536 diff --git a/checkpoint-2000/unet_ema/config.json b/checkpoint-2000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..58f895249fef0e1fe2451b781aad36e3dbbabd5b --- /dev/null +++ b/checkpoint-2000/unet_ema/config.json @@ -0,0 +1,75 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.32.0.dev0", + "_name_or_path": "runwayml/stable-diffusion-v1-5", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "addition_time_embed_dim": null, + "attention_head_dim": 8, + "attention_type": "default", + "block_out_channels": [ + 320, + 640, + 1280, + 1280 + ], + "center_input_sample": false, + "class_embed_type": null, + "class_embeddings_concat": false, + "conv_in_kernel": 3, + "conv_out_kernel": 3, + "cross_attention_dim": 768, + "cross_attention_norm": null, + "decay": 0.9999, + "down_block_types": [ + "CrossAttnDownBlock2D", + "CrossAttnDownBlock2D", + "CrossAttnDownBlock2D", + "DownBlock2D" + ], + "downsample_padding": 1, + "dropout": 0.0, + "dual_cross_attention": false, + "encoder_hid_dim": null, + "encoder_hid_dim_type": null, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 4, + "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_attention_heads": null, + "num_class_embeds": null, + "only_cross_attention": false, + "optimization_step": 2000, + "out_channels": 4, + "power": 0.6666666666666666, + "projection_class_embeddings_input_dim": null, + "resnet_out_scale_factor": 1.0, + "resnet_skip_time_act": false, + "resnet_time_scale_shift": "default", + "reverse_transformer_layers_per_block": null, + "sample_size": 64, + "time_cond_proj_dim": null, + "time_embedding_act_fn": null, + "time_embedding_dim": null, + "time_embedding_type": "positional", + "timestep_post_act": null, + "transformer_layers_per_block": 1, + "up_block_types": [ + "UpBlock2D", + "CrossAttnUpBlock2D", + "CrossAttnUpBlock2D", + "CrossAttnUpBlock2D" + ], + "upcast_attention": false, + "update_after_step": 0, + "use_ema_warmup": false, + "use_linear_projection": false +} diff --git a/checkpoint-2000/unet_ema/diffusion_pytorch_model.safetensors b/checkpoint-2000/unet_ema/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..72324515e6f05acf68f6cbb8e2c12eb9d97b45ce --- /dev/null +++ b/checkpoint-2000/unet_ema/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c03d9cfd0205be3b340f96fe5988a8c688b6ab817ed88efc1f52c1c927c9616e +size 3438167536 diff --git a/checkpoint-2500/optimizer.bin b/checkpoint-2500/optimizer.bin index e6db4767e6f602dd2f53ee419ea5a399a8919722..750f18009311eb72bbb8848de0bdd4a14babd9cb 100644 --- a/checkpoint-2500/optimizer.bin +++ b/checkpoint-2500/optimizer.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:ffe6385d8a76688a1d2ccb46bbe9c30fffbf575af9bfea2d8b92b9f1f1bce84e +oid sha256:72654108e5b8a899cb7904576dbc8fef37a11ced07c914865d44a80ef282117b size 6876750164 diff --git a/checkpoint-2500/random_states_0.pkl b/checkpoint-2500/random_states_0.pkl index fef8a8ae6e45995010e4cf7f5c542bddd7a52b0b..4f5280d6682fdc5a0e6e040731db4dfb27fd8935 100644 --- a/checkpoint-2500/random_states_0.pkl +++ b/checkpoint-2500/random_states_0.pkl @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:d9d915da7f02a86e31ee6f75b8849f0950863ba8994f67a7fc9e9194e604c03e -size 14408 +oid sha256:6f462685dc0baa15da6eeb853e9630e9247effb0fc979b71358c2a7675afe416 +size 14344 diff --git a/checkpoint-2500/scheduler.bin b/checkpoint-2500/scheduler.bin index ee4040d6915f873b14de719954b56a62f4d93222..e5bdbc6bf81a2d7333570a3df444b6d1eecfb6ac 100644 --- a/checkpoint-2500/scheduler.bin +++ b/checkpoint-2500/scheduler.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:87a49c4a26f62f0aa84f4b656eef4c3465103beed2061780e76c86467dbea981 +oid sha256:6353762aaa1c52881aa1d6234c85280b3c734c301ea46eb27bbf7f54a6daf124 size 1000 diff --git a/checkpoint-2500/unet/config.json b/checkpoint-2500/unet/config.json index 4872afd6ff2f1ca9a40ed0bca31b5449b1760578..7635fba9f6130f3988be75419c7700d9db25e562 100644 --- a/checkpoint-2500/unet/config.json +++ b/checkpoint-2500/unet/config.json @@ -1,7 +1,7 @@ { "_class_name": "UNet2DConditionModel", "_diffusers_version": "0.32.0.dev0", - "_name_or_path": "CompVis/stable-diffusion-v1-4", + "_name_or_path": "runwayml/stable-diffusion-v1-5", "act_fn": "silu", "addition_embed_type": null, "addition_embed_type_num_heads": 64, diff --git a/checkpoint-2500/unet/diffusion_pytorch_model.safetensors b/checkpoint-2500/unet/diffusion_pytorch_model.safetensors index 33c84d17e840d808f57de31b4f59dd92f36dbabd..cd06d699e27c6c85f9aa199b158a6530185e91d0 100644 --- a/checkpoint-2500/unet/diffusion_pytorch_model.safetensors +++ b/checkpoint-2500/unet/diffusion_pytorch_model.safetensors @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:e416d3f63a01d1690f5b5fc995ca8b4de4c7b4bd338850b62598b49f3e0b947a +oid sha256:187305089c3f22dbf94267d0e85ecead46b686ef931c42d247d311d24dcb4e5e size 3438167536 diff --git a/checkpoint-2500/unet_ema/config.json b/checkpoint-2500/unet_ema/config.json index 55e49e223eae33b22665f53c4140c1c9bb12c2f0..3838dc8cbe4b6e2f0e6263627a9c81183e95d752 100644 --- a/checkpoint-2500/unet_ema/config.json +++ b/checkpoint-2500/unet_ema/config.json @@ -1,7 +1,7 @@ { "_class_name": "UNet2DConditionModel", "_diffusers_version": "0.32.0.dev0", - "_name_or_path": "CompVis/stable-diffusion-v1-4", + "_name_or_path": "runwayml/stable-diffusion-v1-5", "act_fn": "silu", "addition_embed_type": null, "addition_embed_type_num_heads": 64, diff --git a/checkpoint-2500/unet_ema/diffusion_pytorch_model.safetensors b/checkpoint-2500/unet_ema/diffusion_pytorch_model.safetensors index 8a533725bc40ffefaf7256a73ce9dae893d1c510..b44dec8afac91dcfbb6b7cd819283e117d28c583 100644 --- a/checkpoint-2500/unet_ema/diffusion_pytorch_model.safetensors +++ b/checkpoint-2500/unet_ema/diffusion_pytorch_model.safetensors @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:d303a986868bb1669669824555d32dc7c570cb93cc0c5246b1a2cf933ad35cf6 +oid sha256:0f3abc2318f2e6831343ee2859a9395d1389d72bdbc98630caf5d93a6ad8b2cd size 3438167536 diff --git a/checkpoint-3000/optimizer.bin b/checkpoint-3000/optimizer.bin index 3524188d8bc13fecc32bc94b3ad52fce6afa6dc6..ba038bfdfdf4e30fc63a199ff93008c1f5b6df41 100644 --- a/checkpoint-3000/optimizer.bin +++ b/checkpoint-3000/optimizer.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:ae99e0b8f5f1f60ce1a9006dbf2f7ec9fd98eb84cfbec0bfc7bd55434153b4b7 +oid sha256:ebf72f73691f4547c09f173c0853851d588bbd05e1dd0631ee2f83c58a6652df size 6876750164 diff --git a/checkpoint-3000/random_states_0.pkl b/checkpoint-3000/random_states_0.pkl index a515f9c64323a0a9ad74165e35a10b38820828b0..32752ae7d278dd126d8723a638f79f179827f676 100644 --- a/checkpoint-3000/random_states_0.pkl +++ b/checkpoint-3000/random_states_0.pkl @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:96e3020305e5f7def3f4f8486056afad3a29127a881e8b01294ef39b1a59b78b -size 14408 +oid sha256:9157fcbbeed6d62c0d206cec59002cc4f577b53d6374fa1511009f012caa2371 +size 14344 diff --git a/checkpoint-3000/scheduler.bin b/checkpoint-3000/scheduler.bin index fa3320d629bccfdc7baa5012e4b63a9075a337fd..ba2cc3b01b2417e557da8786431c98c8a94f48d2 100644 --- a/checkpoint-3000/scheduler.bin +++ b/checkpoint-3000/scheduler.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:9cd157c3e520fe67f80ed2bb1dc152730c79b5ece6a75d207b88cbe9517b4bf7 +oid sha256:910e7b81b91dda8cf27d9a8f5ab0f27fcae595c1765ce971c4296041bcfbe551 size 1000 diff --git a/checkpoint-3000/unet/config.json b/checkpoint-3000/unet/config.json index 4872afd6ff2f1ca9a40ed0bca31b5449b1760578..7635fba9f6130f3988be75419c7700d9db25e562 100644 --- a/checkpoint-3000/unet/config.json +++ b/checkpoint-3000/unet/config.json @@ -1,7 +1,7 @@ { "_class_name": "UNet2DConditionModel", "_diffusers_version": "0.32.0.dev0", - "_name_or_path": "CompVis/stable-diffusion-v1-4", + "_name_or_path": "runwayml/stable-diffusion-v1-5", "act_fn": "silu", "addition_embed_type": null, "addition_embed_type_num_heads": 64, diff --git a/checkpoint-3000/unet/diffusion_pytorch_model.safetensors b/checkpoint-3000/unet/diffusion_pytorch_model.safetensors index 93cd04eda84cb802b1208f0c43a972b4f5d435e7..74dbb8b19f4691e0110910ef119a11bc2c41f6d3 100644 --- a/checkpoint-3000/unet/diffusion_pytorch_model.safetensors +++ b/checkpoint-3000/unet/diffusion_pytorch_model.safetensors @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:495c7cf4d59488e75c91360559bd8b9b2e99de21300c50db482985810051c98a +oid sha256:2a20ef583ea805aabd0592292951003db0cc956ea418606e08d494d77e4a07f3 size 3438167536 diff --git a/checkpoint-3000/unet_ema/config.json b/checkpoint-3000/unet_ema/config.json index e656db682ec687168293394993cff1525d0ec01c..5a3aa92790b98e1ee33a1cfdb192f21ed32190d8 100644 --- a/checkpoint-3000/unet_ema/config.json +++ b/checkpoint-3000/unet_ema/config.json @@ -1,7 +1,7 @@ { "_class_name": "UNet2DConditionModel", "_diffusers_version": "0.32.0.dev0", - "_name_or_path": "CompVis/stable-diffusion-v1-4", + "_name_or_path": "runwayml/stable-diffusion-v1-5", "act_fn": "silu", "addition_embed_type": null, "addition_embed_type_num_heads": 64, diff --git a/checkpoint-3000/unet_ema/diffusion_pytorch_model.safetensors b/checkpoint-3000/unet_ema/diffusion_pytorch_model.safetensors index c8e8415a27cfb9826a3a1a751e6972a527bcb813..ddc3f573ac4b3a72b0e01372da18e0fcf2abc383 100644 --- a/checkpoint-3000/unet_ema/diffusion_pytorch_model.safetensors +++ b/checkpoint-3000/unet_ema/diffusion_pytorch_model.safetensors @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:52899fbeea249abae4130e5fc0cc69337da1cce17cffda3d783d6775fe2f393f +oid sha256:620f6f2a2a3deed3be12699aef9d88c571b5d482fe5d82022226d6dd35acd732 size 3438167536 diff --git a/checkpoint-3500/optimizer.bin b/checkpoint-3500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..155e9b8fe2b9ec3431c47647b12782fdf4dd3ea7 --- /dev/null +++ b/checkpoint-3500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:887cbd87195eacf3b4a58af3f91e5dc09c5947bd936cb2b8c2745a567b2e46fe +size 6876750164 diff --git a/checkpoint-3500/random_states_0.pkl b/checkpoint-3500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..35696d2935a171cedb7dee4df960fcfae3ddb697 --- /dev/null +++ b/checkpoint-3500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:954fe4da4d62c79985e1b7a8ccc3db53d09c3b2b1a44671e923d2e78f98e80fa +size 14344 diff --git a/checkpoint-3500/scaler.pt b/checkpoint-3500/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..a924998d94801336044099de2468e03991cb3918 --- /dev/null +++ b/checkpoint-3500/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:f586fda6b71232ded73bd26e574e83cf6f329fb64a2aabd51108cab0deca57ce +size 988 diff --git a/checkpoint-3500/scheduler.bin b/checkpoint-3500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..02ea9d6b2ac679ac729f671ff2fe620f63f00d81 --- /dev/null +++ b/checkpoint-3500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:028c42f5139fef439208dd82620055df43c50636b0df3f5c77562cdc762631c4 +size 1000 diff --git a/checkpoint-3500/unet/config.json b/checkpoint-3500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..7635fba9f6130f3988be75419c7700d9db25e562 --- /dev/null +++ b/checkpoint-3500/unet/config.json @@ -0,0 +1,68 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.32.0.dev0", + "_name_or_path": "runwayml/stable-diffusion-v1-5", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "addition_time_embed_dim": null, + "attention_head_dim": 8, + "attention_type": "default", + "block_out_channels": [ + 320, + 640, + 1280, + 1280 + ], + "center_input_sample": false, + "class_embed_type": null, + "class_embeddings_concat": false, + "conv_in_kernel": 3, + "conv_out_kernel": 3, + "cross_attention_dim": 768, + "cross_attention_norm": null, + "down_block_types": [ + "CrossAttnDownBlock2D", + "CrossAttnDownBlock2D", + "CrossAttnDownBlock2D", + "DownBlock2D" + ], + "downsample_padding": 1, + "dropout": 0.0, + "dual_cross_attention": false, + "encoder_hid_dim": null, + "encoder_hid_dim_type": null, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 4, + "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_attention_heads": null, + "num_class_embeds": null, + "only_cross_attention": false, + "out_channels": 4, + "projection_class_embeddings_input_dim": null, + "resnet_out_scale_factor": 1.0, + "resnet_skip_time_act": false, + "resnet_time_scale_shift": "default", + "reverse_transformer_layers_per_block": null, + "sample_size": 64, + "time_cond_proj_dim": null, + "time_embedding_act_fn": null, + "time_embedding_dim": null, + "time_embedding_type": "positional", + "timestep_post_act": null, + "transformer_layers_per_block": 1, + "up_block_types": [ + "UpBlock2D", + "CrossAttnUpBlock2D", + "CrossAttnUpBlock2D", + "CrossAttnUpBlock2D" + ], + "upcast_attention": false, + "use_linear_projection": false +} diff --git a/checkpoint-3500/unet/diffusion_pytorch_model.safetensors b/checkpoint-3500/unet/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..b3adcaa0a61a7be018d9aea5fe3a3761027894e6 --- /dev/null +++ b/checkpoint-3500/unet/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c5fe2d6a37a34873c6ea3dd34325b8bffd1d656498816e0c589ab3ea0a2893f5 +size 3438167536 diff --git a/checkpoint-3500/unet_ema/config.json b/checkpoint-3500/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..aaec121d1a7b5f2655c1c2f22c43a651b0f58d9f --- /dev/null +++ b/checkpoint-3500/unet_ema/config.json @@ -0,0 +1,75 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.32.0.dev0", + "_name_or_path": "runwayml/stable-diffusion-v1-5", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "addition_time_embed_dim": null, + "attention_head_dim": 8, + "attention_type": "default", + "block_out_channels": [ + 320, + 640, + 1280, + 1280 + ], + "center_input_sample": false, + "class_embed_type": null, + "class_embeddings_concat": false, + "conv_in_kernel": 3, + "conv_out_kernel": 3, + "cross_attention_dim": 768, + "cross_attention_norm": null, + "decay": 0.9999, + "down_block_types": [ + "CrossAttnDownBlock2D", + "CrossAttnDownBlock2D", + "CrossAttnDownBlock2D", + "DownBlock2D" + ], + "downsample_padding": 1, + "dropout": 0.0, + "dual_cross_attention": false, + "encoder_hid_dim": null, + "encoder_hid_dim_type": null, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 4, + "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_attention_heads": null, + "num_class_embeds": null, + "only_cross_attention": false, + "optimization_step": 3500, + "out_channels": 4, + "power": 0.6666666666666666, + "projection_class_embeddings_input_dim": null, + "resnet_out_scale_factor": 1.0, + "resnet_skip_time_act": false, + "resnet_time_scale_shift": "default", + "reverse_transformer_layers_per_block": null, + "sample_size": 64, + "time_cond_proj_dim": null, + "time_embedding_act_fn": null, + "time_embedding_dim": null, + "time_embedding_type": "positional", + "timestep_post_act": null, + "transformer_layers_per_block": 1, + "up_block_types": [ + "UpBlock2D", + "CrossAttnUpBlock2D", + "CrossAttnUpBlock2D", + "CrossAttnUpBlock2D" + ], + "upcast_attention": false, + "update_after_step": 0, + "use_ema_warmup": false, + "use_linear_projection": false +} diff --git a/checkpoint-3500/unet_ema/diffusion_pytorch_model.safetensors b/checkpoint-3500/unet_ema/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..307849d1b0554d570c11aef17d1bd51f52ae635a --- /dev/null +++ b/checkpoint-3500/unet_ema/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:64c135cededad7dbc9b359e5d414451adfab670def2e3b2f66ae213c36c38cd1 +size 3438167536 diff --git a/checkpoint-4000/optimizer.bin b/checkpoint-4000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..8cb6bce9840598c41e5b1e4db2b9f9a7cbe74783 --- /dev/null +++ b/checkpoint-4000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:8e1ed52e9ded4a7f98ded79f501cc3ccabe8626680188f726450383a8e48552d +size 6876750164 diff --git a/checkpoint-4000/random_states_0.pkl b/checkpoint-4000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..496eccb5ec3b91301e9f99509f4bf134e42ac583 --- /dev/null +++ b/checkpoint-4000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:97e15c8649ecf486541830bf3fdcd111b60724a5e69d05d8ef4989e3baa4c758 +size 14344 diff --git a/checkpoint-4000/scaler.pt b/checkpoint-4000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..42f439fb9472f801fa32d0be214992042405bf1e --- /dev/null +++ b/checkpoint-4000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:f55a5bebb6f7643d4bf339b100cc687400da46957cd9793deabb72c0f7aed9ae +size 988 diff --git a/checkpoint-4000/scheduler.bin b/checkpoint-4000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..0b9724f9e172cb1291efb21d869a895d016ab566 --- /dev/null +++ b/checkpoint-4000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:3b0da7609303685e021f64c9a7aa3dfe8810cac29fd2800358a6bfc3b05aaa4d +size 1000 diff --git a/checkpoint-4000/unet/config.json b/checkpoint-4000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..7635fba9f6130f3988be75419c7700d9db25e562 --- /dev/null +++ b/checkpoint-4000/unet/config.json @@ -0,0 +1,68 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.32.0.dev0", + "_name_or_path": "runwayml/stable-diffusion-v1-5", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "addition_time_embed_dim": null, + "attention_head_dim": 8, + "attention_type": "default", + "block_out_channels": [ + 320, + 640, + 1280, + 1280 + ], + "center_input_sample": false, + "class_embed_type": null, + "class_embeddings_concat": false, + "conv_in_kernel": 3, + "conv_out_kernel": 3, + "cross_attention_dim": 768, + "cross_attention_norm": null, + "down_block_types": [ + "CrossAttnDownBlock2D", + "CrossAttnDownBlock2D", + "CrossAttnDownBlock2D", + "DownBlock2D" + ], + "downsample_padding": 1, + "dropout": 0.0, + "dual_cross_attention": false, + "encoder_hid_dim": null, + "encoder_hid_dim_type": null, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 4, + "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_attention_heads": null, + "num_class_embeds": null, + "only_cross_attention": false, + "out_channels": 4, + "projection_class_embeddings_input_dim": null, + "resnet_out_scale_factor": 1.0, + "resnet_skip_time_act": false, + "resnet_time_scale_shift": "default", + "reverse_transformer_layers_per_block": null, + "sample_size": 64, + "time_cond_proj_dim": null, + "time_embedding_act_fn": null, + "time_embedding_dim": null, + "time_embedding_type": "positional", + "timestep_post_act": null, + "transformer_layers_per_block": 1, + "up_block_types": [ + "UpBlock2D", + "CrossAttnUpBlock2D", + "CrossAttnUpBlock2D", + "CrossAttnUpBlock2D" + ], + "upcast_attention": false, + "use_linear_projection": false +} diff --git a/checkpoint-4000/unet/diffusion_pytorch_model.safetensors b/checkpoint-4000/unet/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..606268e4d730f0ffd22905e88e8ff9175cd9865d --- /dev/null +++ b/checkpoint-4000/unet/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e73c1cad37cb7a416dfe30b2fac832a10540fa2afd65f2ac165f1a85078aba8c +size 3438167536 diff --git a/checkpoint-4000/unet_ema/config.json b/checkpoint-4000/unet_ema/config.json new file mode 100644 index 0000000000000000000000000000000000000000..1efc59643a390b9264546d524ecba6fa63c48440 --- /dev/null +++ b/checkpoint-4000/unet_ema/config.json @@ -0,0 +1,75 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.32.0.dev0", + "_name_or_path": "runwayml/stable-diffusion-v1-5", + "act_fn": "silu", + "addition_embed_type": null, + "addition_embed_type_num_heads": 64, + "addition_time_embed_dim": null, + "attention_head_dim": 8, + "attention_type": "default", + "block_out_channels": [ + 320, + 640, + 1280, + 1280 + ], + "center_input_sample": false, + "class_embed_type": null, + "class_embeddings_concat": false, + "conv_in_kernel": 3, + "conv_out_kernel": 3, + "cross_attention_dim": 768, + "cross_attention_norm": null, + "decay": 0.9999, + "down_block_types": [ + "CrossAttnDownBlock2D", + "CrossAttnDownBlock2D", + "CrossAttnDownBlock2D", + "DownBlock2D" + ], + "downsample_padding": 1, + "dropout": 0.0, + "dual_cross_attention": false, + "encoder_hid_dim": null, + "encoder_hid_dim_type": null, + "flip_sin_to_cos": true, + "freq_shift": 0, + "in_channels": 4, + "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_attention_heads": null, + "num_class_embeds": null, + "only_cross_attention": false, + "optimization_step": 4000, + "out_channels": 4, + "power": 0.6666666666666666, + "projection_class_embeddings_input_dim": null, + "resnet_out_scale_factor": 1.0, + "resnet_skip_time_act": false, + "resnet_time_scale_shift": "default", + "reverse_transformer_layers_per_block": null, + "sample_size": 64, + "time_cond_proj_dim": null, + "time_embedding_act_fn": null, + "time_embedding_dim": null, + "time_embedding_type": "positional", + "timestep_post_act": null, + "transformer_layers_per_block": 1, + "up_block_types": [ + "UpBlock2D", + "CrossAttnUpBlock2D", + "CrossAttnUpBlock2D", + "CrossAttnUpBlock2D" + ], + "upcast_attention": false, + "update_after_step": 0, + "use_ema_warmup": false, + "use_linear_projection": false +} diff --git a/checkpoint-4000/unet_ema/diffusion_pytorch_model.safetensors b/checkpoint-4000/unet_ema/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..3e7589096fe0b4ccca135d7ed6c3bf14ba28cefb --- /dev/null +++ b/checkpoint-4000/unet_ema/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:4aee85f03875240810cb3eb31c37080cd99de0b02ce667d3862bc5de537a8605 +size 3438167536 diff --git a/checkpoint-500/optimizer.bin b/checkpoint-500/optimizer.bin index 03f333fd00abcb6bedd517b8340d1120d803d855..8f2efd024f17aec9382396d078e21d5bf9bf4381 100644 --- a/checkpoint-500/optimizer.bin +++ b/checkpoint-500/optimizer.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:bb22811bdb8e53a1d016d3dc58db2e01dcba9367ad387b10df3f8325921bd24e +oid sha256:232462da78e082b4742dd23c884351f2045b7cf0ea99a805da1bdd3ae891235b size 6876750164 diff --git a/checkpoint-500/random_states_0.pkl b/checkpoint-500/random_states_0.pkl index 158ad0d1884ff1a3fae812302f2be9407961d624..411c789bbddd44b160ff2a2fe8f452cb17fb0af7 100644 --- a/checkpoint-500/random_states_0.pkl +++ b/checkpoint-500/random_states_0.pkl @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:e8d971c52410066844d3024a704d1c65b34db02ef0056abbf0d3a418d616c894 -size 14408 +oid sha256:3e289ce2bf6eb78f79e80f621f44bd2df73015c938e589bbab14291978ccdc10 +size 14344 diff --git a/checkpoint-500/scheduler.bin b/checkpoint-500/scheduler.bin index 7e27866f2765547d560cd4789433ac0a20f0fb72..d18007c802bd16df96e58c88c3539bda66bdab22 100644 --- a/checkpoint-500/scheduler.bin +++ b/checkpoint-500/scheduler.bin @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:ef6a9eb437b711407538ca3058645b7ae30bb34bfa6897b80260bb245054aff2 +oid sha256:ade7e52d4d5140c908ec80995ce445acee76c73a13851d1a6cfec1a6da0b4e4a size 1000 diff --git a/checkpoint-500/unet/config.json b/checkpoint-500/unet/config.json index 4872afd6ff2f1ca9a40ed0bca31b5449b1760578..7635fba9f6130f3988be75419c7700d9db25e562 100644 --- a/checkpoint-500/unet/config.json +++ b/checkpoint-500/unet/config.json @@ -1,7 +1,7 @@ { "_class_name": "UNet2DConditionModel", "_diffusers_version": "0.32.0.dev0", - "_name_or_path": "CompVis/stable-diffusion-v1-4", + "_name_or_path": "runwayml/stable-diffusion-v1-5", "act_fn": "silu", "addition_embed_type": null, "addition_embed_type_num_heads": 64, diff --git a/checkpoint-500/unet/diffusion_pytorch_model.safetensors b/checkpoint-500/unet/diffusion_pytorch_model.safetensors index 4c40845ea6b533222d8fb1d5a66d4fa70228ce46..2a0c94d350b38b45ed240210a6a00d639c2a6934 100644 --- a/checkpoint-500/unet/diffusion_pytorch_model.safetensors +++ b/checkpoint-500/unet/diffusion_pytorch_model.safetensors @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:b66301d99b0acbd8c766c4da195bc209c3973912193c5d60cbd1d7bc36dad7ab +oid sha256:d373e06bab1f899931ad5d689afe345d39b79bb4b0fe7aad5d0c1995a52a1bba size 3438167536 diff --git a/checkpoint-500/unet_ema/config.json b/checkpoint-500/unet_ema/config.json index 4d8c67f315d6a253db013c1f45e9cbbcabf34ab6..969546ec8bf59bbe4c62e12db4fd82b00403ba72 100644 --- a/checkpoint-500/unet_ema/config.json +++ b/checkpoint-500/unet_ema/config.json @@ -1,7 +1,7 @@ { "_class_name": "UNet2DConditionModel", "_diffusers_version": "0.32.0.dev0", - "_name_or_path": "CompVis/stable-diffusion-v1-4", + "_name_or_path": "runwayml/stable-diffusion-v1-5", "act_fn": "silu", "addition_embed_type": null, "addition_embed_type_num_heads": 64, diff --git a/checkpoint-500/unet_ema/diffusion_pytorch_model.safetensors b/checkpoint-500/unet_ema/diffusion_pytorch_model.safetensors index 6be7a9c0657192f7ea59bfd75505622404eb43ff..7915c0f6e1540a891cba9dd8b5c5265a80db2492 100644 --- a/checkpoint-500/unet_ema/diffusion_pytorch_model.safetensors +++ b/checkpoint-500/unet_ema/diffusion_pytorch_model.safetensors @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:a072753e43abeb0c4ad94da80e19487617a583e917e6a078ccb1de2c0289b93f +oid sha256:f414eb4f3042e347342705f57e5ee885df87cd697c55de41e38123ce0d9df6ea size 3438167536 diff --git a/logs/text2image-fine-tune/1730235167.1831658/events.out.tfevents.1730235167.IPEG-WS-008.27296.1 b/logs/text2image-fine-tune/1730235167.1831658/events.out.tfevents.1730235167.IPEG-WS-008.27296.1 new file mode 100644 index 0000000000000000000000000000000000000000..aebdda73b09daa10c8b0cd7c6e81a558d9787949 --- /dev/null +++ b/logs/text2image-fine-tune/1730235167.1831658/events.out.tfevents.1730235167.IPEG-WS-008.27296.1 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b51e91f2c3756e372b1f1a5c53f37f11e94fd5ab13b8d938ae76fae9a0cdb43e +size 2528 diff --git a/logs/text2image-fine-tune/1730235167.201874/hparams.yml b/logs/text2image-fine-tune/1730235167.201874/hparams.yml new file mode 100644 index 0000000000000000000000000000000000000000..53d77080de7fea8acba9f7304c014b1ed8a4a9ed --- /dev/null +++ b/logs/text2image-fine-tune/1730235167.201874/hparams.yml @@ -0,0 +1,55 @@ +adam_beta1: 0.9 +adam_beta2: 0.999 +adam_epsilon: 1.0e-08 +adam_weight_decay: 0.01 +allow_tf32: false +cache_dir: null +caption_column: caption +center_crop: true +checkpointing_steps: 500 +checkpoints_total_limit: null +dataloader_num_workers: 0 +dataset_config_name: null +dataset_name: Trkkk/hannover_text_img +dream_detail_preservation: 1.0 +dream_training: false +enable_xformers_memory_efficient_attention: false +foreach_ema: false +gradient_accumulation_steps: 1 +gradient_checkpointing: true +hub_model_id: Trkkk/stable_diffusion +hub_token: null +image_column: image +input_perturbation: 0 +learning_rate: 2.0e-06 +local_rank: -1 +logging_dir: logs +lr_scheduler: constant +lr_warmup_steps: 0 +max_grad_norm: 1.0 +max_train_samples: null +max_train_steps: 3500 +mixed_precision: fp16 +noise_offset: 0 +non_ema_revision: null +num_train_epochs: 125 +offload_ema: false +output_dir: ./fine_tuned_model_v4 +prediction_type: null +pretrained_model_name_or_path: runwayml/stable-diffusion-v1-5 +push_to_hub: true +random_flip: true +report_to: tensorboard +resolution: 256 +resume_from_checkpoint: null +revision: null +scale_lr: false +seed: null +snr_gamma: null +tracker_project_name: text2image-fine-tune +train_batch_size: 4 +train_data_dir: null +use_8bit_adam: false +use_ema: true +validation_epochs: 5 +variant: null diff --git a/logs/text2image-fine-tune/1730240791.2981133/events.out.tfevents.1730240791.IPEG-WS-008.22768.1 b/logs/text2image-fine-tune/1730240791.2981133/events.out.tfevents.1730240791.IPEG-WS-008.22768.1 new file mode 100644 index 0000000000000000000000000000000000000000..2631f87a014da10db86fa01b4c360381fd697dd6 --- /dev/null +++ b/logs/text2image-fine-tune/1730240791.2981133/events.out.tfevents.1730240791.IPEG-WS-008.22768.1 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:71aa2708e8d9ad6695e501dab750bf2c375ad4e24252c54b4ce10cad5c9177ff +size 2528 diff --git a/logs/text2image-fine-tune/1730240791.3410735/hparams.yml b/logs/text2image-fine-tune/1730240791.3410735/hparams.yml new file mode 100644 index 0000000000000000000000000000000000000000..881b74d837bb2531f4d8a8ed27da05e77311bf28 --- /dev/null +++ b/logs/text2image-fine-tune/1730240791.3410735/hparams.yml @@ -0,0 +1,55 @@ +adam_beta1: 0.9 +adam_beta2: 0.999 +adam_epsilon: 1.0e-08 +adam_weight_decay: 0.01 +allow_tf32: false +cache_dir: null +caption_column: caption +center_crop: true +checkpointing_steps: 500 +checkpoints_total_limit: null +dataloader_num_workers: 0 +dataset_config_name: null +dataset_name: Trkkk/hannover_text_img +dream_detail_preservation: 1.0 +dream_training: false +enable_xformers_memory_efficient_attention: false +foreach_ema: false +gradient_accumulation_steps: 1 +gradient_checkpointing: true +hub_model_id: Trkkk/stable_diffusion +hub_token: null +image_column: image +input_perturbation: 0 +learning_rate: 2.0e-06 +local_rank: -1 +logging_dir: logs +lr_scheduler: constant +lr_warmup_steps: 0 +max_grad_norm: 1.0 +max_train_samples: null +max_train_steps: 3500 +mixed_precision: fp16 +noise_offset: 0 +non_ema_revision: null +num_train_epochs: 63 +offload_ema: false +output_dir: ./fine_tuned_model_v4 +prediction_type: null +pretrained_model_name_or_path: runwayml/stable-diffusion-v1-5 +push_to_hub: true +random_flip: true +report_to: tensorboard +resolution: 256 +resume_from_checkpoint: null +revision: null +scale_lr: false +seed: null +snr_gamma: null +tracker_project_name: text2image-fine-tune +train_batch_size: 2 +train_data_dir: null +use_8bit_adam: false +use_ema: true +validation_epochs: 5 +variant: null diff --git a/logs/text2image-fine-tune/1730246792.2625384/events.out.tfevents.1730246792.IPEG-WS-008.27212.1 b/logs/text2image-fine-tune/1730246792.2625384/events.out.tfevents.1730246792.IPEG-WS-008.27212.1 new file mode 100644 index 0000000000000000000000000000000000000000..60063ccae597b58f4207dfbf2e4d49390023bf90 --- /dev/null +++ b/logs/text2image-fine-tune/1730246792.2625384/events.out.tfevents.1730246792.IPEG-WS-008.27212.1 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b844e4ef2fa6464d262bec0f270bc69014c8dcf46ea4d0a1f0a267e8236ff756 +size 2528 diff --git a/logs/text2image-fine-tune/1730246792.30242/hparams.yml b/logs/text2image-fine-tune/1730246792.30242/hparams.yml new file mode 100644 index 0000000000000000000000000000000000000000..efcaa27a71c822510e36bb3cf55d1b0a6aa02a77 --- /dev/null +++ b/logs/text2image-fine-tune/1730246792.30242/hparams.yml @@ -0,0 +1,55 @@ +adam_beta1: 0.9 +adam_beta2: 0.999 +adam_epsilon: 1.0e-08 +adam_weight_decay: 0.01 +allow_tf32: false +cache_dir: null +caption_column: caption +center_crop: true +checkpointing_steps: 500 +checkpoints_total_limit: null +dataloader_num_workers: 0 +dataset_config_name: null +dataset_name: Trkkk/hannover_text_img +dream_detail_preservation: 1.0 +dream_training: false +enable_xformers_memory_efficient_attention: false +foreach_ema: false +gradient_accumulation_steps: 4 +gradient_checkpointing: true +hub_model_id: Trkkk/stable_diffusion +hub_token: null +image_column: image +input_perturbation: 0 +learning_rate: 2.0e-06 +local_rank: -1 +logging_dir: logs +lr_scheduler: constant +lr_warmup_steps: 0 +max_grad_norm: 1.0 +max_train_samples: null +max_train_steps: 4000 +mixed_precision: fp16 +noise_offset: 0 +non_ema_revision: null +num_train_epochs: 286 +offload_ema: false +output_dir: ./fine_tuned_model_v4 +prediction_type: null +pretrained_model_name_or_path: runwayml/stable-diffusion-v1-5 +push_to_hub: true +random_flip: true +report_to: tensorboard +resolution: 256 +resume_from_checkpoint: null +revision: null +scale_lr: false +seed: null +snr_gamma: null +tracker_project_name: text2image-fine-tune +train_batch_size: 2 +train_data_dir: null +use_8bit_adam: false +use_ema: true +validation_epochs: 5 +variant: null diff --git a/logs/text2image-fine-tune/1730256973.6816907/events.out.tfevents.1730256973.IPEG-WS-008.24668.1 b/logs/text2image-fine-tune/1730256973.6816907/events.out.tfevents.1730256973.IPEG-WS-008.24668.1 new file mode 100644 index 0000000000000000000000000000000000000000..c942fd6c081a7d7604230570998e5e2fdacb31ca --- /dev/null +++ b/logs/text2image-fine-tune/1730256973.6816907/events.out.tfevents.1730256973.IPEG-WS-008.24668.1 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:2f05bf3e686787f82834138b31795c1b73ca03e84ac46dd68aa7cdd34ebf70bc +size 2528 diff --git a/logs/text2image-fine-tune/1730256973.7215154/hparams.yml b/logs/text2image-fine-tune/1730256973.7215154/hparams.yml new file mode 100644 index 0000000000000000000000000000000000000000..8d4d003f0cc1eaa7625088bd865564399b5dd9a6 --- /dev/null +++ b/logs/text2image-fine-tune/1730256973.7215154/hparams.yml @@ -0,0 +1,55 @@ +adam_beta1: 0.9 +adam_beta2: 0.999 +adam_epsilon: 1.0e-08 +adam_weight_decay: 0.01 +allow_tf32: false +cache_dir: null +caption_column: caption +center_crop: true +checkpointing_steps: 500 +checkpoints_total_limit: null +dataloader_num_workers: 0 +dataset_config_name: null +dataset_name: Trkkk/hannover_text_img +dream_detail_preservation: 1.0 +dream_training: false +enable_xformers_memory_efficient_attention: false +foreach_ema: false +gradient_accumulation_steps: 4 +gradient_checkpointing: true +hub_model_id: Trkkk/stable_diffusion +hub_token: null +image_column: image +input_perturbation: 0 +learning_rate: 5.0e-06 +local_rank: -1 +logging_dir: logs +lr_scheduler: constant +lr_warmup_steps: 0 +max_grad_norm: 1.0 +max_train_samples: null +max_train_steps: 2000 +mixed_precision: fp16 +noise_offset: 0 +non_ema_revision: null +num_train_epochs: 143 +offload_ema: false +output_dir: ./fine_tuned_model_v4 +prediction_type: null +pretrained_model_name_or_path: runwayml/stable-diffusion-v1-5 +push_to_hub: true +random_flip: true +report_to: tensorboard +resolution: 256 +resume_from_checkpoint: null +revision: null +scale_lr: false +seed: null +snr_gamma: null +tracker_project_name: text2image-fine-tune +train_batch_size: 2 +train_data_dir: null +use_8bit_adam: false +use_ema: true +validation_epochs: 5 +variant: null diff --git a/logs/text2image-fine-tune/events.out.tfevents.1730235167.IPEG-WS-008.27296.0 b/logs/text2image-fine-tune/events.out.tfevents.1730235167.IPEG-WS-008.27296.0 new file mode 100644 index 0000000000000000000000000000000000000000..2bb19fc81bbb5f5d223c4087e6126645e34539bd --- /dev/null +++ b/logs/text2image-fine-tune/events.out.tfevents.1730235167.IPEG-WS-008.27296.0 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c3e02517f3509402437d9da72765d215f488aa77c530aafbde330bca0fb40bda +size 12059626 diff --git a/logs/text2image-fine-tune/events.out.tfevents.1730240791.IPEG-WS-008.22768.0 b/logs/text2image-fine-tune/events.out.tfevents.1730240791.IPEG-WS-008.22768.0 new file mode 100644 index 0000000000000000000000000000000000000000..b9e079458236f92bc4b65953fd65e9c481829d50 --- /dev/null +++ b/logs/text2image-fine-tune/events.out.tfevents.1730240791.IPEG-WS-008.22768.0 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:ece5273dfc17ad3aa170e3b2ea9f9c523668c3562499c1cc143d59eba8b505dc +size 6306183 diff --git a/logs/text2image-fine-tune/events.out.tfevents.1730246792.IPEG-WS-008.27212.0 b/logs/text2image-fine-tune/events.out.tfevents.1730246792.IPEG-WS-008.27212.0 new file mode 100644 index 0000000000000000000000000000000000000000..5c81a7cc8abe9d9ffa0118fe35895fd10f9de7fb --- /dev/null +++ b/logs/text2image-fine-tune/events.out.tfevents.1730246792.IPEG-WS-008.27212.0 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:32ac23cad136452200e0d001484ad29a0773d25f80b9340a2308fb2c66e832ee +size 20774440 diff --git a/logs/text2image-fine-tune/events.out.tfevents.1730256973.IPEG-WS-008.24668.0 b/logs/text2image-fine-tune/events.out.tfevents.1730256973.IPEG-WS-008.24668.0 new file mode 100644 index 0000000000000000000000000000000000000000..b39c1a83cce29fc3f257c4f1c053e0f0a6a26c96 --- /dev/null +++ b/logs/text2image-fine-tune/events.out.tfevents.1730256973.IPEG-WS-008.24668.0 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:78b837f4f872f502d50553ee3a5334f8a266d1a52c40cace6f519c669a4b31d3 +size 10545185 diff --git a/model_index.json b/model_index.json index e1268d2238edb06ff0c2af73b38b539f3bb27a70..7fe9d09ef6e371a328fed822ea04116458941be2 100644 --- a/model_index.json +++ b/model_index.json @@ -1,7 +1,7 @@ { "_class_name": "StableDiffusionPipeline", "_diffusers_version": "0.32.0.dev0", - "_name_or_path": "CompVis/stable-diffusion-v1-4", + "_name_or_path": "runwayml/stable-diffusion-v1-5", "feature_extractor": [ "transformers", "CLIPImageProcessor" diff --git a/safety_checker/config.json b/safety_checker/config.json index 167067be15139794caacd29e7d6350c1c654a3fc..a94851f85f21702b86475cbfd7ed92c57f637316 100644 --- a/safety_checker/config.json +++ b/safety_checker/config.json @@ -1,5 +1,5 @@ { - "_name_or_path": "C:\\Users\\grabe_stud\\.cache\\huggingface\\hub\\models--CompVis--stable-diffusion-v1-4\\snapshots\\133a221b8aa7292a167afc5127cb63fb5005638b\\safety_checker", + "_name_or_path": "C:\\Users\\grabe_stud\\.cache\\huggingface\\hub\\models--runwayml--stable-diffusion-v1-5\\snapshots\\f03de327dd89b501a01da37fc5240cf4fdba85a1\\safety_checker", "architectures": [ "StableDiffusionSafetyChecker" ], diff --git a/text_encoder/config.json b/text_encoder/config.json index 83f769294a7bc0cd3c1333c430118ce4c1ff0991..d49e3f4a3f69b33ac03bf0dcdf4265aeb8e20cf5 100644 --- a/text_encoder/config.json +++ b/text_encoder/config.json @@ -1,5 +1,5 @@ { - "_name_or_path": "CompVis/stable-diffusion-v1-4", + "_name_or_path": "runwayml/stable-diffusion-v1-5", "architectures": [ "CLIPTextModel" ], @@ -18,7 +18,7 @@ "num_attention_heads": 12, "num_hidden_layers": 12, "pad_token_id": 1, - "projection_dim": 512, + "projection_dim": 768, "torch_dtype": "float16", "transformers_version": "4.45.2", "vocab_size": 49408 diff --git a/unet/config.json b/unet/config.json index 4872afd6ff2f1ca9a40ed0bca31b5449b1760578..7635fba9f6130f3988be75419c7700d9db25e562 100644 --- a/unet/config.json +++ b/unet/config.json @@ -1,7 +1,7 @@ { "_class_name": "UNet2DConditionModel", "_diffusers_version": "0.32.0.dev0", - "_name_or_path": "CompVis/stable-diffusion-v1-4", + "_name_or_path": "runwayml/stable-diffusion-v1-5", "act_fn": "silu", "addition_embed_type": null, "addition_embed_type_num_heads": 64, diff --git a/unet/diffusion_pytorch_model.safetensors b/unet/diffusion_pytorch_model.safetensors index c8e8415a27cfb9826a3a1a751e6972a527bcb813..72324515e6f05acf68f6cbb8e2c12eb9d97b45ce 100644 --- a/unet/diffusion_pytorch_model.safetensors +++ b/unet/diffusion_pytorch_model.safetensors @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:52899fbeea249abae4130e5fc0cc69337da1cce17cffda3d783d6775fe2f393f +oid sha256:c03d9cfd0205be3b340f96fe5988a8c688b6ab817ed88efc1f52c1c927c9616e size 3438167536 diff --git a/vae/config.json b/vae/config.json index 460c549fb8f69e7feadf50271674f2ad0fa21ecf..d52783b6e629480db0b6f7b24f800c87aec13347 100644 --- a/vae/config.json +++ b/vae/config.json @@ -1,7 +1,7 @@ { "_class_name": "AutoencoderKL", "_diffusers_version": "0.32.0.dev0", - "_name_or_path": "CompVis/stable-diffusion-v1-4", + "_name_or_path": "runwayml/stable-diffusion-v1-5", "act_fn": "silu", "block_out_channels": [ 128, diff --git a/val_imgs_grid.png b/val_imgs_grid.png index 7ce68181c2a7c7b040c56ef2fe1893ab6b4fd1cc..0e1dda0f8ecfbf41f79555dcde55cc7bb25037fd 100644 Binary files a/val_imgs_grid.png and b/val_imgs_grid.png differ