diff --git a/.gitattributes b/.gitattributes index a6344aac8c09253b3b630fb776ae94478aa0275b..adb11a537ef72c23c986f8dd8c22d84deb7de45e 100644 --- a/.gitattributes +++ b/.gitattributes @@ -33,3 +33,7 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text *.zip filter=lfs diff=lfs merge=lfs -text *.zst filter=lfs diff=lfs merge=lfs -text *tfevents* filter=lfs diff=lfs merge=lfs -text +test_images/image_0.png filter=lfs diff=lfs merge=lfs -text +test_images/image_1.png filter=lfs diff=lfs merge=lfs -text +test_images/image_2.png filter=lfs diff=lfs merge=lfs -text +test_images/image_3.png filter=lfs diff=lfs merge=lfs -text diff --git a/README.md b/README.md index 6a002e8b7932dca8c1ea253487221b1eaf4ab351..de849418dc9b86ec5ae90158ff0ca211b5127e70 100644 --- a/README.md +++ b/README.md @@ -14,4 +14,9 @@ inference: true 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 "my_rhinoplasty" using [DreamBooth](https://dreambooth.github.io/). You can find some example images in the following. +Test prompt: my_rhinoplasty +![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..f3672b691903f61a7765e635b5cbed510d07440f --- /dev/null +++ b/checkpoint-100/custom_checkpoint_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:4d8656b81a3950715b3eebb0f166bd53b416f685555de07a2932e21269da01cc +size 3418585 diff --git a/checkpoint-100/optimizer.bin b/checkpoint-100/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..2aad74f5ac88a329a3323c7e2dc6da13365184c3 --- /dev/null +++ b/checkpoint-100/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:34b4a434b2664ebcdf791ef66eb9619d9d0c241a8e635a87539f06fcdb52cae7 +size 6853829 diff --git a/checkpoint-100/pytorch_model.bin b/checkpoint-100/pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..cffafbbc0dcd1c9ef1fdf54d5a845322ba977e6d --- /dev/null +++ b/checkpoint-100/pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:bfb7a9ec5b6a638253a67a413267b98d8ce59613bcb8bb6091b76d0ed84b5d97 +size 3417037 diff --git a/checkpoint-100/random_states_0.pkl b/checkpoint-100/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..6b403c1f04afa6f0ffbcc90133047680c37e0326 --- /dev/null +++ b/checkpoint-100/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:1a7a1bcae4824b28c8ba36e1abf327b2943f31bf146dc949de9585c1b21d2f47 +size 14599 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..066ca36dc074cc45ce409d073b826f6456e9df23 --- /dev/null +++ b/checkpoint-1000/custom_checkpoint_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:8fecb16578f7038afb4be1bf92fded0545198a10f89a6c3fea5737635fb776d3 +size 3418585 diff --git a/checkpoint-1000/optimizer.bin b/checkpoint-1000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..707aca4a9704f1c2283c33b07a838c7965046d52 --- /dev/null +++ b/checkpoint-1000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:27ff74c35cdcad5b6b61f185cd8c61f13e8a98fa0c2e817fd8349a6b57cd5e45 +size 6853829 diff --git a/checkpoint-1000/pytorch_model.bin b/checkpoint-1000/pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..71bc376d2f156d55d8c19dcc4e5c6211ddba0540 --- /dev/null +++ b/checkpoint-1000/pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:befbfd01ab3f777c3e055175089e3a69befbe8c1b32fe9a7e009a4624cad6086 +size 3417037 diff --git a/checkpoint-1000/random_states_0.pkl b/checkpoint-1000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..7d88546cb1d2e92eef985936a02eccf73c67ab6b --- /dev/null +++ b/checkpoint-1000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e86674fb15efc012db4dfba102ebab75f9dae1b03f4ce1cc1218623a1225271a +size 14599 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..e3f5000f4155afb02a3829d28ebc14b145f695b0 --- /dev/null +++ b/checkpoint-200/custom_checkpoint_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:382f53fa2e07299c5a799c91109df1493d4cfaf15c68c8f6cb397a484d0b9216 +size 3418585 diff --git a/checkpoint-200/optimizer.bin b/checkpoint-200/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..e75defdd24948f547feb69a7ec4d1bd21cb38e45 --- /dev/null +++ b/checkpoint-200/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:2f4af7f8f7af46d12487d2cfed578a08cf75322a696230488908a08873150b9b +size 6853829 diff --git a/checkpoint-200/pytorch_model.bin b/checkpoint-200/pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..6fb1edb424251297b2e280af43aba1298f23dc5c --- /dev/null +++ b/checkpoint-200/pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b5f7d0b6c7c94c513cd2959b206b9090fd3629f23f10636e0c3cd3497333eedd +size 3417037 diff --git a/checkpoint-200/random_states_0.pkl b/checkpoint-200/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..316e2d7bf9d41b7e6bd4cd4b9ca09618cd2ba1ee --- /dev/null +++ b/checkpoint-200/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:a031339fb595cd08efb5a994a0326f4855b87bf05a72bac21c7dc8df30902d24 +size 14599 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..5aef8cf2e3ce60646089fd6aef556953f2ad1465 --- /dev/null +++ b/checkpoint-300/custom_checkpoint_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:f7138a7b23eec57f91d5ce9888108bfd77f98931b2c51b4d6f335d416e00dcf2 +size 3418585 diff --git a/checkpoint-300/optimizer.bin b/checkpoint-300/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..a9d295120eab17333f48d1180ec98f93de0459a4 --- /dev/null +++ b/checkpoint-300/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:1cd0ddd9bb37de6c4749ac41df167ab744bd6497295b66f92ebba53bd736399c +size 6853829 diff --git a/checkpoint-300/pytorch_model.bin b/checkpoint-300/pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..4428afb8a9ae7c2c5e7a5d236d5e1d5b6254c83e --- /dev/null +++ b/checkpoint-300/pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c4399adfb88e577ac0626c61ea8ea04f24adca67a24a587f45d049995de31940 +size 3417037 diff --git a/checkpoint-300/random_states_0.pkl b/checkpoint-300/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..07618facbb82140cc3b02dca26792b0daf8aaed9 --- /dev/null +++ b/checkpoint-300/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b58b12cc23f768be97df74649c6322a3350790f55ec7115b446c8dca94df0c96 +size 14599 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..ee1367957d806fbef14a21873d4d684cd1610dd2 --- /dev/null +++ b/checkpoint-400/custom_checkpoint_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b53b1f4d3836604bf11489e09844a4edb8673afb9ff13ccb70de30c16ff3b1c2 +size 3418585 diff --git a/checkpoint-400/optimizer.bin b/checkpoint-400/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..079804a5d112a821d3b5d23949beb5f827ee4c9e --- /dev/null +++ b/checkpoint-400/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:1bba75676783076329f97778dd0e63d16360d27bf5caa65216f9440ceb3299df +size 6853829 diff --git a/checkpoint-400/pytorch_model.bin b/checkpoint-400/pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..aabba96e0bae35ad1c06fca5bc91edd9c4fcd92b --- /dev/null +++ b/checkpoint-400/pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:1ff532e5a8ccfa3122e120b4140bcc72963d8aa92266511dda6169ef326c4355 +size 3417037 diff --git a/checkpoint-400/random_states_0.pkl b/checkpoint-400/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..71290172afea362503ca0e2b792712e9aaf56127 --- /dev/null +++ b/checkpoint-400/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:73ab26bced5fcff44c0220b44540f5f497af893e0016a25fbf92618951016d75 +size 14599 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..6fa2289baf4bb45355703e9f64c6564cd721f57e --- /dev/null +++ b/checkpoint-500/custom_checkpoint_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:ceb56e3d6bd1d766c5f506ffdee816cf8abc7aa19df58e585543ea1ecebf35f8 +size 3418585 diff --git a/checkpoint-500/optimizer.bin b/checkpoint-500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..565a0dfe8961aee073418eff17f964458f864a0f --- /dev/null +++ b/checkpoint-500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7c1aa032cb390d380f466c27677b94fa495fa58095f5f7d38ba11375e96b0046 +size 6853829 diff --git a/checkpoint-500/pytorch_model.bin b/checkpoint-500/pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..9b85717672174653088bd1d28e8ebd84b79cbbac --- /dev/null +++ b/checkpoint-500/pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:f04bfeb2f8b8d4643265b7f390c00ba165a8661a507898750ecbea0109f7318a +size 3417037 diff --git a/checkpoint-500/random_states_0.pkl b/checkpoint-500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..10845f0c002a7743e6037be503dc38a0603aa4e0 --- /dev/null +++ b/checkpoint-500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:411c943c11ceaa555861e3e1206a12401c7cc533016fde9fb581ec28892b3d98 +size 14599 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..2ff49617fbd19a709e19c25d499059a4e40397f2 --- /dev/null +++ b/checkpoint-600/custom_checkpoint_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:4b040decbd29b1d1cfadf0cd1196cd36e3fb12f6c8a73cae589dc19d248bc5e8 +size 3418585 diff --git a/checkpoint-600/optimizer.bin b/checkpoint-600/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..fcf328476f4e00a5f10f99bfcfb4181e539a48ba --- /dev/null +++ b/checkpoint-600/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:6c3a27aa3a22b7f12c657c0137fce428915769d34d2fa43c7e9a26fc70d55a8b +size 6853829 diff --git a/checkpoint-600/pytorch_model.bin b/checkpoint-600/pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..4cb154d904476b7c1c0af3c12c2ac59f3bf19bc3 --- /dev/null +++ b/checkpoint-600/pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b948159c1c84d6268cf5fc0e572d59b4dc7fe2e071bf9f3bb500fae844a7e190 +size 3417037 diff --git a/checkpoint-600/random_states_0.pkl b/checkpoint-600/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..ba1278c6043b340fac251759da171e3b2e3f8d00 --- /dev/null +++ b/checkpoint-600/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:8419c2cf4d0ca26a5e4ad15578da2c8af50da2c8f0b3f3dd6f4896e86da24818 +size 14599 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..0237f73e0f7276f5149ebca091facc8472b1a20d --- /dev/null +++ b/checkpoint-700/custom_checkpoint_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b32175dac10e999cd1004ad7e9c7c4fe71bbef0e9a1abf2e99683d3c5ae6f673 +size 3418585 diff --git a/checkpoint-700/optimizer.bin b/checkpoint-700/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..c050b2749c51f00494272a28a44f5ea41db71c3d --- /dev/null +++ b/checkpoint-700/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:3e4510d598443ce40f2802020bb245f7f18e97adcc1668716e2b913f97594584 +size 6853829 diff --git a/checkpoint-700/pytorch_model.bin b/checkpoint-700/pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..c68eefdaa3a4df661d9aa1dbf39d00c3beba1a36 --- /dev/null +++ b/checkpoint-700/pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:bcb45d6ab99cd7e64564e480a3519270ca9b78a7097dd506b2a479fd474cafda +size 3417037 diff --git a/checkpoint-700/random_states_0.pkl b/checkpoint-700/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..6688ed98dab6a10bf254497e4f125fbf0ea7c0ff --- /dev/null +++ b/checkpoint-700/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:4504cbf581c25ab461e385a9ec640b94f31c05944c9b22f4372a94c8c945919a +size 14599 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..66046c99309f9f8324ae0a002ad61b19ae176fc6 --- /dev/null +++ b/checkpoint-800/custom_checkpoint_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:9ac5b6da35d0b05f9d018f3aa610fd4aa01f085d5f5c5895cbc385a9328702db +size 3418585 diff --git a/checkpoint-800/optimizer.bin b/checkpoint-800/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..148762ba52b1a93a34c4a7d00b463de196a17536 --- /dev/null +++ b/checkpoint-800/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:27e197e65a7891c73ac38f304700518a02436f646b22cd45d658e897ae24548d +size 6853829 diff --git a/checkpoint-800/pytorch_model.bin b/checkpoint-800/pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..d182fb6f0f2c9df9c718200f6b092c3a5a8ea677 --- /dev/null +++ b/checkpoint-800/pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b5e497df92d9e776a645a3ac0aba7405093c76b0635402903682fff09e3a1ec2 +size 3417037 diff --git a/checkpoint-800/random_states_0.pkl b/checkpoint-800/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..7a782589384fbba9e51761c5ba621252d0604908 --- /dev/null +++ b/checkpoint-800/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:9d34b3043b6ac89474750314b86516913d879b5111439afc56276f89f2b5c630 +size 14599 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..77b018709739cb9ea100b987b5b551450e6f7ffa --- /dev/null +++ b/checkpoint-900/custom_checkpoint_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b6726b2bafacaac7f0ae981ad8e4355a2248bc7645bace8481b199592a4561a4 +size 3418585 diff --git a/checkpoint-900/optimizer.bin b/checkpoint-900/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..4efacc5667dbff358b56278c9acb63958c27beb2 --- /dev/null +++ b/checkpoint-900/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:6d32ab4f95104fbd7a0931acff11fbe84e7e6e983d01baf741dd13cdb5afa1bf +size 6853829 diff --git a/checkpoint-900/pytorch_model.bin b/checkpoint-900/pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..cdbe77f4fbbfcaa8fee630e45393088e7f3460ab --- /dev/null +++ b/checkpoint-900/pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:55c8b5eb86b7a391b0231b419be87f21340e5574e8d51778a2ff2b406464548a +size 3417037 diff --git a/checkpoint-900/random_states_0.pkl b/checkpoint-900/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..4ba0a599b85f9edac0ac57d5c5b783dbb0968004 --- /dev/null +++ b/checkpoint-900/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:dd5f9248b360ac93b2c040553c8f22ccfa25a5708167a78ac00d3d5de362bce4 +size 14599 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/pytorch_lora_weights.safetensors b/pytorch_lora_weights.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..c32f06f260dfc8e891b5953865495d83f9d26e8e --- /dev/null +++ b/pytorch_lora_weights.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:9674fa58f16f384e85d58f3d99a5c62136c15a820d4ba4f4d833f5ed5d09ad72 +size 3358448 diff --git a/test_images/image_0.png b/test_images/image_0.png new file mode 100644 index 0000000000000000000000000000000000000000..af66f3ffeb427c59dbc58111486943c4e604e8e7 --- /dev/null +++ b/test_images/image_0.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:bfca001ff132690dc3add628098eab57665c55a2dd1fe2eead7641831e62f299 +size 375475 diff --git a/test_images/image_1.png b/test_images/image_1.png new file mode 100644 index 0000000000000000000000000000000000000000..7bf00165f507a4ef47740e1edbb9c0b3faf42cb1 --- /dev/null +++ b/test_images/image_1.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:caf6bace8a6bbbf86a3e8424fc491999d88f33bbe705b09bd9f2cf6be39105cb +size 326428 diff --git a/test_images/image_2.png b/test_images/image_2.png new file mode 100644 index 0000000000000000000000000000000000000000..2888f11670ff6941f4e7d7305aa0c78c0a675958 --- /dev/null +++ b/test_images/image_2.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:a8a84ee2d36164aae88522908ffa8ccf0427093df3982c616ac4234cd22eca60 +size 331440 diff --git a/test_images/image_3.png b/test_images/image_3.png new file mode 100644 index 0000000000000000000000000000000000000000..e303828e79f3f11ad0d12e8f07f6461f70e80167 --- /dev/null +++ b/test_images/image_3.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:19c77428ebb2a41489d8b4adf659db58f9a3ce191d44cc17c1345d956a278592 +size 319450 diff --git a/train.sh b/train.sh index 2e20fb30c864c3c308000c64f97776aaf2e8afe9..aab6a9063a892b553b8e271d83f0b101207d78f2 100644 --- a/train.sh +++ b/train.sh @@ -1 +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/rhinoplasty-lora --output_dir=/home/user/app/experiments/rhinoplasty-lora --instance_prompt="my_rhinoplasty" --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="my_rhinoplasty" --validation_epochs=100 --seed=0 --mixed_precision fp16 --use_8bit_adam \ No newline at end of file +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/rhinoplasty-lora --output_dir=/home/user/app/experiments/rhinoplasty-lora --instance_prompt="my_rhinoplasty" --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="my_rhinoplasty" --validation_epochs=100 --seed=0 \ No newline at end of file