diff --git a/README.md b/README.md index 1c8ebfbead7a7b5e35faa81d774c04fb3291fd7e..c1d6bf3d9b1c4e452e635b3ed241018bcabb14fa 100644 --- a/README.md +++ b/README.md @@ -1,7 +1,7 @@ --- license: creativeml-openrail-m -base_model: runwayml/stable-diffusion-v1-5 +base_model: crom87/sd_base-db-selfies4-1e-06-priorp-100-gc instance_prompt: TOKstyle georgeclooney tags: - stable-diffusion @@ -14,21 +14,9 @@ inference: true # DreamBooth - crom87/sd_base-db-selfies4-1e-06-priorp-100-gc -This is a dreambooth model derived from runwayml/stable-diffusion-v1-5. The weights were trained on TOKstyle georgeclooney using [DreamBooth](https://dreambooth.github.io/). +This is a dreambooth model derived from crom87/sd_base-db-selfies4-1e-06-priorp-100-gc. The weights were trained on TOKstyle georgeclooney using [DreamBooth](https://dreambooth.github.io/). You can find some example images in the following. -![img_0](./image_0.png) -![img_1](./image_1.png) -![img_2](./image_2.png) -![img_3](./image_3.png) -![img_4](./image_4.png) -![img_5](./image_5.png) -![img_6](./image_6.png) -![img_7](./image_7.png) -![img_8](./image_8.png) -![img_9](./image_9.png) -![img_10](./image_10.png) -![img_11](./image_11.png) DreamBooth for the text encoder was enabled: True. diff --git a/checkpoint-1000/optimizer.bin b/checkpoint-1000/optimizer.bin index 007f9bf6af50d1729e073bf54f00e76f14f5180e..da30b503858415a1d0447d0839c4ab49c0a3efd4 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:95a82bf1b72577925d7301867ab408b822b0c7271ff64b967aaf2576aa084cd5 +oid sha256:2dd5ac3cd96e7227e2663bf94b04a5fb4a4823493b5de6d8798022673b9954c8 size 1972598422 diff --git a/checkpoint-1000/random_states_0.pkl b/checkpoint-1000/random_states_0.pkl index b4a27f9b0d1259424b3abe66ef4817afa812196e..3d631f43cc7d9954167cf04991e49e0733e41562 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:950bc0547accca960f96005db4e96de299045ec924c76e138ed23307cd83da5a +oid sha256:9128549b0e977d02a1fe61ee230d80b247e2c4d30d2cf595b247f85650f22a7d size 14344 diff --git a/checkpoint-1000/scheduler.bin b/checkpoint-1000/scheduler.bin index c497e7a070f42af3631c6af9a20001bf6c2fddb9..1487b066ca28a560852c5e73d3ad5dcf08805936 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:3cc9d529863b9458dbb9b91d10c2e89741ffa38f99dbc44be25770289ed29416 +oid sha256:3e36a7249b5e783f64c0e801daf11cc70246e8a1ac48638aea268a17bbe20f36 size 1000 diff --git a/checkpoint-1000/text_encoder/config.json b/checkpoint-1000/text_encoder/config.json index 1bd2ec5b8f36b41c778d23c96db888069188deb9..40471de7789e1ca424aefef71b8dc124b869f269 100644 --- a/checkpoint-1000/text_encoder/config.json +++ b/checkpoint-1000/text_encoder/config.json @@ -1,5 +1,5 @@ { - "_name_or_path": "runwayml/stable-diffusion-v1-5", + "_name_or_path": "crom87/sd_base-db-selfies4-1e-06-priorp-100-gc", "architectures": [ "CLIPTextModel" ], diff --git a/checkpoint-1000/text_encoder/model.safetensors b/checkpoint-1000/text_encoder/model.safetensors index ac7d4dc2786b374a140bd6066a18fc9101a35809..569fe8802c856940d481a8e7091e9bffdb9f799f 100644 --- a/checkpoint-1000/text_encoder/model.safetensors +++ b/checkpoint-1000/text_encoder/model.safetensors @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:7d323651f8e35a07ee8228b2beea52dfe1c64fc711823453a3267a5cf9d6bed7 +oid sha256:ce6c50836dd7bb59735fc231e9a9208b5ffe5e5eed6da688331bcdcd1e5c7c1b size 492265168 diff --git a/checkpoint-1000/unet/config.json b/checkpoint-1000/unet/config.json index adad02d1cdec217d04412031372cc60434aaedce..fe43629b8071d2d7c03ac116d928f403c7772c6b 100644 --- a/checkpoint-1000/unet/config.json +++ b/checkpoint-1000/unet/config.json @@ -1,7 +1,7 @@ { "_class_name": "UNet2DConditionModel", "_diffusers_version": "0.23.1", - "_name_or_path": "runwayml/stable-diffusion-v1-5", + "_name_or_path": "crom87/sd_base-db-selfies4-1e-06-priorp-100-gc", "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 2c5cb72c41f8730dd908b1d772792aa3d5d0b852..33b2baf29376810e04dc1bd156534fdf14fcb0a6 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:1ef6e6ce917966f052ff288a314832447009f85fe05e439fd1d77e8457dfda8b +oid sha256:406a92e309f1107d320cd5aff9d064099b640df2b2b8bcba05e9f7e08dee828a size 3438167536 diff --git a/checkpoint-1500/optimizer.bin b/checkpoint-1500/optimizer.bin index e53e8166104f1a9729832513893436190055fdf8..d8c9db32fb3a6d606a65a2537e1c4d49119e0a23 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:55ac3f7cf149e95cf17eff0246bf94ed26d9b5a4cc320a8de25b7db8cbc2f718 +oid sha256:fadc113337a2155ee7e20e9cad443d0337eb8dd7a6f207d922a74021af14f5d8 size 1972598422 diff --git a/checkpoint-1500/random_states_0.pkl b/checkpoint-1500/random_states_0.pkl index 6d0d27dc393f34cb17e721b07a39f88dc3835d79..78ed2201b125a554f8a886f8169afacfe9ff27e4 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:18fe3fded2f77d5a63a61e47cb020a2f48d556ff73c46e527541fe392a8b4ad4 +oid sha256:b6d06c2c3d36a524751b2deb88e0161b9432b9b3b13207b9e93050380180b277 size 14344 diff --git a/checkpoint-1500/scheduler.bin b/checkpoint-1500/scheduler.bin index 7aa8e90feaa2be810f1ff1362d6cc185e60bc845..112915f939d8850db5e5fe284fa28a08c2ca9ff4 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:511502cb8e486bc7704b13f50561c780285bc0785605731cccd9a6fe2798bb6b +oid sha256:8f8108c9f3ee073c12b3784e0f4fdb33e21478fc1e37ff98e67bf025cabaa140 size 1000 diff --git a/checkpoint-1500/text_encoder/config.json b/checkpoint-1500/text_encoder/config.json index 1bd2ec5b8f36b41c778d23c96db888069188deb9..40471de7789e1ca424aefef71b8dc124b869f269 100644 --- a/checkpoint-1500/text_encoder/config.json +++ b/checkpoint-1500/text_encoder/config.json @@ -1,5 +1,5 @@ { - "_name_or_path": "runwayml/stable-diffusion-v1-5", + "_name_or_path": "crom87/sd_base-db-selfies4-1e-06-priorp-100-gc", "architectures": [ "CLIPTextModel" ], diff --git a/checkpoint-1500/text_encoder/model.safetensors b/checkpoint-1500/text_encoder/model.safetensors index d6b88770fd4fcb75d5406d76aece00e70c42fed4..ee3c66a72a1f33b5da5f6cb5acfd6721163f44a1 100644 --- a/checkpoint-1500/text_encoder/model.safetensors +++ b/checkpoint-1500/text_encoder/model.safetensors @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:616f6fd55074cdb89184c7fbdd678d7a4193b13ce540a303ef2740542217dd0f +oid sha256:c3602b2cc27e8e4139bd2a933559d8f6c5a4f15a9b6774e797399639fd30f6ca size 492265168 diff --git a/checkpoint-1500/unet/config.json b/checkpoint-1500/unet/config.json index adad02d1cdec217d04412031372cc60434aaedce..fe43629b8071d2d7c03ac116d928f403c7772c6b 100644 --- a/checkpoint-1500/unet/config.json +++ b/checkpoint-1500/unet/config.json @@ -1,7 +1,7 @@ { "_class_name": "UNet2DConditionModel", "_diffusers_version": "0.23.1", - "_name_or_path": "runwayml/stable-diffusion-v1-5", + "_name_or_path": "crom87/sd_base-db-selfies4-1e-06-priorp-100-gc", "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 4a75997af3df72cc68b8d2d29fc0d1bffde85d74..bd2fa5544a2cfa7e756c409db024265caa9f70e0 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:37971510ad440d8e668b94ead1753accb18cac31724601d4e623ae1df50078f5 +oid sha256:1319347ae16b4748ce42eb19da6c973e6224d998731dd03cf1e8ebe9b7abe8fd size 3438167536 diff --git a/checkpoint-2000/optimizer.bin b/checkpoint-2000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..f7d28ff5132fe6d83d177185c0b340d8f495d863 --- /dev/null +++ b/checkpoint-2000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:ec271c78c17a1cc8ea69817ec721d7325500ce4446e28c2f87520f8a788e61e7 +size 1972598422 diff --git a/checkpoint-2000/random_states_0.pkl b/checkpoint-2000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..6ffbe712db3069738e1fe6d6574770d0a71ead01 --- /dev/null +++ b/checkpoint-2000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:318a0c02d9d72d3bf3598250c829fdf40570c9f1cac74b49236353a04c36b101 +size 14344 diff --git a/checkpoint-2000/scaler.pt b/checkpoint-2000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..99282b0e304ef68a38f5a2dd4a6ac123700f2f53 --- /dev/null +++ b/checkpoint-2000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:3c50a9cebe5d66d453d25b140738bff479749ac03e0a43597d8776bc22f6ed0c +size 988 diff --git a/checkpoint-2000/scheduler.bin b/checkpoint-2000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..6dc6ebd68f5e303eb9ab0b1850ff5a538d2eb3c5 --- /dev/null +++ b/checkpoint-2000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:ab4b295f733a984101b9ee74fddce9007b95aad8c979511e8676285d70d391b2 +size 1000 diff --git a/checkpoint-2000/text_encoder/config.json b/checkpoint-2000/text_encoder/config.json new file mode 100644 index 0000000000000000000000000000000000000000..40471de7789e1ca424aefef71b8dc124b869f269 --- /dev/null +++ b/checkpoint-2000/text_encoder/config.json @@ -0,0 +1,25 @@ +{ + "_name_or_path": "crom87/sd_base-db-selfies4-1e-06-priorp-100-gc", + "architectures": [ + "CLIPTextModel" + ], + "attention_dropout": 0.0, + "bos_token_id": 0, + "dropout": 0.0, + "eos_token_id": 2, + "hidden_act": "quick_gelu", + "hidden_size": 768, + "initializer_factor": 1.0, + "initializer_range": 0.02, + "intermediate_size": 3072, + "layer_norm_eps": 1e-05, + "max_position_embeddings": 77, + "model_type": "clip_text_model", + "num_attention_heads": 12, + "num_hidden_layers": 12, + "pad_token_id": 1, + "projection_dim": 768, + "torch_dtype": "float32", + "transformers_version": "4.35.2", + "vocab_size": 49408 +} diff --git a/checkpoint-2000/text_encoder/model.safetensors b/checkpoint-2000/text_encoder/model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..083a06455cbe952b94898e73f5a07d5b27ff0f47 --- /dev/null +++ b/checkpoint-2000/text_encoder/model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:0d1ac28c3129f726347318495ec1ce7339cc020be466d51209eaab737b7dde60 +size 492265168 diff --git a/checkpoint-2000/unet/config.json b/checkpoint-2000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..fe43629b8071d2d7c03ac116d928f403c7772c6b --- /dev/null +++ b/checkpoint-2000/unet/config.json @@ -0,0 +1,68 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.23.1", + "_name_or_path": "crom87/sd_base-db-selfies4-1e-06-priorp-100-gc", + "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..f073a6610e90218e2d732c4b8ccd8f93d64ccb99 --- /dev/null +++ b/checkpoint-2000/unet/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:f8b35a6aa057c11076e38874f2a9bcf5adbbc4b9b42793bd8632dcc193976e22 +size 3438167536 diff --git a/checkpoint-2500/optimizer.bin b/checkpoint-2500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..4f355fcb8cdd61bf8833004847b147a79aeae491 --- /dev/null +++ b/checkpoint-2500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:8814111ec12ebcb9865a6188f208a9c39f903afe5f513238cd3af1213212bf72 +size 1972598422 diff --git a/checkpoint-2500/random_states_0.pkl b/checkpoint-2500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..7e49ec97ba8ca8ea4c36b731d1a96728dab87b82 --- /dev/null +++ b/checkpoint-2500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:6a5024b6f5fa904b77fd5f489a8c128787598c9ff1b52a5efdeb61fb0691d7ea +size 14344 diff --git a/checkpoint-2500/scaler.pt b/checkpoint-2500/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..4b9902c4e2dee2972ff0943d57b3678364435cdf --- /dev/null +++ b/checkpoint-2500/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:48e2d97f563bb838328076a1666504681962151a3975a2f064be3a03e6500740 +size 988 diff --git a/checkpoint-2500/scheduler.bin b/checkpoint-2500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..6b4cb13ae206a0cca1105a34d103aed67a3bb36f --- /dev/null +++ b/checkpoint-2500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:3eb36b143f6f6bcf02310ded9ee4b045d16d7e6a70ace4a142cebc8f13c3c5a9 +size 1000 diff --git a/checkpoint-2500/text_encoder/config.json b/checkpoint-2500/text_encoder/config.json new file mode 100644 index 0000000000000000000000000000000000000000..40471de7789e1ca424aefef71b8dc124b869f269 --- /dev/null +++ b/checkpoint-2500/text_encoder/config.json @@ -0,0 +1,25 @@ +{ + "_name_or_path": "crom87/sd_base-db-selfies4-1e-06-priorp-100-gc", + "architectures": [ + "CLIPTextModel" + ], + "attention_dropout": 0.0, + "bos_token_id": 0, + "dropout": 0.0, + "eos_token_id": 2, + "hidden_act": "quick_gelu", + "hidden_size": 768, + "initializer_factor": 1.0, + "initializer_range": 0.02, + "intermediate_size": 3072, + "layer_norm_eps": 1e-05, + "max_position_embeddings": 77, + "model_type": "clip_text_model", + "num_attention_heads": 12, + "num_hidden_layers": 12, + "pad_token_id": 1, + "projection_dim": 768, + "torch_dtype": "float32", + "transformers_version": "4.35.2", + "vocab_size": 49408 +} diff --git a/checkpoint-2500/text_encoder/model.safetensors b/checkpoint-2500/text_encoder/model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..99ffb1ffd93c48a3ebdb3b27eaa05c387691ac1c --- /dev/null +++ b/checkpoint-2500/text_encoder/model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:bbf2d27133f6e7e7a61e96763f253d76306d55702cee0f0c7dc9969cfbee3c16 +size 492265168 diff --git a/checkpoint-2500/unet/config.json b/checkpoint-2500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..fe43629b8071d2d7c03ac116d928f403c7772c6b --- /dev/null +++ b/checkpoint-2500/unet/config.json @@ -0,0 +1,68 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.23.1", + "_name_or_path": "crom87/sd_base-db-selfies4-1e-06-priorp-100-gc", + "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-2500/unet/diffusion_pytorch_model.safetensors b/checkpoint-2500/unet/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..9f0bbd0df6916cc8b26ebbf56fc52327c248acaf --- /dev/null +++ b/checkpoint-2500/unet/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:fbd6702c54b0b11537c5fb5580b838026155ddc9947396c9f1ace2355d74a92b +size 3438167536 diff --git a/checkpoint-3000/optimizer.bin b/checkpoint-3000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..176ab1c92e56a143d06c175bc78f456c5946bd1d --- /dev/null +++ b/checkpoint-3000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c8c7d5dda3b813f2751cc77c1e6580e83f529faf2e28fefe676c8a403023dff3 +size 1972598422 diff --git a/checkpoint-3000/random_states_0.pkl b/checkpoint-3000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..897174fb75001b1f5db1eab8d86ba23815f7acae --- /dev/null +++ b/checkpoint-3000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:381569c59fda8f49ae8429ca1c2107686d1e265468aade17b7320d7a6b7f17fd +size 14344 diff --git a/checkpoint-3000/scaler.pt b/checkpoint-3000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..6c13b5448a0852139c8a78d48ae5f4dc71f8a023 --- /dev/null +++ b/checkpoint-3000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:21aba8ed0f38ed1c04994c10a9ca7e9925e55ef2ed51283c43ff8e2cce78585f +size 988 diff --git a/checkpoint-3000/scheduler.bin b/checkpoint-3000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..e2dc60dd47f9eefa849085610be6f8db14b6aace --- /dev/null +++ b/checkpoint-3000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:6c772b33d2c1b5cb1bd9d21f99842709beccfe980b27c16da6393109c6d968d5 +size 1000 diff --git a/checkpoint-3000/text_encoder/config.json b/checkpoint-3000/text_encoder/config.json new file mode 100644 index 0000000000000000000000000000000000000000..40471de7789e1ca424aefef71b8dc124b869f269 --- /dev/null +++ b/checkpoint-3000/text_encoder/config.json @@ -0,0 +1,25 @@ +{ + "_name_or_path": "crom87/sd_base-db-selfies4-1e-06-priorp-100-gc", + "architectures": [ + "CLIPTextModel" + ], + "attention_dropout": 0.0, + "bos_token_id": 0, + "dropout": 0.0, + "eos_token_id": 2, + "hidden_act": "quick_gelu", + "hidden_size": 768, + "initializer_factor": 1.0, + "initializer_range": 0.02, + "intermediate_size": 3072, + "layer_norm_eps": 1e-05, + "max_position_embeddings": 77, + "model_type": "clip_text_model", + "num_attention_heads": 12, + "num_hidden_layers": 12, + "pad_token_id": 1, + "projection_dim": 768, + "torch_dtype": "float32", + "transformers_version": "4.35.2", + "vocab_size": 49408 +} diff --git a/checkpoint-3000/text_encoder/model.safetensors b/checkpoint-3000/text_encoder/model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..293333f075319dabca0ddff5c3bd0436d14a83f8 --- /dev/null +++ b/checkpoint-3000/text_encoder/model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:896e3a9246950fff903e21a07e9b02a0548d008547c4ac8561e516ee2a0e64aa +size 492265168 diff --git a/checkpoint-3000/unet/config.json b/checkpoint-3000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..fe43629b8071d2d7c03ac116d928f403c7772c6b --- /dev/null +++ b/checkpoint-3000/unet/config.json @@ -0,0 +1,68 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.23.1", + "_name_or_path": "crom87/sd_base-db-selfies4-1e-06-priorp-100-gc", + "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-3000/unet/diffusion_pytorch_model.safetensors b/checkpoint-3000/unet/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..b2001ee6b69a8d716cc0745838f91d14912c6539 --- /dev/null +++ b/checkpoint-3000/unet/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:acf522c9537128426f40d8d7919cd79575bfe24b346a9abf3264029b21c2c7c5 +size 3438167536 diff --git a/checkpoint-3500/optimizer.bin b/checkpoint-3500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..4afaf5f5d1e0cd7c7a9eeeca6815804704962751 --- /dev/null +++ b/checkpoint-3500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:5159dc8cf2ef754f9351bdb4d603e18a56769176a176797ebafc11ef11aa0696 +size 1972598422 diff --git a/checkpoint-3500/random_states_0.pkl b/checkpoint-3500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..795c811bf4d519711a5842f09192b633eeb61f7b --- /dev/null +++ b/checkpoint-3500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c69e002c37e3356c95fe2598b74694760c83cd42c7e1d29e3d94899e13d01057 +size 14344 diff --git a/checkpoint-3500/scaler.pt b/checkpoint-3500/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..642d9ddc499e11af6d65baf9612222717fa34b7d --- /dev/null +++ b/checkpoint-3500/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:9d25c9e5c384ba91142c829ef5432ebc4ae7d8c71f3de723046dd3aa202e08a2 +size 988 diff --git a/checkpoint-3500/scheduler.bin b/checkpoint-3500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..87a8d92cd677a0ee09b09838d26dbb93ea22393e --- /dev/null +++ b/checkpoint-3500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:cbbeac5d69a9a48ec6877ac36a96d71d0dfd06a1b8871ac96b299c3192f8ca0d +size 1000 diff --git a/checkpoint-3500/text_encoder/config.json b/checkpoint-3500/text_encoder/config.json new file mode 100644 index 0000000000000000000000000000000000000000..40471de7789e1ca424aefef71b8dc124b869f269 --- /dev/null +++ b/checkpoint-3500/text_encoder/config.json @@ -0,0 +1,25 @@ +{ + "_name_or_path": "crom87/sd_base-db-selfies4-1e-06-priorp-100-gc", + "architectures": [ + "CLIPTextModel" + ], + "attention_dropout": 0.0, + "bos_token_id": 0, + "dropout": 0.0, + "eos_token_id": 2, + "hidden_act": "quick_gelu", + "hidden_size": 768, + "initializer_factor": 1.0, + "initializer_range": 0.02, + "intermediate_size": 3072, + "layer_norm_eps": 1e-05, + "max_position_embeddings": 77, + "model_type": "clip_text_model", + "num_attention_heads": 12, + "num_hidden_layers": 12, + "pad_token_id": 1, + "projection_dim": 768, + "torch_dtype": "float32", + "transformers_version": "4.35.2", + "vocab_size": 49408 +} diff --git a/checkpoint-3500/text_encoder/model.safetensors b/checkpoint-3500/text_encoder/model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..a3031e80320d59a01ea57bcadb7a5d1c1aad780c --- /dev/null +++ b/checkpoint-3500/text_encoder/model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:41d6b5c22cf725a1c047061d25c1d4b9774df536fd0cc12c471d4f2509f6aedc +size 492265168 diff --git a/checkpoint-3500/unet/config.json b/checkpoint-3500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..fe43629b8071d2d7c03ac116d928f403c7772c6b --- /dev/null +++ b/checkpoint-3500/unet/config.json @@ -0,0 +1,68 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.23.1", + "_name_or_path": "crom87/sd_base-db-selfies4-1e-06-priorp-100-gc", + "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..124228181783fa4e5783b6d0e959d813f39d9df7 --- /dev/null +++ b/checkpoint-3500/unet/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:349a866d1718a60e06cb7232a550c49e3753bce31db4b37fc0f6e0f79448ee14 +size 3438167536 diff --git a/checkpoint-4000/optimizer.bin b/checkpoint-4000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..14cf6df909517e903058c674de2e464f3ec06f35 --- /dev/null +++ b/checkpoint-4000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:a0ef9c5210e149fea4d22ab4d77c65c3b38d4740d3ea74d1b2380f04a8e1b053 +size 1972598422 diff --git a/checkpoint-4000/random_states_0.pkl b/checkpoint-4000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..c7318dd2df551bbe318e02e6d3c3bfaea467a5bd --- /dev/null +++ b/checkpoint-4000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:0bb012d0c23f9cb722f8564effe26b246b06d0899233efb91a7dbd0242173247 +size 14344 diff --git a/checkpoint-4000/scaler.pt b/checkpoint-4000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..8e3c148798a27b525e5967dae1f764aa32ba855f --- /dev/null +++ b/checkpoint-4000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:59a4acfb0da74c479080613978839dd3cbb4608fd2c07e764b4c844401d8dd5f +size 988 diff --git a/checkpoint-4000/scheduler.bin b/checkpoint-4000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..16b5f89839dbf6e82f79bff5b795a24c14311e94 --- /dev/null +++ b/checkpoint-4000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:d973fe349997563d1e89ce50a1d2a191c9252dfa6ca467e32f9d204a4a251a8e +size 1000 diff --git a/checkpoint-4000/text_encoder/config.json b/checkpoint-4000/text_encoder/config.json new file mode 100644 index 0000000000000000000000000000000000000000..40471de7789e1ca424aefef71b8dc124b869f269 --- /dev/null +++ b/checkpoint-4000/text_encoder/config.json @@ -0,0 +1,25 @@ +{ + "_name_or_path": "crom87/sd_base-db-selfies4-1e-06-priorp-100-gc", + "architectures": [ + "CLIPTextModel" + ], + "attention_dropout": 0.0, + "bos_token_id": 0, + "dropout": 0.0, + "eos_token_id": 2, + "hidden_act": "quick_gelu", + "hidden_size": 768, + "initializer_factor": 1.0, + "initializer_range": 0.02, + "intermediate_size": 3072, + "layer_norm_eps": 1e-05, + "max_position_embeddings": 77, + "model_type": "clip_text_model", + "num_attention_heads": 12, + "num_hidden_layers": 12, + "pad_token_id": 1, + "projection_dim": 768, + "torch_dtype": "float32", + "transformers_version": "4.35.2", + "vocab_size": 49408 +} diff --git a/checkpoint-4000/text_encoder/model.safetensors b/checkpoint-4000/text_encoder/model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..5336f1ba71b8af11664ef76438fef90a2dfeb3e4 --- /dev/null +++ b/checkpoint-4000/text_encoder/model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:a6bbe09ac0d94c927b386f1b2e578629b727933c2992c6a34b8b7674b27a7de3 +size 492265168 diff --git a/checkpoint-4000/unet/config.json b/checkpoint-4000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..fe43629b8071d2d7c03ac116d928f403c7772c6b --- /dev/null +++ b/checkpoint-4000/unet/config.json @@ -0,0 +1,68 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.23.1", + "_name_or_path": "crom87/sd_base-db-selfies4-1e-06-priorp-100-gc", + "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..955b2a35a5d343b4b7438d16b6900252f982ab64 --- /dev/null +++ b/checkpoint-4000/unet/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c0e194df74287eb7317c92baf28884e37af56e31b66174548be7750d6e93f494 +size 3438167536 diff --git a/checkpoint-4500/optimizer.bin b/checkpoint-4500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..a39f2ca583a66bd2ab0f0cad13bc165f371fe449 --- /dev/null +++ b/checkpoint-4500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:8075805926f529a3afd5ce02ff12b159fbbc707ad7950e3f2a4dd79b43a50a36 +size 1972598422 diff --git a/checkpoint-4500/random_states_0.pkl b/checkpoint-4500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..165faf876067ae217b0bdf0e490c0b61652f6ca1 --- /dev/null +++ b/checkpoint-4500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:bca1fbe79c8cbda5d12d21d56274c2b09214853d675ce182b3b361b2dc057132 +size 14344 diff --git a/checkpoint-4500/scaler.pt b/checkpoint-4500/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..0eb1718239e9101b6a6e609988d0ab4c61c50ef6 --- /dev/null +++ b/checkpoint-4500/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:25aed798df1ea1f11d3cdd0a7de18dd9b632beaa14981af70f644c899539b4cc +size 988 diff --git a/checkpoint-4500/scheduler.bin b/checkpoint-4500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..b363d0c86903459864f6aa94f077c722d0193ae5 --- /dev/null +++ b/checkpoint-4500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:f02cf01510d68fc1c326cfbf38e5502e3711a5f8683b4dcd8d0fa73169edd64b +size 1000 diff --git a/checkpoint-4500/text_encoder/config.json b/checkpoint-4500/text_encoder/config.json new file mode 100644 index 0000000000000000000000000000000000000000..40471de7789e1ca424aefef71b8dc124b869f269 --- /dev/null +++ b/checkpoint-4500/text_encoder/config.json @@ -0,0 +1,25 @@ +{ + "_name_or_path": "crom87/sd_base-db-selfies4-1e-06-priorp-100-gc", + "architectures": [ + "CLIPTextModel" + ], + "attention_dropout": 0.0, + "bos_token_id": 0, + "dropout": 0.0, + "eos_token_id": 2, + "hidden_act": "quick_gelu", + "hidden_size": 768, + "initializer_factor": 1.0, + "initializer_range": 0.02, + "intermediate_size": 3072, + "layer_norm_eps": 1e-05, + "max_position_embeddings": 77, + "model_type": "clip_text_model", + "num_attention_heads": 12, + "num_hidden_layers": 12, + "pad_token_id": 1, + "projection_dim": 768, + "torch_dtype": "float32", + "transformers_version": "4.35.2", + "vocab_size": 49408 +} diff --git a/checkpoint-4500/text_encoder/model.safetensors b/checkpoint-4500/text_encoder/model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..a3e6169c5085f1fe548b4a31a6a2835218e912c7 --- /dev/null +++ b/checkpoint-4500/text_encoder/model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:a46c7425b38ad4ca0fb43e13faee142c3e669683b9e292ff457277b13ff21dee +size 492265168 diff --git a/checkpoint-4500/unet/config.json b/checkpoint-4500/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..fe43629b8071d2d7c03ac116d928f403c7772c6b --- /dev/null +++ b/checkpoint-4500/unet/config.json @@ -0,0 +1,68 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.23.1", + "_name_or_path": "crom87/sd_base-db-selfies4-1e-06-priorp-100-gc", + "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-4500/unet/diffusion_pytorch_model.safetensors b/checkpoint-4500/unet/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..1abfa3529266dcd7024b3c285ba4d7c7229bec0c --- /dev/null +++ b/checkpoint-4500/unet/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:4ccc6d51d0bd1773619bb99cc78eba782523f453815dc8e0c9d0d01cff78d1a9 +size 3438167536 diff --git a/checkpoint-500/optimizer.bin b/checkpoint-500/optimizer.bin index da87744eea8bda24b0d60b1ff819d50090d5333c..d2531fe530306ab947fbf1db2b4352852bd5e5ac 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:89cca5a5383fec6d8c11ebe5cf4b7d0ba350093b96b2b8db698031b75cea32b2 +oid sha256:5846a9673e82c99ed60e94c4d1cdba82c6a0ec98c156338859c2540c158244bf size 1972598422 diff --git a/checkpoint-500/random_states_0.pkl b/checkpoint-500/random_states_0.pkl index fc0cff16d6bf06156d14e718c8718a33bd703317..11e3621ba959b3838f0e7fd6a0bea51af183e50c 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:80aee2e3baf03d1009ee4c874688e511984797bf20f9102b38e51283ce44b225 +oid sha256:a172fbd8904717d6e76600376f496c36c6cddd441781bb283b8fd9b2a13344a4 size 14344 diff --git a/checkpoint-500/scheduler.bin b/checkpoint-500/scheduler.bin index 1c524c45bedbefb2a8dac988a06c1693f046a173..25ef8afe13c726fcce3b2f18b32a7ffde048de07 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:1ff7e0081931acdf638a5b6dc6f89d270dc53395b51e189cd30a4b0fd9d14285 +oid sha256:298e7e75502de41a98ece7ac5311b3b598cf308af851c910dc4da2b9a52082ad size 1000 diff --git a/checkpoint-500/text_encoder/config.json b/checkpoint-500/text_encoder/config.json index 1bd2ec5b8f36b41c778d23c96db888069188deb9..40471de7789e1ca424aefef71b8dc124b869f269 100644 --- a/checkpoint-500/text_encoder/config.json +++ b/checkpoint-500/text_encoder/config.json @@ -1,5 +1,5 @@ { - "_name_or_path": "runwayml/stable-diffusion-v1-5", + "_name_or_path": "crom87/sd_base-db-selfies4-1e-06-priorp-100-gc", "architectures": [ "CLIPTextModel" ], diff --git a/checkpoint-500/text_encoder/model.safetensors b/checkpoint-500/text_encoder/model.safetensors index d98f265beb9387d38c789cfa462e9fde75020396..c3704438d954b5c49ecee4a5fd9c3b21714d6621 100644 --- a/checkpoint-500/text_encoder/model.safetensors +++ b/checkpoint-500/text_encoder/model.safetensors @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:edd8dec0e3bdf7ce22441b6e13a5cbe0e3ad1e9e3cab34a424f8ecbf2558910a +oid sha256:d4ddd669f8dbd0b4d0850d38da9d89973b5c2f8e62dc5dae90bcee823638cc94 size 492265168 diff --git a/checkpoint-500/unet/config.json b/checkpoint-500/unet/config.json index adad02d1cdec217d04412031372cc60434aaedce..fe43629b8071d2d7c03ac116d928f403c7772c6b 100644 --- a/checkpoint-500/unet/config.json +++ b/checkpoint-500/unet/config.json @@ -1,7 +1,7 @@ { "_class_name": "UNet2DConditionModel", "_diffusers_version": "0.23.1", - "_name_or_path": "runwayml/stable-diffusion-v1-5", + "_name_or_path": "crom87/sd_base-db-selfies4-1e-06-priorp-100-gc", "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 e38a93952dd58da747463ee140ccbf11e5e088d3..1d076d0011a368496399a700f09574011841c78a 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:b5e4a6073704d3d3170535f40f77dacbcc929be5bff85aa2a52571bbfc6fe5cf +oid sha256:8f02d4428ac33e146c4fa74c58edb2db7a12fc323e8e69996a3107bd091c7860 size 3438167536 diff --git a/checkpoint-5000/optimizer.bin b/checkpoint-5000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..4cfd0e9bdf91c5bccbb52992bce74d0672494cba --- /dev/null +++ b/checkpoint-5000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:16d4a4145f28682e33b27862aa5ef21607831f58c883e19e423ad3add0c8094e +size 1972598422 diff --git a/checkpoint-5000/random_states_0.pkl b/checkpoint-5000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..fb5b3e55288c3f8e16624b209cbbaee96ece15cc --- /dev/null +++ b/checkpoint-5000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:cb370ab9638565873e8a52a6ef59519ba0bd104bebf0b9d8cc15b91761eec504 +size 14344 diff --git a/checkpoint-5000/scaler.pt b/checkpoint-5000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..da81849ef9b20a8870285eaa92c68ae84344e397 --- /dev/null +++ b/checkpoint-5000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:41216328ab75de937007afb04d76156949bedb908461001a451c8991c2ba8cca +size 988 diff --git a/checkpoint-5000/scheduler.bin b/checkpoint-5000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..9b98348bf874ecd5dc6edc9666563a340ec18f01 --- /dev/null +++ b/checkpoint-5000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:0d979454babc0c1f1273a1517c0ac0b9b150b6bb4b299c97c72282cd672319d4 +size 1000 diff --git a/checkpoint-5000/text_encoder/config.json b/checkpoint-5000/text_encoder/config.json new file mode 100644 index 0000000000000000000000000000000000000000..40471de7789e1ca424aefef71b8dc124b869f269 --- /dev/null +++ b/checkpoint-5000/text_encoder/config.json @@ -0,0 +1,25 @@ +{ + "_name_or_path": "crom87/sd_base-db-selfies4-1e-06-priorp-100-gc", + "architectures": [ + "CLIPTextModel" + ], + "attention_dropout": 0.0, + "bos_token_id": 0, + "dropout": 0.0, + "eos_token_id": 2, + "hidden_act": "quick_gelu", + "hidden_size": 768, + "initializer_factor": 1.0, + "initializer_range": 0.02, + "intermediate_size": 3072, + "layer_norm_eps": 1e-05, + "max_position_embeddings": 77, + "model_type": "clip_text_model", + "num_attention_heads": 12, + "num_hidden_layers": 12, + "pad_token_id": 1, + "projection_dim": 768, + "torch_dtype": "float32", + "transformers_version": "4.35.2", + "vocab_size": 49408 +} diff --git a/checkpoint-5000/text_encoder/model.safetensors b/checkpoint-5000/text_encoder/model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..2bf54dc15871a4f28f8955d7f2fa9f2cf4f3dd99 --- /dev/null +++ b/checkpoint-5000/text_encoder/model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:69e318ad4f25b6db797200f3d60bb61e362908c57231e926ccba4228695e8903 +size 492265168 diff --git a/checkpoint-5000/unet/config.json b/checkpoint-5000/unet/config.json new file mode 100644 index 0000000000000000000000000000000000000000..fe43629b8071d2d7c03ac116d928f403c7772c6b --- /dev/null +++ b/checkpoint-5000/unet/config.json @@ -0,0 +1,68 @@ +{ + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.23.1", + "_name_or_path": "crom87/sd_base-db-selfies4-1e-06-priorp-100-gc", + "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-5000/unet/diffusion_pytorch_model.safetensors b/checkpoint-5000/unet/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..81baadd900af8a8fd2724dca9a1b6dd9f52a9c6c --- /dev/null +++ b/checkpoint-5000/unet/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:66f8c34998e349efa32906c44dda0d5b783a31b6fb3610c489ac1eab771b8c0c +size 3438167536 diff --git a/logs/dreambooth/1700466613.9881477/events.out.tfevents.1700466613.r-crom87-db-training-test-space-vdp8ibyr-af963-qphx2.1.1 b/logs/dreambooth/1700466613.9881477/events.out.tfevents.1700466613.r-crom87-db-training-test-space-vdp8ibyr-af963-qphx2.1.1 new file mode 100644 index 0000000000000000000000000000000000000000..ee7da8f56d06b0aa39be2d7e1191ced354b73f37 --- /dev/null +++ b/logs/dreambooth/1700466613.9881477/events.out.tfevents.1700466613.r-crom87-db-training-test-space-vdp8ibyr-af963-qphx2.1.1 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:08bd41f67d7c4cdf0ce52e8b7a5fb3becd469730fab58e5568571bc1547668df +size 3309 diff --git a/logs/dreambooth/1700466613.9899673/hparams.yml b/logs/dreambooth/1700466613.9899673/hparams.yml new file mode 100644 index 0000000000000000000000000000000000000000..19a16e49de0bd1a0ec87b7792110cc41797d7fd2 --- /dev/null +++ b/logs/dreambooth/1700466613.9899673/hparams.yml @@ -0,0 +1,60 @@ +adam_beta1: 0.9 +adam_beta2: 0.999 +adam_epsilon: 1.0e-08 +adam_weight_decay: 0.01 +allow_tf32: false +center_crop: false +checkpointing_steps: 500 +checkpoints_total_limit: null +class_data_dir: class_george_clooney +class_labels_conditioning: null +class_prompt: georgeclooney +dataloader_num_workers: 0 +enable_xformers_memory_efficient_attention: false +gradient_accumulation_steps: 1 +gradient_checkpointing: false +hub_model_id: crom87/sd_base-db-selfies4-1e-06-priorp-100-gc +hub_token: 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 +instance_data_dir: instance_images +instance_prompt: TOKstyle georgeclooney +learning_rate: 2.0e-06 +local_rank: -1 +logging_dir: logs +lr_num_cycles: 1 +lr_power: 1.0 +lr_scheduler: constant +lr_warmup_steps: 0 +max_grad_norm: 1.0 +max_train_steps: 5000 +mixed_precision: fp16 +num_class_images: 200 +num_train_epochs: 25 +num_validation_images: 3 +offset_noise: false +output_dir: output_model +pre_compute_text_embeddings: false +pretrained_model_name_or_path: crom87/sd_base-db-selfies4-1e-06-priorp-100-gc +prior_generation_precision: null +prior_loss_weight: 1.0 +push_to_hub: true +report_to: tensorboard +resolution: 512 +resume_from_checkpoint: latest +revision: null +sample_batch_size: 4 +samples_per_plugin: images=100 +scale_lr: false +seed: 42 +set_grads_to_none: false +skip_save_text_encoder: false +snr_gamma: null +text_encoder_use_attention_mask: false +tokenizer_max_length: null +tokenizer_name: null +train_batch_size: 1 +train_text_encoder: true +use_8bit_adam: true +validation_prompt: fixed in script +validation_scheduler: DPMSolverMultistepScheduler +validation_steps: 300 +with_prior_preservation: true diff --git a/logs/dreambooth/events.out.tfevents.1700466613.r-crom87-db-training-test-space-vdp8ibyr-af963-qphx2.1.0 b/logs/dreambooth/events.out.tfevents.1700466613.r-crom87-db-training-test-space-vdp8ibyr-af963-qphx2.1.0 new file mode 100644 index 0000000000000000000000000000000000000000..2b167b1306ec934584996c80501206b797abdad6 --- /dev/null +++ b/logs/dreambooth/events.out.tfevents.1700466613.r-crom87-db-training-test-space-vdp8ibyr-af963-qphx2.1.0 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:d7a3b2b38beb6e71089bbc7210adb6cd9925a7fc630b1c837915de1cbd464ee7 +size 18578151 diff --git a/model_index.json b/model_index.json index 233485ea6edc7db248c176eb91ef45f519b1586e..4f4c729e5d854775ab2ac33ef4f80f9a85618af0 100644 --- a/model_index.json +++ b/model_index.json @@ -1,7 +1,7 @@ { "_class_name": "StableDiffusionPipeline", "_diffusers_version": "0.23.1", - "_name_or_path": "runwayml/stable-diffusion-v1-5", + "_name_or_path": "crom87/sd_base-db-selfies4-1e-06-priorp-100-gc", "feature_extractor": [ "transformers", "CLIPImageProcessor" diff --git a/safety_checker/config.json b/safety_checker/config.json index 73e23c653d04d0ae07d9798f4eacb79fa5f3b823..fbb39d767819ed27129c3f84f555365ce3904345 100644 --- a/safety_checker/config.json +++ b/safety_checker/config.json @@ -1,5 +1,5 @@ { - "_name_or_path": "/home/user/.cache/huggingface/hub/models--runwayml--stable-diffusion-v1-5/snapshots/1d0c4ebf6ff58a5caecab40fa1406526bca4b5b9/safety_checker", + "_name_or_path": "/home/user/.cache/huggingface/hub/models--crom87--sd_base-db-selfies4-1e-06-priorp-100-gc/snapshots/a26bb4f9c4df48643e698dc3bff55f14bf504bff/safety_checker", "architectures": [ "StableDiffusionSafetyChecker" ], diff --git a/text_encoder/config.json b/text_encoder/config.json index 1bd2ec5b8f36b41c778d23c96db888069188deb9..40471de7789e1ca424aefef71b8dc124b869f269 100644 --- a/text_encoder/config.json +++ b/text_encoder/config.json @@ -1,5 +1,5 @@ { - "_name_or_path": "runwayml/stable-diffusion-v1-5", + "_name_or_path": "crom87/sd_base-db-selfies4-1e-06-priorp-100-gc", "architectures": [ "CLIPTextModel" ], diff --git a/text_encoder/model.safetensors b/text_encoder/model.safetensors index d6b88770fd4fcb75d5406d76aece00e70c42fed4..2bf54dc15871a4f28f8955d7f2fa9f2cf4f3dd99 100644 --- a/text_encoder/model.safetensors +++ b/text_encoder/model.safetensors @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:616f6fd55074cdb89184c7fbdd678d7a4193b13ce540a303ef2740542217dd0f +oid sha256:69e318ad4f25b6db797200f3d60bb61e362908c57231e926ccba4228695e8903 size 492265168 diff --git a/tokenizer/special_tokens_map.json b/tokenizer/special_tokens_map.json index 2c2130b544c0c5a72d5d00da071ba130a9800fb2..b5018cd2c0d0f31e08ba78d639e05736101ca08a 100644 --- a/tokenizer/special_tokens_map.json +++ b/tokenizer/special_tokens_map.json @@ -13,7 +13,13 @@ "rstrip": false, "single_word": false }, - "pad_token": "<|endoftext|>", + "pad_token": { + "content": "<|endoftext|>", + "lstrip": false, + "normalized": true, + "rstrip": false, + "single_word": false + }, "unk_token": { "content": "<|endoftext|>", "lstrip": false, diff --git a/unet/config.json b/unet/config.json index adad02d1cdec217d04412031372cc60434aaedce..fe43629b8071d2d7c03ac116d928f403c7772c6b 100644 --- a/unet/config.json +++ b/unet/config.json @@ -1,7 +1,7 @@ { "_class_name": "UNet2DConditionModel", "_diffusers_version": "0.23.1", - "_name_or_path": "runwayml/stable-diffusion-v1-5", + "_name_or_path": "crom87/sd_base-db-selfies4-1e-06-priorp-100-gc", "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 4a75997af3df72cc68b8d2d29fc0d1bffde85d74..81baadd900af8a8fd2724dca9a1b6dd9f52a9c6c 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:37971510ad440d8e668b94ead1753accb18cac31724601d4e623ae1df50078f5 +oid sha256:66f8c34998e349efa32906c44dda0d5b783a31b6fb3610c489ac1eab771b8c0c size 3438167536 diff --git a/vae/config.json b/vae/config.json index 1f309bbf3f9f12b79a618722f9ae84764c46eaba..56c6b94bd0b73d6b449463473e2ae18fa56dba9f 100644 --- a/vae/config.json +++ b/vae/config.json @@ -1,7 +1,7 @@ { "_class_name": "AutoencoderKL", "_diffusers_version": "0.23.1", - "_name_or_path": "/home/user/.cache/huggingface/hub/models--runwayml--stable-diffusion-v1-5/snapshots/1d0c4ebf6ff58a5caecab40fa1406526bca4b5b9/vae", + "_name_or_path": "/home/user/.cache/huggingface/hub/models--crom87--sd_base-db-selfies4-1e-06-priorp-100-gc/snapshots/a26bb4f9c4df48643e698dc3bff55f14bf504bff/vae", "act_fn": "silu", "block_out_channels": [ 128,