diff --git a/README.md b/README.md
new file mode 100644
index 0000000000000000000000000000000000000000..80de1ec0f8ba1d87380309b7a0f86acdb02ce46f
--- /dev/null
+++ b/README.md
@@ -0,0 +1,22 @@
+---
+license: creativeml-openrail-m
+base_model: stabilityai/stable-diffusion-2-1-base
+instance_prompt: black hat, hat ornament, flower hat
+tags:
+- stable-diffusion
+- stable-diffusion-diffusers
+- text-to-image
+- diffusers
+- lora
+inference: true
+---
+# LoRA DreamBooth - hu-tao-hat
+
+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 "black hat, hat ornament, flower hat" using [DreamBooth](https://dreambooth.github.io/). You can find some example images in the following.
+
+Test prompt: black hat, hat ornament, flower hat ornament
+![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-100/custom_checkpoint_0.pkl b/checkpoint-100/custom_checkpoint_0.pkl
new file mode 100644
index 0000000000000000000000000000000000000000..3428b7e92e2bea8819a5e51aab2ac4ab967698cb
--- /dev/null
+++ b/checkpoint-100/custom_checkpoint_0.pkl
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a9bd14e60f0383879b5e7410a183661b6449a19a2474ec2f59ca7dc845d172a3
+size 3418585
diff --git a/checkpoint-100/optimizer.bin b/checkpoint-100/optimizer.bin
new file mode 100644
index 0000000000000000000000000000000000000000..5d4c7380747270b47ce404130e2ba552f083bdce
--- /dev/null
+++ b/checkpoint-100/optimizer.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:502e7d529c2004fc7eb471b007a3e9f008099a1b2c1563d896ddafa1e5e2e89a
+size 3492869
diff --git a/checkpoint-100/pytorch_model.bin b/checkpoint-100/pytorch_model.bin
new file mode 100644
index 0000000000000000000000000000000000000000..31d20c078b1c2d920f7016c25cf5450ce344f6de
--- /dev/null
+++ b/checkpoint-100/pytorch_model.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4691799601a183552dddc88211a5da4ae8aac41680e7d7cb801626587cd957d8
+size 3417037
diff --git a/checkpoint-100/random_states_0.pkl b/checkpoint-100/random_states_0.pkl
new file mode 100644
index 0000000000000000000000000000000000000000..181ff9419140820085612313bca24eb79c51e0fe
--- /dev/null
+++ b/checkpoint-100/random_states_0.pkl
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ae2de888894feba30b5b0fb1c3588c2b09037a5bc8f7401b71ec5231b2af2322
+size 14599
diff --git a/checkpoint-100/scaler.pt b/checkpoint-100/scaler.pt
new file mode 100644
index 0000000000000000000000000000000000000000..1e00f588d3f0176a99d362447a49f57ff6e1b1ad
--- /dev/null
+++ b/checkpoint-100/scaler.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:cfa44e8523f62833816d29aa6c576eaa7783e3bbdb3e132e248b1d8aaee6132b
+size 557
diff --git a/checkpoint-100/scheduler.bin b/checkpoint-100/scheduler.bin
new file mode 100644
index 0000000000000000000000000000000000000000..c1705b3250b4c68673300431fac9562424a3f1ca
--- /dev/null
+++ b/checkpoint-100/scheduler.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:906bdd391189fd16d3160fd1e198eb53687dae580f930cf3f64365609fd5f606
+size 563
diff --git a/checkpoint-1000/custom_checkpoint_0.pkl b/checkpoint-1000/custom_checkpoint_0.pkl
new file mode 100644
index 0000000000000000000000000000000000000000..a113a4d09280ab312696e5e53b357178ae82ee94
--- /dev/null
+++ b/checkpoint-1000/custom_checkpoint_0.pkl
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:72f147f992947c9ea4b6fa31fb816a1a22cf5c09e485374746edebcb304d333c
+size 3418585
diff --git a/checkpoint-1000/optimizer.bin b/checkpoint-1000/optimizer.bin
new file mode 100644
index 0000000000000000000000000000000000000000..94c4bd8504c1fedd4e6a7cb41c5fff881fc9cee8
--- /dev/null
+++ b/checkpoint-1000/optimizer.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5cf9715c734e3c6125a9d177916fb9f32aee4e8612c3dbe49339a71b8844d473
+size 3493125
diff --git a/checkpoint-1000/pytorch_model.bin b/checkpoint-1000/pytorch_model.bin
new file mode 100644
index 0000000000000000000000000000000000000000..52138955cb3c3d707c1395a9376cf95b81bba187
--- /dev/null
+++ b/checkpoint-1000/pytorch_model.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:34904f940b2f41781a9764f42c78a7f68ec83908b9c0b44dc021627aa4cdc1b9
+size 3417037
diff --git a/checkpoint-1000/random_states_0.pkl b/checkpoint-1000/random_states_0.pkl
new file mode 100644
index 0000000000000000000000000000000000000000..7e99a909dd615d06a744dd3e7566436ccdc35e69
--- /dev/null
+++ b/checkpoint-1000/random_states_0.pkl
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:22607e60f87bbd596c9d7b18bd81272245b97661c7821b89dfadb82622f0587f
+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-200/custom_checkpoint_0.pkl b/checkpoint-200/custom_checkpoint_0.pkl
new file mode 100644
index 0000000000000000000000000000000000000000..589f6105421a7b310fd9867c7f305f633e2e7c9e
--- /dev/null
+++ b/checkpoint-200/custom_checkpoint_0.pkl
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b40d56729af30b1c46c1a3443c9a51f09bf78fe294f6d6e164ed55533f0d73c4
+size 3418585
diff --git a/checkpoint-200/optimizer.bin b/checkpoint-200/optimizer.bin
new file mode 100644
index 0000000000000000000000000000000000000000..00c83bfc7d86376b5684bbb7df7d5d51c4f47558
--- /dev/null
+++ b/checkpoint-200/optimizer.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:df8073abe8d71ed08ed15c291652c1e75b8d1ac213b6eeb056f5a2e2dd1beab2
+size 3492869
diff --git a/checkpoint-200/pytorch_model.bin b/checkpoint-200/pytorch_model.bin
new file mode 100644
index 0000000000000000000000000000000000000000..9115badc5bdc30bacf9a96501ede6b46128fca25
--- /dev/null
+++ b/checkpoint-200/pytorch_model.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7d2b93d5d8c6496ec002ad067132ad33fde764dcc5adc16fce33c9a4fc9ab3f3
+size 3417037
diff --git a/checkpoint-200/random_states_0.pkl b/checkpoint-200/random_states_0.pkl
new file mode 100644
index 0000000000000000000000000000000000000000..fb410c8dfaac27858b14345c0ee05b6850135fbe
--- /dev/null
+++ b/checkpoint-200/random_states_0.pkl
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:80eb60edf0a6889fb1499e6de5dd50c5f0f8e5c94f6706ee9573a9de4318778c
+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-300/custom_checkpoint_0.pkl b/checkpoint-300/custom_checkpoint_0.pkl
new file mode 100644
index 0000000000000000000000000000000000000000..b04d17b6da81bf9875c1f42874c19a14ebe8dc28
--- /dev/null
+++ b/checkpoint-300/custom_checkpoint_0.pkl
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:61822fe1dd4fecbb9744262a42ea8fa74172bb56a32249fcfdc55e27c0a9215f
+size 3418585
diff --git a/checkpoint-300/optimizer.bin b/checkpoint-300/optimizer.bin
new file mode 100644
index 0000000000000000000000000000000000000000..665741b72c4c549fc84735b948bf3ad5b7d6d654
--- /dev/null
+++ b/checkpoint-300/optimizer.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:00461fc69fcb295972db019b957b0b71e0fd08a1f756bfa5efa189044ca0221b
+size 3493125
diff --git a/checkpoint-300/pytorch_model.bin b/checkpoint-300/pytorch_model.bin
new file mode 100644
index 0000000000000000000000000000000000000000..052b4959c00e8a3a6c578e850a50d9580cc3e424
--- /dev/null
+++ b/checkpoint-300/pytorch_model.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4afa29cbab2de6d55ea192b83aaa0466247d6658ece238722c3956f7a94634bb
+size 3417037
diff --git a/checkpoint-300/random_states_0.pkl b/checkpoint-300/random_states_0.pkl
new file mode 100644
index 0000000000000000000000000000000000000000..41f75f116a4829820ea50e2c7bfb7e0b2cbc895a
--- /dev/null
+++ b/checkpoint-300/random_states_0.pkl
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:58bbcc070d2e0c0cd8b7fa9b40f300ca82e213688ebcd0c6eab78775b5e642d0
+size 14599
diff --git a/checkpoint-300/scaler.pt b/checkpoint-300/scaler.pt
new file mode 100644
index 0000000000000000000000000000000000000000..96d1b8ea2cbad321ffe9c10840a99bb1bcef18f5
--- /dev/null
+++ b/checkpoint-300/scaler.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:efafd90182e3d39d1b7c4a686f86e5913f5abc094dc3e2f827a6d479c6cef247
+size 557
diff --git a/checkpoint-300/scheduler.bin b/checkpoint-300/scheduler.bin
new file mode 100644
index 0000000000000000000000000000000000000000..5774569f77af24c12b1bdb79b70039de17478231
--- /dev/null
+++ b/checkpoint-300/scheduler.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a84fb1fc02dc9adb8c44beeab2c46fbc41e10a07e99dad3fcd85350ca2bda9d5
+size 563
diff --git a/checkpoint-400/custom_checkpoint_0.pkl b/checkpoint-400/custom_checkpoint_0.pkl
new file mode 100644
index 0000000000000000000000000000000000000000..92805985c363eca6fef17e1f890d3a4c428cbdda
--- /dev/null
+++ b/checkpoint-400/custom_checkpoint_0.pkl
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b1459494abdda7b1557134d7edadab93e424306f0da5014b343ed6e17469ad93
+size 3418585
diff --git a/checkpoint-400/optimizer.bin b/checkpoint-400/optimizer.bin
new file mode 100644
index 0000000000000000000000000000000000000000..fb8a5078248dcede2ca0665e2101b70ac5b909d3
--- /dev/null
+++ b/checkpoint-400/optimizer.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2bdce006dd83e1424c5ceb6c0ad92ebc1d8b15538a561e5e3286ba5c029d8d11
+size 3493125
diff --git a/checkpoint-400/pytorch_model.bin b/checkpoint-400/pytorch_model.bin
new file mode 100644
index 0000000000000000000000000000000000000000..3acbc194b27cf57820d2bdf12f0dbfad972181d9
--- /dev/null
+++ b/checkpoint-400/pytorch_model.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e36f9fe78b4478b89fa4452b81df916890318111b09c84c3954860492b94d9bc
+size 3417037
diff --git a/checkpoint-400/random_states_0.pkl b/checkpoint-400/random_states_0.pkl
new file mode 100644
index 0000000000000000000000000000000000000000..d3f160ccf4892b9143936c6de3c030ce42990b04
--- /dev/null
+++ b/checkpoint-400/random_states_0.pkl
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:636be054d39efa14a0c4c8ed7b24ca354c1419570060f077ff27dc5c8baca54f
+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-500/custom_checkpoint_0.pkl b/checkpoint-500/custom_checkpoint_0.pkl
new file mode 100644
index 0000000000000000000000000000000000000000..ef4f775ec3e294275c9cc6d2d12855ecb0c4dbd1
--- /dev/null
+++ b/checkpoint-500/custom_checkpoint_0.pkl
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9e4da7696d37a38d0c4d116c3f6e9e700e0a7a10b9e50393cff72e9d09b60d64
+size 3418585
diff --git a/checkpoint-500/optimizer.bin b/checkpoint-500/optimizer.bin
new file mode 100644
index 0000000000000000000000000000000000000000..984df912ff740828db2e2e43ccab4b932ef56a69
--- /dev/null
+++ b/checkpoint-500/optimizer.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:558219fc94b152bf2b61d72b9dc86af6b1b21fae461d683e7d3a31c93ca2bc7c
+size 3493125
diff --git a/checkpoint-500/pytorch_model.bin b/checkpoint-500/pytorch_model.bin
new file mode 100644
index 0000000000000000000000000000000000000000..f47a0a3fefa8aa0d788690a8a4b0164612c7a39b
--- /dev/null
+++ b/checkpoint-500/pytorch_model.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2a04d7ecec17d2b18d6c4259f320fd672b8420b197806c22ec34afe1c588068e
+size 3417037
diff --git a/checkpoint-500/random_states_0.pkl b/checkpoint-500/random_states_0.pkl
new file mode 100644
index 0000000000000000000000000000000000000000..bb81edffe8c1b05e5a1871e9e96e7c4b880fa3c0
--- /dev/null
+++ b/checkpoint-500/random_states_0.pkl
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9c92e2f56c64bb9646e38c7a05a270b63255873783d6c51e3502e944a447a273
+size 14599
diff --git a/checkpoint-500/scaler.pt b/checkpoint-500/scaler.pt
new file mode 100644
index 0000000000000000000000000000000000000000..b5f2e5ee77c2cef5eff816170558ce9209855685
--- /dev/null
+++ b/checkpoint-500/scaler.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a3f196a54202bb4ba1220e8c59f42f9cda0702d68ea83147d814c2fb2f36b8f2
+size 557
diff --git a/checkpoint-500/scheduler.bin b/checkpoint-500/scheduler.bin
new file mode 100644
index 0000000000000000000000000000000000000000..8c94c928cf7a00c3c46efdbe7bb69aaa33a599d3
--- /dev/null
+++ b/checkpoint-500/scheduler.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:62783ff6f0ae2c2ccc177b097e430c03940c5e70c38a191df273c5cf7ab1227c
+size 563
diff --git a/checkpoint-600/custom_checkpoint_0.pkl b/checkpoint-600/custom_checkpoint_0.pkl
new file mode 100644
index 0000000000000000000000000000000000000000..50943cffe5a5770d9a6fe8ec7be8ac54819fb0b8
--- /dev/null
+++ b/checkpoint-600/custom_checkpoint_0.pkl
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b9359f25bf3c95bdc43527dc4985e066e0ceab0ad7b67e932b568e78ec1421ae
+size 3418585
diff --git a/checkpoint-600/optimizer.bin b/checkpoint-600/optimizer.bin
new file mode 100644
index 0000000000000000000000000000000000000000..b886423cc573f7f1ee36188db98a52cb61b38abe
--- /dev/null
+++ b/checkpoint-600/optimizer.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:49fd604b381e3e36fc800a043d225d0aa66ae7bda0e4fabd996a33241fe99abf
+size 3493125
diff --git a/checkpoint-600/pytorch_model.bin b/checkpoint-600/pytorch_model.bin
new file mode 100644
index 0000000000000000000000000000000000000000..662698e797686dbd4b5f4d8aba011b5bf8bdb8fc
--- /dev/null
+++ b/checkpoint-600/pytorch_model.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7ddaf1f2c31cbd25880b6bcefc930c72949d87800a92de14ff4507c1108d3140
+size 3417037
diff --git a/checkpoint-600/random_states_0.pkl b/checkpoint-600/random_states_0.pkl
new file mode 100644
index 0000000000000000000000000000000000000000..260a2ab9785dc58182238f7a646f6d30553b54be
--- /dev/null
+++ b/checkpoint-600/random_states_0.pkl
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ad3b863332ad6a1859019f1eba1eff43cd385ad2e6a9fbc63168da35f39806a1
+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-700/custom_checkpoint_0.pkl b/checkpoint-700/custom_checkpoint_0.pkl
new file mode 100644
index 0000000000000000000000000000000000000000..cfc380b0a31445a09e1dc13968744f08d3e91e7a
--- /dev/null
+++ b/checkpoint-700/custom_checkpoint_0.pkl
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8bd36dedb365ba470babfe5d9fd807a4f00df98bd32745598e9ff1b9e5ae0b05
+size 3418585
diff --git a/checkpoint-700/optimizer.bin b/checkpoint-700/optimizer.bin
new file mode 100644
index 0000000000000000000000000000000000000000..10a890cd23e29371d93e5af464cd70d67326c529
--- /dev/null
+++ b/checkpoint-700/optimizer.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9fa05f4820f939c8a61dc17ab5ac98bbdebdd18608990a687bb82a39d3c8c0d7
+size 3493125
diff --git a/checkpoint-700/pytorch_model.bin b/checkpoint-700/pytorch_model.bin
new file mode 100644
index 0000000000000000000000000000000000000000..4f9863fb9973e7c501ab612d9f4536b7095eb797
--- /dev/null
+++ b/checkpoint-700/pytorch_model.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:aae7dbdc532c4f3da9f10e35f5ec43b2abec50ea190125b1ceb3ed97abe05646
+size 3417037
diff --git a/checkpoint-700/random_states_0.pkl b/checkpoint-700/random_states_0.pkl
new file mode 100644
index 0000000000000000000000000000000000000000..02338a39fdc868915642196b425defe854a0199f
--- /dev/null
+++ b/checkpoint-700/random_states_0.pkl
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b1db45423189931bb3b4821742c5a57835e0bb49e3149d3b23ec2fbbeb991259
+size 14599
diff --git a/checkpoint-700/scaler.pt b/checkpoint-700/scaler.pt
new file mode 100644
index 0000000000000000000000000000000000000000..5f851bd509f220aa69f1c2ca24b4f35d1b63e8a5
--- /dev/null
+++ b/checkpoint-700/scaler.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:eedd043df6cf348e21712f6001f51ee5b59434655346c4271bdeabe13333a8a9
+size 557
diff --git a/checkpoint-700/scheduler.bin b/checkpoint-700/scheduler.bin
new file mode 100644
index 0000000000000000000000000000000000000000..5b16e7cee616ce268a4b972ce7047eea19b7c821
--- /dev/null
+++ b/checkpoint-700/scheduler.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:795892f33080f59efa85c7dfcdba41f3e40832e394382f5c58eb3a75366c3a45
+size 563
diff --git a/checkpoint-800/custom_checkpoint_0.pkl b/checkpoint-800/custom_checkpoint_0.pkl
new file mode 100644
index 0000000000000000000000000000000000000000..bb19d5020ad1c99a344ebe9e72ada76cd0847216
--- /dev/null
+++ b/checkpoint-800/custom_checkpoint_0.pkl
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c369df27addc34138d987efe5a52a4242a9dd9723eb95d270c4ed181164ab299
+size 3418585
diff --git a/checkpoint-800/optimizer.bin b/checkpoint-800/optimizer.bin
new file mode 100644
index 0000000000000000000000000000000000000000..6a73a66560862c92876080a73c3af3cca9879d3b
--- /dev/null
+++ b/checkpoint-800/optimizer.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d9ecaad4cb843c512d04cbbd813ba0b5989c492ac2cd8e4b647c5efd745266f5
+size 3493125
diff --git a/checkpoint-800/pytorch_model.bin b/checkpoint-800/pytorch_model.bin
new file mode 100644
index 0000000000000000000000000000000000000000..66d7f84e168f2a4b1125ef9922aeb48a53e1331d
--- /dev/null
+++ b/checkpoint-800/pytorch_model.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:59c940df80af08a1ed17ab52c30c2248f2a9e5a29dddb39f3c4e3105cdf324f8
+size 3417037
diff --git a/checkpoint-800/random_states_0.pkl b/checkpoint-800/random_states_0.pkl
new file mode 100644
index 0000000000000000000000000000000000000000..81f7ba68cd37c4db64f2933d588b51c2a801e3cc
--- /dev/null
+++ b/checkpoint-800/random_states_0.pkl
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7bac367ba732177bfced01f8de0241ee42db5745746f43bdd8988b5045c1ad44
+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/checkpoint-900/custom_checkpoint_0.pkl b/checkpoint-900/custom_checkpoint_0.pkl
new file mode 100644
index 0000000000000000000000000000000000000000..3ecd57723e7265c8ffc8e775996a19e248d68625
--- /dev/null
+++ b/checkpoint-900/custom_checkpoint_0.pkl
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3fe0729921056adbd73ddbd0061f3023eaa41f3f53e15ff8657587c4385007a4
+size 3418585
diff --git a/checkpoint-900/optimizer.bin b/checkpoint-900/optimizer.bin
new file mode 100644
index 0000000000000000000000000000000000000000..51a3e645ff9c9739f90d4334d09b785a71e8d27e
--- /dev/null
+++ b/checkpoint-900/optimizer.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:142bf8f73873e439deabf01c3beea091dddbc46f065b50d7e33bcca1bd7af4ca
+size 3493125
diff --git a/checkpoint-900/pytorch_model.bin b/checkpoint-900/pytorch_model.bin
new file mode 100644
index 0000000000000000000000000000000000000000..cdc9de017297be0d6d3d455efae935f58380b549
--- /dev/null
+++ b/checkpoint-900/pytorch_model.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:494e631c59707f9ba1db2e2552d21fd11709d92bcd3cb187ba1b8985ab768849
+size 3417037
diff --git a/checkpoint-900/random_states_0.pkl b/checkpoint-900/random_states_0.pkl
new file mode 100644
index 0000000000000000000000000000000000000000..8a16cc0d87703e209831f8cc80e2beb9e43c4659
--- /dev/null
+++ b/checkpoint-900/random_states_0.pkl
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c54101277426e60059c8d3bd0acd8a71ae5e94fa1704b0de681fabb4138b94c8
+size 14599
diff --git a/checkpoint-900/scaler.pt b/checkpoint-900/scaler.pt
new file mode 100644
index 0000000000000000000000000000000000000000..ce88e214e8dfdcd84c4762e445a17b775da1addc
--- /dev/null
+++ b/checkpoint-900/scaler.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f35d70411c5f2791415a19167968224d98e615ab85a34b84d6ff010df96a1eb6
+size 557
diff --git a/checkpoint-900/scheduler.bin b/checkpoint-900/scheduler.bin
new file mode 100644
index 0000000000000000000000000000000000000000..8d4c3230242e4e92d0ec5552e8b36d2571b1a58c
--- /dev/null
+++ b/checkpoint-900/scheduler.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f0819f1ed37263eee386bbf633042a77c7529d9816a9e2f0a21be55a80dffb69
+size 563
diff --git a/logs/dreambooth-lora/1678107809.0264351/events.out.tfevents.1678107809.s-antonellaavad-lora-dreambooth-training-ui-33f5a-7bf69c5dcr2f6.21077.1 b/logs/dreambooth-lora/1678107809.0264351/events.out.tfevents.1678107809.s-antonellaavad-lora-dreambooth-training-ui-33f5a-7bf69c5dcr2f6.21077.1
new file mode 100644
index 0000000000000000000000000000000000000000..97e1a9af7279769bed14fa55553e6f4999e66f8c
--- /dev/null
+++ b/logs/dreambooth-lora/1678107809.0264351/events.out.tfevents.1678107809.s-antonellaavad-lora-dreambooth-training-ui-33f5a-7bf69c5dcr2f6.21077.1
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ac8880a2694c63a2d4057c6d84519e359d93f6f57f35374c99f619a81dd39f72
+size 2494
diff --git a/logs/dreambooth-lora/1678107809.028312/hparams.yml b/logs/dreambooth-lora/1678107809.028312/hparams.yml
new file mode 100644
index 0000000000000000000000000000000000000000..baded419894c3df78b3268dfa46cad4485e334dd
--- /dev/null
+++ b/logs/dreambooth-lora/1678107809.028312/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: 100
+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/hu-tao-hat
+instance_prompt: black hat, hat ornament, flower hat
+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: 1000
+mixed_precision: fp16
+num_class_images: 100
+num_train_epochs: 1000
+num_validation_images: 4
+output_dir: /home/user/app/experiments/hu-tao-hat
+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: black hat, hat ornament, flower hat ornament
+with_prior_preservation: false
diff --git a/logs/dreambooth-lora/events.out.tfevents.1678107809.s-antonellaavad-lora-dreambooth-training-ui-33f5a-7bf69c5dcr2f6.21077.0 b/logs/dreambooth-lora/events.out.tfevents.1678107809.s-antonellaavad-lora-dreambooth-training-ui-33f5a-7bf69c5dcr2f6.21077.0
new file mode 100644
index 0000000000000000000000000000000000000000..6d1a73aac19f9c7e94b383c827c928eb9a2bd242
--- /dev/null
+++ b/logs/dreambooth-lora/events.out.tfevents.1678107809.s-antonellaavad-lora-dreambooth-training-ui-33f5a-7bf69c5dcr2f6.21077.0
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9ddbc30869736f4d03819ee4f6783aff1cba7b83b1333acf227ef484e63830da
+size 15248679
diff --git a/pytorch_lora_weights.bin b/pytorch_lora_weights.bin
new file mode 100644
index 0000000000000000000000000000000000000000..f25694b580bfc313d2898517818ee370ef1dab18
--- /dev/null
+++ b/pytorch_lora_weights.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7f272ad12cb8fa693db61acedfbeb6a1440cf0ce60459793b1820bcab0a51b63
+size 3418843
diff --git a/test_images/image_0.png b/test_images/image_0.png
new file mode 100644
index 0000000000000000000000000000000000000000..7c6b16ec1a0e07c227a7ad8cb9eac5751fe3fd6e
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..999f803445f285d311aa9539b161346d6e4884a0
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..eea6be72c224513a63ff026acad89e50ab15207a
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..06080ef45166e823859353dd3b7ec09d65dde488
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..92cb56f054fa6040c5234063bdf0dd677547e755
--- /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/hu-tao-hat --output_dir=/home/user/app/experiments/hu-tao-hat --instance_prompt="black hat, hat ornament, flower hat" --resolution=512 --train_batch_size=1 --gradient_accumulation_steps=1 --learning_rate=0.0001 --lr_scheduler=constant --lr_warmup_steps=0 --max_train_steps=1000 --checkpointing_steps=100 --validation_prompt="black hat, hat ornament, flower hat ornament" --validation_epochs=100 --seed=0 --mixed_precision fp16 --use_8bit_adam
\ No newline at end of file