diff --git a/README.md b/README.md new file mode 100644 index 0000000000000000000000000000000000000000..016fedbbfed8ea0a95b6ab7a421f72bef0bcdc8a --- /dev/null +++ b/README.md @@ -0,0 +1,22 @@ +--- +license: creativeml-openrail-m +base_model: stabilityai/stable-diffusion-2-1-base +instance_prompt: https://huggingface.co./spaces/antonellaavad/LoRA-DreamBooth-Training-UI +tags: +- stable-diffusion +- stable-diffusion-diffusers +- text-to-image +- diffusers +- lora +inference: true +--- +# LoRA DreamBooth - https-huggingface-co-spaces-antonellaavad-lora-dreambooth-training-ui + +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 "https://huggingface.co./spaces/antonellaavad/LoRA-DreamBooth-Training-UI" using [DreamBooth](https://dreambooth.github.io/). You can find some example images in the following. + +Test prompt: https://huggingface.co./spaces/antonellaavad/LoRA-DreamBooth-Training-UI +![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..bf278f84566f4e9ee418bab89a3e7806aedd78f0 --- /dev/null +++ b/checkpoint-100/custom_checkpoint_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:0deef2b5b7dead4a553c7caea5a8a0725bd11f3d001f16f372c5b9c460de1fae +size 3418585 diff --git a/checkpoint-100/optimizer.bin b/checkpoint-100/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..36748f3b73344b65a1558e31053d3d3cb0a9d4eb --- /dev/null +++ b/checkpoint-100/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:fb2576e6bf5c5019cf3b2ca023c9d362f83b897cff7eaf5bbd17c25ab5e06cc9 +size 3492869 diff --git a/checkpoint-100/pytorch_model.bin b/checkpoint-100/pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..c0a5e47f004fb9ae0545a1888572545979d1f6e7 --- /dev/null +++ b/checkpoint-100/pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:87fbe4e04fc7a372725c6c4a4351b3c42669eb8e1a3c55643ab1268f2a057f06 +size 3417037 diff --git a/checkpoint-100/random_states_0.pkl b/checkpoint-100/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..f89daede10e396000c207a418d3ca33c955bee0d --- /dev/null +++ b/checkpoint-100/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:d6117668bdc0cb897dfd1caf9c8c1b2e8ddcafd83225caa424cbbf3fc675c1c5 +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..b1bc01125efac3b989099ba63841de358994cb73 --- /dev/null +++ b/checkpoint-1000/custom_checkpoint_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e83db424af6f4d9071958cf487501768739c2dd67673a7a6dec4afa6ade5a2db +size 3418585 diff --git a/checkpoint-1000/optimizer.bin b/checkpoint-1000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..778f5f0facdf44a3c4f60681a6aa9ed555ba044e --- /dev/null +++ b/checkpoint-1000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b0a49ff5a27c2216f72a2052f26a680a88b2497a01704868baa293e3b3f47f80 +size 3493125 diff --git a/checkpoint-1000/pytorch_model.bin b/checkpoint-1000/pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..91e1c1203edfbf346eb65785b930272c3faa42e3 --- /dev/null +++ b/checkpoint-1000/pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e0731b02f7fa8689089bc79cdc3d4358b79e97a593d6caab0667dab4fc702314 +size 3417037 diff --git a/checkpoint-1000/random_states_0.pkl b/checkpoint-1000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..2ce5466c882d44268612b1298cb3aafac86ebb9c --- /dev/null +++ b/checkpoint-1000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:f167e68662dbd25ec2776ebc383d095d13fed9d1ed5a6e1adec5c06c10319931 +size 14599 diff --git a/checkpoint-1000/scaler.pt b/checkpoint-1000/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..115263c9235918fc2a72413ed7e00c969b8d3527 --- /dev/null +++ b/checkpoint-1000/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7fa181fa360d46feed4180ea17c8b6a4a879a9b4231c2e91aff2be20be9076cc +size 557 diff --git a/checkpoint-1000/scheduler.bin b/checkpoint-1000/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..14f94902b976cce57ed30cc2be50060a200dd145 --- /dev/null +++ b/checkpoint-1000/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:77b9ebea123ef27976d26da3da6318c1ad6e54aba8cea7a2c88d3b85e8811d78 +size 563 diff --git a/checkpoint-200/custom_checkpoint_0.pkl b/checkpoint-200/custom_checkpoint_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..acc67f8f6a22952da254c8edc4694b117e4fc76f --- /dev/null +++ b/checkpoint-200/custom_checkpoint_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:5f4bef63898a71f8ccff4dbc2aa6bfc63a6cac3d920fd2102a28340db66e8f18 +size 3418585 diff --git a/checkpoint-200/optimizer.bin b/checkpoint-200/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..591b9c1f9b03c60302f7cb732dddb22bf633f720 --- /dev/null +++ b/checkpoint-200/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:38f504d3100381576fc27ee619279a704b4f4d14543f98def92926ae04d87e95 +size 3492869 diff --git a/checkpoint-200/pytorch_model.bin b/checkpoint-200/pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..3ae3742493894d11dcbe0fb689f17d4e35492376 --- /dev/null +++ b/checkpoint-200/pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:a15759f0f402ead92c524a8eb16767b9c7e9e45c521c87347773c187bf17be22 +size 3417037 diff --git a/checkpoint-200/random_states_0.pkl b/checkpoint-200/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..a147b9f2197d0e39ee19532a927ab934d2983fff --- /dev/null +++ b/checkpoint-200/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:fdf253ef6cef711c1f4cfc3d77d4963daecad2b8df20b4453ef3c3e756dfede5 +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..c0d163ee91d7b645b02ec77d7ab24e539353ff5c --- /dev/null +++ b/checkpoint-300/custom_checkpoint_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:1dbe4a7aaf096c57aed3b31bdbcd66c3ab320f923979eb194a4f6efc3c0ba600 +size 3418585 diff --git a/checkpoint-300/optimizer.bin b/checkpoint-300/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..6e847adbb77798d6bfb65b787fb8d7dc425bccb8 --- /dev/null +++ b/checkpoint-300/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:291e211862d9ab046fce8a71f8149b1fd5d4de3e130c9c1e0d87699442736b0a +size 3493125 diff --git a/checkpoint-300/pytorch_model.bin b/checkpoint-300/pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..7d1c04192dc521e23e5405636827e5ad48305000 --- /dev/null +++ b/checkpoint-300/pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:2b59ee0f9db36e3f6b8f24ee37a7629eee3d24e6865f3c81099453efd766d154 +size 3417037 diff --git a/checkpoint-300/random_states_0.pkl b/checkpoint-300/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..e9f9c2ddd5f22c8a703e813244abea8d2b9f1eb8 --- /dev/null +++ b/checkpoint-300/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:f4463a69ad500bdb682a5f0e26e07ae724106bcb71b5410c2c475d6412048c11 +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..bc41a3c3f7dca7d892dfdce09c06b465fa8624a7 --- /dev/null +++ b/checkpoint-400/custom_checkpoint_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e8673ab0e506a90ae88a321cab6a2fdfaa568dc57a7cd4474085b9fe82c3094a +size 3418585 diff --git a/checkpoint-400/optimizer.bin b/checkpoint-400/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..6505221ed6dc10c32fd1f5ebbe4977c7080ed77b --- /dev/null +++ b/checkpoint-400/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:084699de4cbf8642cf0ecac86e4e0524018759150eebc9125a1ad0ed521273d0 +size 3493125 diff --git a/checkpoint-400/pytorch_model.bin b/checkpoint-400/pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..25e7f09ac072d7fc811237ea850c06fdadfdb380 --- /dev/null +++ b/checkpoint-400/pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:1d2303a679eb81e2a7b9d507c1926f306b9003f0c5875a2224679b6022c8716c +size 3417037 diff --git a/checkpoint-400/random_states_0.pkl b/checkpoint-400/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..8e428b2518c4ad531efcbc8c3b7087526b5246a3 --- /dev/null +++ b/checkpoint-400/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b2d9adb5f613daa2fe2c95853cf90d9ae005dd9a80cf58fd37c23f24170400a2 +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..b1bc01125efac3b989099ba63841de358994cb73 --- /dev/null +++ b/checkpoint-500/custom_checkpoint_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e83db424af6f4d9071958cf487501768739c2dd67673a7a6dec4afa6ade5a2db +size 3418585 diff --git a/checkpoint-500/optimizer.bin b/checkpoint-500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..778f5f0facdf44a3c4f60681a6aa9ed555ba044e --- /dev/null +++ b/checkpoint-500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b0a49ff5a27c2216f72a2052f26a680a88b2497a01704868baa293e3b3f47f80 +size 3493125 diff --git a/checkpoint-500/pytorch_model.bin b/checkpoint-500/pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..91e1c1203edfbf346eb65785b930272c3faa42e3 --- /dev/null +++ b/checkpoint-500/pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e0731b02f7fa8689089bc79cdc3d4358b79e97a593d6caab0667dab4fc702314 +size 3417037 diff --git a/checkpoint-500/random_states_0.pkl b/checkpoint-500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..e3b949bcf043ec56e2ffe0968f0807b27e704a88 --- /dev/null +++ b/checkpoint-500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:871a63ffc8e89a1f2b7902546538c10805ef6d10e87e6f7c7eda2ba85b3a5aaf +size 14599 diff --git a/checkpoint-500/scaler.pt b/checkpoint-500/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..b6238cec9b7df564ad9b284e8aab874ac6332ea4 --- /dev/null +++ b/checkpoint-500/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:dddeac69865edf49ade8db0d259c588f65b7c9fa05d442ce27c1dd2f261e2076 +size 557 diff --git a/checkpoint-500/scheduler.bin b/checkpoint-500/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..7f6e584df3aeab88bd70ef8389b2e7c3137d6322 --- /dev/null +++ b/checkpoint-500/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:1b9d0b803c68c073d47f8430c988da8aa613959dd13c52d0c8e795c8c38209f8 +size 563 diff --git a/checkpoint-600/custom_checkpoint_0.pkl b/checkpoint-600/custom_checkpoint_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..b1bc01125efac3b989099ba63841de358994cb73 --- /dev/null +++ b/checkpoint-600/custom_checkpoint_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e83db424af6f4d9071958cf487501768739c2dd67673a7a6dec4afa6ade5a2db +size 3418585 diff --git a/checkpoint-600/optimizer.bin b/checkpoint-600/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..778f5f0facdf44a3c4f60681a6aa9ed555ba044e --- /dev/null +++ b/checkpoint-600/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b0a49ff5a27c2216f72a2052f26a680a88b2497a01704868baa293e3b3f47f80 +size 3493125 diff --git a/checkpoint-600/pytorch_model.bin b/checkpoint-600/pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..91e1c1203edfbf346eb65785b930272c3faa42e3 --- /dev/null +++ b/checkpoint-600/pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e0731b02f7fa8689089bc79cdc3d4358b79e97a593d6caab0667dab4fc702314 +size 3417037 diff --git a/checkpoint-600/random_states_0.pkl b/checkpoint-600/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..48cfa4f92f937c5dabbd09f7d43bfc7f49bf9bfb --- /dev/null +++ b/checkpoint-600/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:6689cfa2878268de087637cbd0b5d7023158a2f4e00f3fa057e0f80425dd45e8 +size 14599 diff --git a/checkpoint-600/scaler.pt b/checkpoint-600/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..115263c9235918fc2a72413ed7e00c969b8d3527 --- /dev/null +++ b/checkpoint-600/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7fa181fa360d46feed4180ea17c8b6a4a879a9b4231c2e91aff2be20be9076cc +size 557 diff --git a/checkpoint-600/scheduler.bin b/checkpoint-600/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..7f6e584df3aeab88bd70ef8389b2e7c3137d6322 --- /dev/null +++ b/checkpoint-600/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:1b9d0b803c68c073d47f8430c988da8aa613959dd13c52d0c8e795c8c38209f8 +size 563 diff --git a/checkpoint-700/custom_checkpoint_0.pkl b/checkpoint-700/custom_checkpoint_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..b1bc01125efac3b989099ba63841de358994cb73 --- /dev/null +++ b/checkpoint-700/custom_checkpoint_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e83db424af6f4d9071958cf487501768739c2dd67673a7a6dec4afa6ade5a2db +size 3418585 diff --git a/checkpoint-700/optimizer.bin b/checkpoint-700/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..778f5f0facdf44a3c4f60681a6aa9ed555ba044e --- /dev/null +++ b/checkpoint-700/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b0a49ff5a27c2216f72a2052f26a680a88b2497a01704868baa293e3b3f47f80 +size 3493125 diff --git a/checkpoint-700/pytorch_model.bin b/checkpoint-700/pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..91e1c1203edfbf346eb65785b930272c3faa42e3 --- /dev/null +++ b/checkpoint-700/pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e0731b02f7fa8689089bc79cdc3d4358b79e97a593d6caab0667dab4fc702314 +size 3417037 diff --git a/checkpoint-700/random_states_0.pkl b/checkpoint-700/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..e4bceacc0107f43b1d4bf52f2fb2d5a7f007cf35 --- /dev/null +++ b/checkpoint-700/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c64adee0b0e44f8c149768b53e70769b5eddc7f3a22198b52cc8d1e256b90c3c +size 14599 diff --git a/checkpoint-700/scaler.pt b/checkpoint-700/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..115263c9235918fc2a72413ed7e00c969b8d3527 --- /dev/null +++ b/checkpoint-700/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7fa181fa360d46feed4180ea17c8b6a4a879a9b4231c2e91aff2be20be9076cc +size 557 diff --git a/checkpoint-700/scheduler.bin b/checkpoint-700/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..285cceb7d27c147add965ac8d5d3b208358af958 --- /dev/null +++ b/checkpoint-700/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:311b1077da680dbb14fab98223d95dacc4609c2292bc4f06e86a81d292de2c21 +size 563 diff --git a/checkpoint-800/custom_checkpoint_0.pkl b/checkpoint-800/custom_checkpoint_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..b1bc01125efac3b989099ba63841de358994cb73 --- /dev/null +++ b/checkpoint-800/custom_checkpoint_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e83db424af6f4d9071958cf487501768739c2dd67673a7a6dec4afa6ade5a2db +size 3418585 diff --git a/checkpoint-800/optimizer.bin b/checkpoint-800/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..778f5f0facdf44a3c4f60681a6aa9ed555ba044e --- /dev/null +++ b/checkpoint-800/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b0a49ff5a27c2216f72a2052f26a680a88b2497a01704868baa293e3b3f47f80 +size 3493125 diff --git a/checkpoint-800/pytorch_model.bin b/checkpoint-800/pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..91e1c1203edfbf346eb65785b930272c3faa42e3 --- /dev/null +++ b/checkpoint-800/pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e0731b02f7fa8689089bc79cdc3d4358b79e97a593d6caab0667dab4fc702314 +size 3417037 diff --git a/checkpoint-800/random_states_0.pkl b/checkpoint-800/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..3d0b951193b60a3da0f88af64dff5bc19270bf9f --- /dev/null +++ b/checkpoint-800/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b60525214a5457aa8f731ad560e159d4b859c915e3f43dddabaf8b21151933e0 +size 14599 diff --git a/checkpoint-800/scaler.pt b/checkpoint-800/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..115263c9235918fc2a72413ed7e00c969b8d3527 --- /dev/null +++ b/checkpoint-800/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7fa181fa360d46feed4180ea17c8b6a4a879a9b4231c2e91aff2be20be9076cc +size 557 diff --git a/checkpoint-800/scheduler.bin b/checkpoint-800/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..86c185e633ac25adbd742ecc32b63f7795ad9e65 --- /dev/null +++ b/checkpoint-800/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:289de169e52c4fd55a89b14bbaef3ac9dd949a7f54c1ec83b6b6d3dbe7e0a9f0 +size 563 diff --git a/checkpoint-900/custom_checkpoint_0.pkl b/checkpoint-900/custom_checkpoint_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..b1bc01125efac3b989099ba63841de358994cb73 --- /dev/null +++ b/checkpoint-900/custom_checkpoint_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e83db424af6f4d9071958cf487501768739c2dd67673a7a6dec4afa6ade5a2db +size 3418585 diff --git a/checkpoint-900/optimizer.bin b/checkpoint-900/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..778f5f0facdf44a3c4f60681a6aa9ed555ba044e --- /dev/null +++ b/checkpoint-900/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b0a49ff5a27c2216f72a2052f26a680a88b2497a01704868baa293e3b3f47f80 +size 3493125 diff --git a/checkpoint-900/pytorch_model.bin b/checkpoint-900/pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..91e1c1203edfbf346eb65785b930272c3faa42e3 --- /dev/null +++ b/checkpoint-900/pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e0731b02f7fa8689089bc79cdc3d4358b79e97a593d6caab0667dab4fc702314 +size 3417037 diff --git a/checkpoint-900/random_states_0.pkl b/checkpoint-900/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..657f82af261e8e4ae077c6d5c73a2a85aff86fc2 --- /dev/null +++ b/checkpoint-900/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:9c1f77d009bd0253d6b44902af34a4422ed93354ec2f8810b73b115dee147d00 +size 14599 diff --git a/checkpoint-900/scaler.pt b/checkpoint-900/scaler.pt new file mode 100644 index 0000000000000000000000000000000000000000..115263c9235918fc2a72413ed7e00c969b8d3527 --- /dev/null +++ b/checkpoint-900/scaler.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7fa181fa360d46feed4180ea17c8b6a4a879a9b4231c2e91aff2be20be9076cc +size 557 diff --git a/checkpoint-900/scheduler.bin b/checkpoint-900/scheduler.bin new file mode 100644 index 0000000000000000000000000000000000000000..eff24e9c7f36588f5c142947aeee6a30e5886ee5 --- /dev/null +++ b/checkpoint-900/scheduler.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:1418c4dc75236568d4730c7083075f3a8b3c80301e0311c458f887e35edcca58 +size 563 diff --git a/logs/dreambooth-lora/1677716706.6606646/events.out.tfevents.1677716706.s-antonellaavad-lora-dreambooth-training-ui-33f5a-7bf69c5dcr2f6.17415.1 b/logs/dreambooth-lora/1677716706.6606646/events.out.tfevents.1677716706.s-antonellaavad-lora-dreambooth-training-ui-33f5a-7bf69c5dcr2f6.17415.1 new file mode 100644 index 0000000000000000000000000000000000000000..8cf4fc4af3c9119b6cfb335fb0e53b1d5f28eec2 --- /dev/null +++ b/logs/dreambooth-lora/1677716706.6606646/events.out.tfevents.1677716706.s-antonellaavad-lora-dreambooth-training-ui-33f5a-7bf69c5dcr2f6.17415.1 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c2ac8470cf5024f832efa9dde6239aa481c66b92f7ab0e6025eae006b1ec8498 +size 2675 diff --git a/logs/dreambooth-lora/1677716706.6628745/hparams.yml b/logs/dreambooth-lora/1677716706.6628745/hparams.yml new file mode 100644 index 0000000000000000000000000000000000000000..fece78331be99f31e6e66221edc78d98a5b4a560 --- /dev/null +++ b/logs/dreambooth-lora/1677716706.6628745/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/https-huggingface-co-spaces-antonellaavad-lora-dreambooth-training-ui +instance_prompt: https://huggingface.co./spaces/antonellaavad/LoRA-DreamBooth-Training-UI +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: 5 +num_validation_images: 4 +output_dir: /home/user/app/experiments/https-huggingface-co-spaces-antonellaavad-lora-dreambooth-training-ui +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: https://huggingface.co./spaces/antonellaavad/LoRA-DreamBooth-Training-UI +with_prior_preservation: false diff --git a/logs/dreambooth-lora/events.out.tfevents.1677716706.s-antonellaavad-lora-dreambooth-training-ui-33f5a-7bf69c5dcr2f6.17415.0 b/logs/dreambooth-lora/events.out.tfevents.1677716706.s-antonellaavad-lora-dreambooth-training-ui-33f5a-7bf69c5dcr2f6.17415.0 new file mode 100644 index 0000000000000000000000000000000000000000..ecb89a844b7ccb48f7b3f8f55de177ef2670af5c --- /dev/null +++ b/logs/dreambooth-lora/events.out.tfevents.1677716706.s-antonellaavad-lora-dreambooth-training-ui-33f5a-7bf69c5dcr2f6.17415.0 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:5bbb675c285da974e54e47949ce7c157096fe7269ece8c3d84c6ff50fdb350d8 +size 1564191 diff --git a/pytorch_lora_weights.bin b/pytorch_lora_weights.bin new file mode 100644 index 0000000000000000000000000000000000000000..1582fc17a5a4025530c8073ca684fdd557e0e003 --- /dev/null +++ b/pytorch_lora_weights.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:d2baca0a63db296d4cdb29ca14d4a639ddf7f37f29900eaa1681db0f82d40621 +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..edde5602a9775d6b9ab9098402ec055a68bada82 --- /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/https-huggingface-co-spaces-antonellaavad-lora-dreambooth-training-ui --output_dir=/home/user/app/experiments/https-huggingface-co-spaces-antonellaavad-lora-dreambooth-training-ui --instance_prompt="https://huggingface.co./spaces/antonellaavad/LoRA-DreamBooth-Training-UI" --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="https://huggingface.co./spaces/antonellaavad/LoRA-DreamBooth-Training-UI" --validation_epochs=100 --seed=0 --mixed_precision fp16 --use_8bit_adam \ No newline at end of file