diff --git a/README.md b/README.md
new file mode 100644
index 0000000000000000000000000000000000000000..2d607a31439116bdcacf54134601ca5ea5334d6f
--- /dev/null
+++ b/README.md
@@ -0,0 +1,22 @@
+---
+license: creativeml-openrail-m
+base_model: stabilityai/stable-diffusion-2-1-base
+instance_prompt: wangyanting
+tags:
+- stable-diffusion
+- stable-diffusion-diffusers
+- text-to-image
+- diffusers
+- lora
+inference: true
+---
+# LoRA DreamBooth - wyt
+
+These are LoRA adaption weights for [stabilityai/stable-diffusion-2-1-base](https://huggingface.co/stabilityai/stable-diffusion-2-1-base). The weights were trained on the instance prompt "wangyanting" using [DreamBooth](https://dreambooth.github.io/). You can find some example images in the following.
+
+Test prompt: wyt
+![image_0](test_images/image_0.png)
+![image_1](test_images/image_1.png)
+![image_2](test_images/image_2.png)
+![image_3](test_images/image_3.png)
+
diff --git a/checkpoint-1000/custom_checkpoint_0.pkl b/checkpoint-1000/custom_checkpoint_0.pkl
new file mode 100644
index 0000000000000000000000000000000000000000..8d5320096d22b5e1a823b1e844e76607fb9dbb01
--- /dev/null
+++ b/checkpoint-1000/custom_checkpoint_0.pkl
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1d3816a5ad466fba54ea4faaae896a6025fd21b716170104af1dc873c7db9023
+size 3418585
diff --git a/checkpoint-1000/optimizer.bin b/checkpoint-1000/optimizer.bin
new file mode 100644
index 0000000000000000000000000000000000000000..c13bae6c43beec7780f9d3abf9ca55ed8e494a66
--- /dev/null
+++ b/checkpoint-1000/optimizer.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0d0372186a2c30a9ce83640bd50d1a5d5dd232f49031c6a49de944074102489f
+size 3493125
diff --git a/checkpoint-1000/pytorch_model.bin b/checkpoint-1000/pytorch_model.bin
new file mode 100644
index 0000000000000000000000000000000000000000..56766e019ea41fc524c557ee8268306516b6ddd2
--- /dev/null
+++ b/checkpoint-1000/pytorch_model.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:15b6ed9d352a58c87bcb8a1e154f69bd9065a2ce56a2da9d9a14f5f5fd825401
+size 3417037
diff --git a/checkpoint-1000/random_states_0.pkl b/checkpoint-1000/random_states_0.pkl
new file mode 100644
index 0000000000000000000000000000000000000000..653a70b82c68e9ef90e3474ff6e65b8afa5bbf53
--- /dev/null
+++ b/checkpoint-1000/random_states_0.pkl
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:347deeb43811276595747ed9fc784b80c25893672ebc1a00cb54a456961738c0
+size 14599
diff --git a/checkpoint-1000/scaler.pt b/checkpoint-1000/scaler.pt
new file mode 100644
index 0000000000000000000000000000000000000000..efdbd3c795f6b0d4144e68355e99c220ccdedd09
--- /dev/null
+++ b/checkpoint-1000/scaler.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:68cff80b680ddf6e7abbef98b5f336b97f9b5963e2209307f639383870e8cc71
+size 557
diff --git a/checkpoint-1000/scheduler.bin b/checkpoint-1000/scheduler.bin
new file mode 100644
index 0000000000000000000000000000000000000000..9386744f4ecd3382f2baaea35ec573968fbf81cd
--- /dev/null
+++ b/checkpoint-1000/scheduler.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:963c3085720a3a3ca1d178559a0f2dc5d5b36fa395a4663f0194bfac4a754038
+size 563
diff --git a/checkpoint-1200/custom_checkpoint_0.pkl b/checkpoint-1200/custom_checkpoint_0.pkl
new file mode 100644
index 0000000000000000000000000000000000000000..3e211359690945f3b88e37a1fa7869980386aed7
--- /dev/null
+++ b/checkpoint-1200/custom_checkpoint_0.pkl
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d8b2dafaab70e50d74dbcce43b6575a51c0fa175977db0e360c1e5b5978351ed
+size 3418585
diff --git a/checkpoint-1200/optimizer.bin b/checkpoint-1200/optimizer.bin
new file mode 100644
index 0000000000000000000000000000000000000000..8cfc19babe8ff27569712e930c97121c52ddae1a
--- /dev/null
+++ b/checkpoint-1200/optimizer.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d8fd502941b51ca35db2c3981a1ff6932321a1aeb5ace156304f31a9e29474cb
+size 3493125
diff --git a/checkpoint-1200/pytorch_model.bin b/checkpoint-1200/pytorch_model.bin
new file mode 100644
index 0000000000000000000000000000000000000000..1e22668d05ba48f2eff5e8e52d2d3dc2332b57b7
--- /dev/null
+++ b/checkpoint-1200/pytorch_model.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8ec1846cec4f8439838e0599a34b0ae204b7f1e1a7431bd295c8c00bd87b79db
+size 3417037
diff --git a/checkpoint-1200/random_states_0.pkl b/checkpoint-1200/random_states_0.pkl
new file mode 100644
index 0000000000000000000000000000000000000000..2e6593498f63316f3f7f6d0daf49fe60bf67f980
--- /dev/null
+++ b/checkpoint-1200/random_states_0.pkl
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:72de7f263347e90620bf76a74643b3f792be751cc130575e6777430c7e565d9c
+size 14599
diff --git a/checkpoint-1200/scaler.pt b/checkpoint-1200/scaler.pt
new file mode 100644
index 0000000000000000000000000000000000000000..c73b6e7148d8ae7026711173634e0a11b1b94e2d
--- /dev/null
+++ b/checkpoint-1200/scaler.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:741cefeca9ef427f92406d2d10b81996655e2a9d50eb7aaa9614e6fdd1c9f529
+size 557
diff --git a/checkpoint-1200/scheduler.bin b/checkpoint-1200/scheduler.bin
new file mode 100644
index 0000000000000000000000000000000000000000..6f41b0cd958cdad128205ad924553c301aeff377
--- /dev/null
+++ b/checkpoint-1200/scheduler.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:efb395eb1965763fbdfe27e8a4533b780593fd2bd4a41491d17f107e94a35f26
+size 563
diff --git a/checkpoint-1400/custom_checkpoint_0.pkl b/checkpoint-1400/custom_checkpoint_0.pkl
new file mode 100644
index 0000000000000000000000000000000000000000..b1f4e36751ceaa11113f10856d26e69cc8838342
--- /dev/null
+++ b/checkpoint-1400/custom_checkpoint_0.pkl
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f786697dd8e58a430b2b54c7501c5f8b2b36e368cb289bc94c9109d1038015f2
+size 3418585
diff --git a/checkpoint-1400/optimizer.bin b/checkpoint-1400/optimizer.bin
new file mode 100644
index 0000000000000000000000000000000000000000..a23873bb6a1eedd53acd0c675a67f6c066218491
--- /dev/null
+++ b/checkpoint-1400/optimizer.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b0b87a7673c536e95f8cc02f57387edfcb10f51cb9e121718f70f954bbb9c8cc
+size 3493125
diff --git a/checkpoint-1400/pytorch_model.bin b/checkpoint-1400/pytorch_model.bin
new file mode 100644
index 0000000000000000000000000000000000000000..64d5cf0c801295f03fa9caba9f779d3a6e65e3f9
--- /dev/null
+++ b/checkpoint-1400/pytorch_model.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:66a8c125d896884fa8c0c48c04e9956132e2e7c7311f4ed21fbdade603ab47c4
+size 3417037
diff --git a/checkpoint-1400/random_states_0.pkl b/checkpoint-1400/random_states_0.pkl
new file mode 100644
index 0000000000000000000000000000000000000000..50afe8464d10eb96af31459e001dd505f57c3738
--- /dev/null
+++ b/checkpoint-1400/random_states_0.pkl
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ac0f487244f38cadca03885bba7f64f85e1b1082955bc13ec7c12d380acfa86f
+size 14599
diff --git a/checkpoint-1400/scaler.pt b/checkpoint-1400/scaler.pt
new file mode 100644
index 0000000000000000000000000000000000000000..a8cb12388904652e2007207f80582007b39a2051
--- /dev/null
+++ b/checkpoint-1400/scaler.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:16fdfc03b58220402968eacaac23fb5471cdb9061302380bd3c8d4d326c02ade
+size 557
diff --git a/checkpoint-1400/scheduler.bin b/checkpoint-1400/scheduler.bin
new file mode 100644
index 0000000000000000000000000000000000000000..e2788e4c509e9cd6e042498ed766593746f2f189
--- /dev/null
+++ b/checkpoint-1400/scheduler.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7a1ca140eeb22d398d44c278637bb16bcea285ccaf1b0e02b1a13c5d4df53161
+size 563
diff --git a/checkpoint-1600/custom_checkpoint_0.pkl b/checkpoint-1600/custom_checkpoint_0.pkl
new file mode 100644
index 0000000000000000000000000000000000000000..805a896931fcbaee09009e1fe58b426e71778e24
--- /dev/null
+++ b/checkpoint-1600/custom_checkpoint_0.pkl
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:eb1c8acf128318a5f01d739447b0556b557e34c8000226fb301bf8e0dba74ec2
+size 3418585
diff --git a/checkpoint-1600/optimizer.bin b/checkpoint-1600/optimizer.bin
new file mode 100644
index 0000000000000000000000000000000000000000..054929b5ae0972707827b1ceb720564e9fd64cb8
--- /dev/null
+++ b/checkpoint-1600/optimizer.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3ea72c9eeb91d840d1a1c0a7cd480d4338730ea4a0e05bf33d613e7e94a80730
+size 3493125
diff --git a/checkpoint-1600/pytorch_model.bin b/checkpoint-1600/pytorch_model.bin
new file mode 100644
index 0000000000000000000000000000000000000000..a1f1383c0dc29ed58b9c11c9e296250a2fd450c5
--- /dev/null
+++ b/checkpoint-1600/pytorch_model.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:27744e18e1a1593c791d8fa37e774966e988781306cd598b93466d30d5933915
+size 3417037
diff --git a/checkpoint-1600/random_states_0.pkl b/checkpoint-1600/random_states_0.pkl
new file mode 100644
index 0000000000000000000000000000000000000000..1783f428e5887c3ecc332b3b45d105f99451a2ff
--- /dev/null
+++ b/checkpoint-1600/random_states_0.pkl
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ca0a27d8521d255872db57f7009fc64b7ad709e3c10a4a75306a13f1c0d17fad
+size 14599
diff --git a/checkpoint-1600/scaler.pt b/checkpoint-1600/scaler.pt
new file mode 100644
index 0000000000000000000000000000000000000000..2f03c4ce6fd4620d95ae66b6787fdbcf34c6622a
--- /dev/null
+++ b/checkpoint-1600/scaler.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:cbbb3b0bb9c64d37bf898d0431b3ed1f0a6f5c9d8c2b563e0f884424fb8bd92a
+size 557
diff --git a/checkpoint-1600/scheduler.bin b/checkpoint-1600/scheduler.bin
new file mode 100644
index 0000000000000000000000000000000000000000..369e0c9a1c6c5ed4943f67d541116999ae68e09e
--- /dev/null
+++ b/checkpoint-1600/scheduler.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:144193f9b4afefdc81b5b9bbeb013f085ce761d3aa64df98f5910666001ebe7a
+size 563
diff --git a/checkpoint-1800/custom_checkpoint_0.pkl b/checkpoint-1800/custom_checkpoint_0.pkl
new file mode 100644
index 0000000000000000000000000000000000000000..8c3619dadf174f4c99ff2e00e371d61a9855a2fe
--- /dev/null
+++ b/checkpoint-1800/custom_checkpoint_0.pkl
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c17c588f8d2405d5c6010aeac65715f461040635e46e63ca0edb88c40b6f5464
+size 3418585
diff --git a/checkpoint-1800/optimizer.bin b/checkpoint-1800/optimizer.bin
new file mode 100644
index 0000000000000000000000000000000000000000..81b157449985ad9946e3adae7dfd951f199b6678
--- /dev/null
+++ b/checkpoint-1800/optimizer.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:32c08936d26ce8eab5a58025db1c1b350ebbda49d3cf8f7e3063341584a52bc8
+size 3493125
diff --git a/checkpoint-1800/pytorch_model.bin b/checkpoint-1800/pytorch_model.bin
new file mode 100644
index 0000000000000000000000000000000000000000..52b60124a56bb976d09ff03793cc98bfd15154eb
--- /dev/null
+++ b/checkpoint-1800/pytorch_model.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c42777913fda604c7fa8cdb08e35060583a5eb4261bd8299e5dc5f535b720a88
+size 3417037
diff --git a/checkpoint-1800/random_states_0.pkl b/checkpoint-1800/random_states_0.pkl
new file mode 100644
index 0000000000000000000000000000000000000000..10b9f6b229176ad04ecd17d7fc113bff240ed1f9
--- /dev/null
+++ b/checkpoint-1800/random_states_0.pkl
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b30ff5ef7d734fd6e4abf2d96062d12c00949e2c1e6f3dbcdbbffe29e17febec
+size 14599
diff --git a/checkpoint-1800/scaler.pt b/checkpoint-1800/scaler.pt
new file mode 100644
index 0000000000000000000000000000000000000000..a042e3b3d1804ba0c9a30ddfdd3014cb8aed4b24
--- /dev/null
+++ b/checkpoint-1800/scaler.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6f56a7c2178708093b1569b9704efb1c2791b9551cfff05b50263fc281598b68
+size 557
diff --git a/checkpoint-1800/scheduler.bin b/checkpoint-1800/scheduler.bin
new file mode 100644
index 0000000000000000000000000000000000000000..49d01880413042b61997654e7cfb0661966511c2
--- /dev/null
+++ b/checkpoint-1800/scheduler.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:32cd70a7cbd8536497bdbd5cbf497eaa0589cb08438d4bbce6614a2ba684214f
+size 563
diff --git a/checkpoint-200/custom_checkpoint_0.pkl b/checkpoint-200/custom_checkpoint_0.pkl
new file mode 100644
index 0000000000000000000000000000000000000000..1dd4a179bb676afa6dd716f18f89030440977b18
--- /dev/null
+++ b/checkpoint-200/custom_checkpoint_0.pkl
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:87d8e6a959e9c954180fa91138ece040402b74dab9b43e1b5fc2197e138ac4f2
+size 3418585
diff --git a/checkpoint-200/optimizer.bin b/checkpoint-200/optimizer.bin
new file mode 100644
index 0000000000000000000000000000000000000000..a155442f1f3ecf8fa7aa1fdd6d9e446dbdb29190
--- /dev/null
+++ b/checkpoint-200/optimizer.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a4db38efbb61b8d7889b17e785aa9ab9513cb8ef5a93a7e22f3c1e728626c130
+size 3492869
diff --git a/checkpoint-200/pytorch_model.bin b/checkpoint-200/pytorch_model.bin
new file mode 100644
index 0000000000000000000000000000000000000000..d8020f9cc151d34293dbc754de4f0f162b7962a8
--- /dev/null
+++ b/checkpoint-200/pytorch_model.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b55ef53b05bc03d918a326a0943bb137d9e97ae4ffdb3199b27ddf242ae1ffec
+size 3417037
diff --git a/checkpoint-200/random_states_0.pkl b/checkpoint-200/random_states_0.pkl
new file mode 100644
index 0000000000000000000000000000000000000000..4df093e6a8dbee0ece4fb31eabbfb529ffc1ef39
--- /dev/null
+++ b/checkpoint-200/random_states_0.pkl
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:364fe549e670efd13316c8b6b038f2b9d5dde8486a932767b455cba6a58898b2
+size 14599
diff --git a/checkpoint-200/scaler.pt b/checkpoint-200/scaler.pt
new file mode 100644
index 0000000000000000000000000000000000000000..87710d0ddc627c070366fcb3112b07dc60d97295
--- /dev/null
+++ b/checkpoint-200/scaler.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4fccf0f9be1bb8f24861e4393745b3e09cc2687125a69e3757955fb0f0925ea5
+size 557
diff --git a/checkpoint-200/scheduler.bin b/checkpoint-200/scheduler.bin
new file mode 100644
index 0000000000000000000000000000000000000000..576b209b652e9ae5872edbd1a3aac9a3d11f88b4
--- /dev/null
+++ b/checkpoint-200/scheduler.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d2565f1f317aadcb95557955732a7a05efb023cacf90620d89fcb0c7ac0f8e04
+size 563
diff --git a/checkpoint-2000/custom_checkpoint_0.pkl b/checkpoint-2000/custom_checkpoint_0.pkl
new file mode 100644
index 0000000000000000000000000000000000000000..8c3619dadf174f4c99ff2e00e371d61a9855a2fe
--- /dev/null
+++ b/checkpoint-2000/custom_checkpoint_0.pkl
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c17c588f8d2405d5c6010aeac65715f461040635e46e63ca0edb88c40b6f5464
+size 3418585
diff --git a/checkpoint-2000/optimizer.bin b/checkpoint-2000/optimizer.bin
new file mode 100644
index 0000000000000000000000000000000000000000..81b157449985ad9946e3adae7dfd951f199b6678
--- /dev/null
+++ b/checkpoint-2000/optimizer.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:32c08936d26ce8eab5a58025db1c1b350ebbda49d3cf8f7e3063341584a52bc8
+size 3493125
diff --git a/checkpoint-2000/pytorch_model.bin b/checkpoint-2000/pytorch_model.bin
new file mode 100644
index 0000000000000000000000000000000000000000..52b60124a56bb976d09ff03793cc98bfd15154eb
--- /dev/null
+++ b/checkpoint-2000/pytorch_model.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c42777913fda604c7fa8cdb08e35060583a5eb4261bd8299e5dc5f535b720a88
+size 3417037
diff --git a/checkpoint-2000/random_states_0.pkl b/checkpoint-2000/random_states_0.pkl
new file mode 100644
index 0000000000000000000000000000000000000000..0659e1dba3e46fe573e15b6ff79355365140fc99
--- /dev/null
+++ b/checkpoint-2000/random_states_0.pkl
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6b660af934d9940faf44ee7e99eb8039dc2bbade341c0afefd9ea0f0af2e2572
+size 14599
diff --git a/checkpoint-2000/scaler.pt b/checkpoint-2000/scaler.pt
new file mode 100644
index 0000000000000000000000000000000000000000..115263c9235918fc2a72413ed7e00c969b8d3527
--- /dev/null
+++ b/checkpoint-2000/scaler.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7fa181fa360d46feed4180ea17c8b6a4a879a9b4231c2e91aff2be20be9076cc
+size 557
diff --git a/checkpoint-2000/scheduler.bin b/checkpoint-2000/scheduler.bin
new file mode 100644
index 0000000000000000000000000000000000000000..c6460047f3f26edc6ff163b41053d86034b9cfd1
--- /dev/null
+++ b/checkpoint-2000/scheduler.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:456e0f82e1f69c1c7c3f294d8d9ee5a27f11a3af5067518517fb2354b1dae333
+size 563
diff --git a/checkpoint-400/custom_checkpoint_0.pkl b/checkpoint-400/custom_checkpoint_0.pkl
new file mode 100644
index 0000000000000000000000000000000000000000..1c9c4f6be1f0a1967d8badc55d19dfb2a5f2fcb5
--- /dev/null
+++ b/checkpoint-400/custom_checkpoint_0.pkl
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d3d9ddf78815bb6db6a52c43cd8fa6ca281cef0d62b84daeed03bb1934655fe0
+size 3418585
diff --git a/checkpoint-400/optimizer.bin b/checkpoint-400/optimizer.bin
new file mode 100644
index 0000000000000000000000000000000000000000..a4348556d938d57b2cf025fbbffb1c85a6378fa8
--- /dev/null
+++ b/checkpoint-400/optimizer.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:cfb96e573dcda4f137e97d5c924d6b5b5e936f8c551415d2aa31559ca4c64d78
+size 3493125
diff --git a/checkpoint-400/pytorch_model.bin b/checkpoint-400/pytorch_model.bin
new file mode 100644
index 0000000000000000000000000000000000000000..994f4e96331c5b302d9631cba2149fc2343bf212
--- /dev/null
+++ b/checkpoint-400/pytorch_model.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7477fd092120097cff83a010ba5221867d06ec5f39f9d589bb582bb5eb13c404
+size 3417037
diff --git a/checkpoint-400/random_states_0.pkl b/checkpoint-400/random_states_0.pkl
new file mode 100644
index 0000000000000000000000000000000000000000..24af965e9b7179784c4263ffee249b13e9fa66a6
--- /dev/null
+++ b/checkpoint-400/random_states_0.pkl
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:99635dec3e8138cd0049b49397201455cd6b02a3a768c965c3d525c343a2d4af
+size 14599
diff --git a/checkpoint-400/scaler.pt b/checkpoint-400/scaler.pt
new file mode 100644
index 0000000000000000000000000000000000000000..3ee84e1b463b64410ae6f3d5c680aef0d7e2b14f
--- /dev/null
+++ b/checkpoint-400/scaler.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:cc40a4be6a52cee4d7658df4041e660ffa02b0d8b5bd143bb8bb397f7b71b1a5
+size 557
diff --git a/checkpoint-400/scheduler.bin b/checkpoint-400/scheduler.bin
new file mode 100644
index 0000000000000000000000000000000000000000..a16de5e0d963432c63202f0a7f8393a1d5987503
--- /dev/null
+++ b/checkpoint-400/scheduler.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5e33b45756ec30d1126f6571b94c7435f6d9cb7b6d5e5e650c81523775e7216f
+size 563
diff --git a/checkpoint-600/custom_checkpoint_0.pkl b/checkpoint-600/custom_checkpoint_0.pkl
new file mode 100644
index 0000000000000000000000000000000000000000..926bff41197edc41e5884988a950315b3fc52b8a
--- /dev/null
+++ b/checkpoint-600/custom_checkpoint_0.pkl
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:49a5e65be35d9070086aa7891c3f05bd1a657b1e45f9fc10413799c4f013e7df
+size 3418585
diff --git a/checkpoint-600/optimizer.bin b/checkpoint-600/optimizer.bin
new file mode 100644
index 0000000000000000000000000000000000000000..f7c0ca0e5dedc8de7c55b72985427ba0f32d87bc
--- /dev/null
+++ b/checkpoint-600/optimizer.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:872716b5a58324c46f77027eeff5cfc4cb2c2030e0ae614ec2a5024c94e2a248
+size 3493125
diff --git a/checkpoint-600/pytorch_model.bin b/checkpoint-600/pytorch_model.bin
new file mode 100644
index 0000000000000000000000000000000000000000..143f61ad3eb7b69e9e08b091ec26e31a1e3c28ff
--- /dev/null
+++ b/checkpoint-600/pytorch_model.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:faec7cff618d8003ac93910415ea3dc2e3578490d45745c4c6e8b987c0bd5716
+size 3417037
diff --git a/checkpoint-600/random_states_0.pkl b/checkpoint-600/random_states_0.pkl
new file mode 100644
index 0000000000000000000000000000000000000000..f03d742e0f53baa5d97a4548841751fd0dd00c7f
--- /dev/null
+++ b/checkpoint-600/random_states_0.pkl
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:82bab10a49760c00d4c65032d21faae1a089f63ecbeca4f6e1112477adf447a3
+size 14599
diff --git a/checkpoint-600/scaler.pt b/checkpoint-600/scaler.pt
new file mode 100644
index 0000000000000000000000000000000000000000..842791b612283ceb8e68b64ed8e40e81c5a97bce
--- /dev/null
+++ b/checkpoint-600/scaler.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5dc9eacfeb00bd0bfeb98934a2309be01be65b288e0d747bbfc423b32679169f
+size 557
diff --git a/checkpoint-600/scheduler.bin b/checkpoint-600/scheduler.bin
new file mode 100644
index 0000000000000000000000000000000000000000..6dff0a6b30941e160b51c76827a3d5909a696333
--- /dev/null
+++ b/checkpoint-600/scheduler.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:864c924c403e116aaa2ce6d169c54f498fb0609bfb8d79139ec317756b219c42
+size 563
diff --git a/checkpoint-800/custom_checkpoint_0.pkl b/checkpoint-800/custom_checkpoint_0.pkl
new file mode 100644
index 0000000000000000000000000000000000000000..341ca8b3df6eee13f11202a7da12afa78a696e3c
--- /dev/null
+++ b/checkpoint-800/custom_checkpoint_0.pkl
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ece3ef0b72ba10bacd0c02674b3fe26c1f8f7c959a4805593adbb3d86736ee87
+size 3418585
diff --git a/checkpoint-800/optimizer.bin b/checkpoint-800/optimizer.bin
new file mode 100644
index 0000000000000000000000000000000000000000..2ed33771def10d0842e8d97f825fafdee38e6c6f
--- /dev/null
+++ b/checkpoint-800/optimizer.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:580511e274848c0149af0b81aea0864f753ac31d8936b2e18e020564505948c5
+size 3493125
diff --git a/checkpoint-800/pytorch_model.bin b/checkpoint-800/pytorch_model.bin
new file mode 100644
index 0000000000000000000000000000000000000000..85c3c63e59227093e9d1a10dce4f73659d40a9ad
--- /dev/null
+++ b/checkpoint-800/pytorch_model.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a907a57cd667529e922aac7c48ad0dc8ee263f904a10b6e439c1310ac1dbf67d
+size 3417037
diff --git a/checkpoint-800/random_states_0.pkl b/checkpoint-800/random_states_0.pkl
new file mode 100644
index 0000000000000000000000000000000000000000..a2b775e9ecc1dfabbace2378a1899bb5a85fcdbc
--- /dev/null
+++ b/checkpoint-800/random_states_0.pkl
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e43c5da7ab78ae802ba2983ed7dad57084172ca244389319eeab7b3582c2367b
+size 14599
diff --git a/checkpoint-800/scaler.pt b/checkpoint-800/scaler.pt
new file mode 100644
index 0000000000000000000000000000000000000000..5e01dd7b5d3a8968bb4c73a805f08f0f65c9b57f
--- /dev/null
+++ b/checkpoint-800/scaler.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:27ec07a12731ae6f9765d05fe7c8495505f1d0f90b4cc6255a0853fec3970808
+size 557
diff --git a/checkpoint-800/scheduler.bin b/checkpoint-800/scheduler.bin
new file mode 100644
index 0000000000000000000000000000000000000000..3c4b34af859664768bf138a270c17169a950c506
--- /dev/null
+++ b/checkpoint-800/scheduler.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6f100ed37bb5af168a49362839776be9a5d1ff011bec2887fb25ef66be063de9
+size 563
diff --git a/logs/dreambooth-lora/1678872369.2098682/events.out.tfevents.1678872369.s-antonellaavad-lora-dreambooth-training-ui-33f5a-7bf69c5d4llbm.918.1 b/logs/dreambooth-lora/1678872369.2098682/events.out.tfevents.1678872369.s-antonellaavad-lora-dreambooth-training-ui-33f5a-7bf69c5d4llbm.918.1
new file mode 100644
index 0000000000000000000000000000000000000000..44dbf61f7e20d1efda44d9fa08500a34f5c8ed3f
--- /dev/null
+++ b/logs/dreambooth-lora/1678872369.2098682/events.out.tfevents.1678872369.s-antonellaavad-lora-dreambooth-training-ui-33f5a-7bf69c5d4llbm.918.1
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a1281e3f5209e126650478c55cfee359682d5fe4e971623759f368cdb915b2ac
+size 2415
diff --git a/logs/dreambooth-lora/1678872369.2121255/hparams.yml b/logs/dreambooth-lora/1678872369.2121255/hparams.yml
new file mode 100644
index 0000000000000000000000000000000000000000..daa31e6a4fe16d41bc64abc4c31e5440aac76a0d
--- /dev/null
+++ b/logs/dreambooth-lora/1678872369.2121255/hparams.yml
@@ -0,0 +1,48 @@
+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: 200
+class_data_dir: null
+class_prompt: null
+dataloader_num_workers: 0
+enable_xformers_memory_efficient_attention: false
+gradient_accumulation_steps: 1
+gradient_checkpointing: false
+hub_model_id: null
+hub_token: null
+instance_data_dir: /home/user/app/training_data/wyt
+instance_prompt: wangyanting
+learning_rate: 0.0001
+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: 2000
+mixed_precision: fp16
+num_class_images: 100
+num_train_epochs: 58
+num_validation_images: 4
+output_dir: /home/user/app/experiments/wyt
+pretrained_model_name_or_path: stabilityai/stable-diffusion-2-1-base
+prior_generation_precision: null
+prior_loss_weight: 1.0
+push_to_hub: false
+report_to: tensorboard
+resolution: 512
+resume_from_checkpoint: null
+revision: null
+sample_batch_size: 4
+scale_lr: false
+seed: 0
+tokenizer_name: null
+train_batch_size: 1
+use_8bit_adam: true
+validation_epochs: 100
+validation_prompt: wyt
+with_prior_preservation: false
diff --git a/logs/dreambooth-lora/events.out.tfevents.1678872369.s-antonellaavad-lora-dreambooth-training-ui-33f5a-7bf69c5d4llbm.918.0 b/logs/dreambooth-lora/events.out.tfevents.1678872369.s-antonellaavad-lora-dreambooth-training-ui-33f5a-7bf69c5d4llbm.918.0
new file mode 100644
index 0000000000000000000000000000000000000000..81442303194f7477bd0fa56101e8574a7f3417ba
--- /dev/null
+++ b/logs/dreambooth-lora/events.out.tfevents.1678872369.s-antonellaavad-lora-dreambooth-training-ui-33f5a-7bf69c5d4llbm.918.0
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:77bbd700404bad5433b017a0897c7d186bd2ea50fdc84702b0f80689235085f1
+size 2049972
diff --git a/pytorch_lora_weights.bin b/pytorch_lora_weights.bin
new file mode 100644
index 0000000000000000000000000000000000000000..b1dd0ba34e80bd83798f2fb2bfa361dc574e5501
--- /dev/null
+++ b/pytorch_lora_weights.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:65751e753795735d35a470d774c0c9c62fc854a249dafd0f1533d3c1c4f3ca26
+size 3418843
diff --git a/test_images/image_0.png b/test_images/image_0.png
new file mode 100644
index 0000000000000000000000000000000000000000..5220d0f9dbfa747cdb356761a73beb1b1f1c96bb
Binary files /dev/null and b/test_images/image_0.png differ
diff --git a/test_images/image_1.png b/test_images/image_1.png
new file mode 100644
index 0000000000000000000000000000000000000000..5220d0f9dbfa747cdb356761a73beb1b1f1c96bb
Binary files /dev/null and b/test_images/image_1.png differ
diff --git a/test_images/image_2.png b/test_images/image_2.png
new file mode 100644
index 0000000000000000000000000000000000000000..5220d0f9dbfa747cdb356761a73beb1b1f1c96bb
Binary files /dev/null and b/test_images/image_2.png differ
diff --git a/test_images/image_3.png b/test_images/image_3.png
new file mode 100644
index 0000000000000000000000000000000000000000..5220d0f9dbfa747cdb356761a73beb1b1f1c96bb
Binary files /dev/null and b/test_images/image_3.png differ
diff --git a/train.sh b/train.sh
new file mode 100644
index 0000000000000000000000000000000000000000..c974b16bcc3ccd5eb2843d9933e8eb2955575f6b
--- /dev/null
+++ b/train.sh
@@ -0,0 +1 @@
+accelerate launch train_dreambooth_lora.py --pretrained_model_name_or_path=stabilityai/stable-diffusion-2-1-base --instance_data_dir=/home/user/app/training_data/wyt --output_dir=/home/user/app/experiments/wyt --instance_prompt="wangyanting" --resolution=512 --train_batch_size=1 --gradient_accumulation_steps=1 --learning_rate=0.0001 --lr_scheduler=constant --lr_warmup_steps=0 --max_train_steps=2000 --checkpointing_steps=200 --validation_prompt="wyt" --validation_epochs=100 --seed=0 --mixed_precision fp16 --use_8bit_adam
\ No newline at end of file