diff --git a/README.md b/README.md new file mode 100644 index 0000000000000000000000000000000000000000..3c1ad20f91668cfc9697d45745c47d1dd290763f --- /dev/null +++ b/README.md @@ -0,0 +1,22 @@ +--- +license: creativeml-openrail-m +base_model: stabilityai/stable-diffusion-2-1-base +instance_prompt: break bad +tags: +- stable-diffusion +- stable-diffusion-diffusers +- text-to-image +- diffusers +- lora +inference: true +--- +# LoRA DreamBooth - walter-white + +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 "break bad" using [DreamBooth](https://dreambooth.github.io/). You can find some example images in the following. + +Test prompt: break bad +![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..3a341ee9684a8bae2e6dac13cd6ec203aea53cf9 --- /dev/null +++ b/checkpoint-100/custom_checkpoint_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:4c0282fbdd1d717c94baa4e143fb24cd39977294445225764fc06bdce937595b +size 3418585 diff --git a/checkpoint-100/optimizer.bin b/checkpoint-100/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..c09ff035430e080e02db39143de88d8d52fb1cf4 --- /dev/null +++ b/checkpoint-100/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:2d55bc291e9cadaefe7c3a28cbf4bf4efd6eba3e443497c0399dea30747f8460 +size 3492869 diff --git a/checkpoint-100/pytorch_model.bin b/checkpoint-100/pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..ea1900fe6169683844e80b0cef0b31844b2a7d68 --- /dev/null +++ b/checkpoint-100/pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:1e49bcce1a2c0d2f6a993ebc5ce7552f337b92a835c0d0ddf99a133c2a427b12 +size 3417037 diff --git a/checkpoint-100/random_states_0.pkl b/checkpoint-100/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..398439342f9fd8a0d109dcb4eed1da4f5411cd19 --- /dev/null +++ b/checkpoint-100/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:91177f3a308e8cbe2c6c25d87a1b3a74e734cf45085918f169866bb8e267aae1 +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..43338d51b91aa278eb8beb17ca6721bcc2aaea72 --- /dev/null +++ b/checkpoint-1000/custom_checkpoint_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e142373ea63d318717514940fa2e80bb56f6fb43de54961ffeffd0f48cd18a46 +size 3418585 diff --git a/checkpoint-1000/optimizer.bin b/checkpoint-1000/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..f153bd27ddb8b7a854b7b74e18b8ec4bc0948935 --- /dev/null +++ b/checkpoint-1000/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:9c5eb6a8b308080822577f39f7e531e595985328629af2c4cb442f67825d1277 +size 3493125 diff --git a/checkpoint-1000/pytorch_model.bin b/checkpoint-1000/pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..b6b2920712eab67cb4c5b8b712439a53f2e4e65a --- /dev/null +++ b/checkpoint-1000/pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7d117ab808cb41f2bd953094fd3e63ac0d82ca692242be94e9f85c326483a42d +size 3417037 diff --git a/checkpoint-1000/random_states_0.pkl b/checkpoint-1000/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..a98763cc91dc6674a9f4a9e8684e36508dc99442 --- /dev/null +++ b/checkpoint-1000/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:92d8b428bf14aa7daa9a58200c7e1218149196ae65ff6d1271366f1ecf0d506b +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..06d94d0269787d0935073a9a4e3e88911df4d79d --- /dev/null +++ b/checkpoint-200/custom_checkpoint_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:d41250720dcca247de14ce354d8a8e2040b87cf40a4495145b32a828fa1c57d1 +size 3418585 diff --git a/checkpoint-200/optimizer.bin b/checkpoint-200/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..fd3c6ec7a058c27d96c3c5dd9a0122b90b03a579 --- /dev/null +++ b/checkpoint-200/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:9a339acb1b062986961e4000044f84f8528942a70fa88c413a79b5fa41cec018 +size 3492869 diff --git a/checkpoint-200/pytorch_model.bin b/checkpoint-200/pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..f66075fa461c2ce79b518a98ca179fae42c82d94 --- /dev/null +++ b/checkpoint-200/pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:9d035bde463b769660fdcc73b9966f4056da54f7a50c27657fa272f217a8fd07 +size 3417037 diff --git a/checkpoint-200/random_states_0.pkl b/checkpoint-200/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..1648b6b3ffc3790173d2ce29a108456e09d9ebc8 --- /dev/null +++ b/checkpoint-200/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:4ef223bd579117efb38c8361deda8afbbfac41b41c71f744606c6ff2a322a090 +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..9e14fcdc615cd13edcf1545af36609b97d8077ab --- /dev/null +++ b/checkpoint-300/custom_checkpoint_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:4225a5f5a07aea63d59b11640563d37c9eee2c83cced24276c8fac40eb7483b3 +size 3418585 diff --git a/checkpoint-300/optimizer.bin b/checkpoint-300/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..9b613929d8f53ba90ff9e8640a80d0183264ae4b --- /dev/null +++ b/checkpoint-300/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:6d70358592f29919b260de7bcd2014b617b94b06f47a4df1c0574f63a34a7bc7 +size 3493125 diff --git a/checkpoint-300/pytorch_model.bin b/checkpoint-300/pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..cfea1f95ec48ce63e4f301ea57987fb76746d52f --- /dev/null +++ b/checkpoint-300/pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:106d05c2c462d070ceb08d3383161495f96e08ee086c94c64e160c490ee828cc +size 3417037 diff --git a/checkpoint-300/random_states_0.pkl b/checkpoint-300/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..3a04402ab5c1ae45d679ff2f8b59025a800db35c --- /dev/null +++ b/checkpoint-300/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:baf99e2cd60cff6f75c04bf2117e08765a302541ca4b534efbba2067a2ee2946 +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..1e199063b3dd81dd15b78042d0a6b574cad59ef1 --- /dev/null +++ b/checkpoint-400/custom_checkpoint_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:8116fd9aa67929ad164848da8a0b97719836b074bc7774a2f33219d08d813fcd +size 3418585 diff --git a/checkpoint-400/optimizer.bin b/checkpoint-400/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..6c038df07f5a6af7d197b11dcf3854e4c3e8a670 --- /dev/null +++ b/checkpoint-400/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:08d627dfb571fad31987bd84cd735ea98e257d8b0aa1a217f7c25d712d8bfa9f +size 3493125 diff --git a/checkpoint-400/pytorch_model.bin b/checkpoint-400/pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..b9ae492f896b212fadedcd6653d0b4f512e16a37 --- /dev/null +++ b/checkpoint-400/pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:31516cb532c37c20058b6b983efdf65e83748b4d10774b5e6eb84442ec1532e2 +size 3417037 diff --git a/checkpoint-400/random_states_0.pkl b/checkpoint-400/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..25788419c0385c4f8f84ed1464014e1442f636ed --- /dev/null +++ b/checkpoint-400/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:43e618d25ee445a84211668155fd9e049ad749a56ae0506b44f4bbb2e4056d35 +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..56dbf0a31bb503dc7c09b1a0778acffa5be0bf6f --- /dev/null +++ b/checkpoint-500/custom_checkpoint_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:8402ba3b8a6c312181074cd9e9ad1c52fefd85ebda9beb051402c5518aa7e98b +size 3418585 diff --git a/checkpoint-500/optimizer.bin b/checkpoint-500/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..3e0362af1a3329a6eb1bc82bc53be9d218100094 --- /dev/null +++ b/checkpoint-500/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:5a1c4dea78470a3e596dbc42084410765137af13619faf2cff366f101d1f3841 +size 3493125 diff --git a/checkpoint-500/pytorch_model.bin b/checkpoint-500/pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..dc07aca28b1c3485b9ec220719b31f5d112caf22 --- /dev/null +++ b/checkpoint-500/pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:0f2a93f85eaebdcb37ec58fe04610eab5932c5465e94212db889086b264454f7 +size 3417037 diff --git a/checkpoint-500/random_states_0.pkl b/checkpoint-500/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..134851d4dc20d71806ac2c7bc7361a4429051b09 --- /dev/null +++ b/checkpoint-500/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c4c76616c95d6f945a69782b65fd8174733b612a6c2d05cb86d77a5a3bf83985 +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..6e1119eb8d35f7082384ca0bb74fa6b59349fab9 --- /dev/null +++ b/checkpoint-600/custom_checkpoint_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:a74b9076e5b25e1b4e8df48e0db7149cc7a8eb052b8a973d45a5059518d0eef8 +size 3418585 diff --git a/checkpoint-600/optimizer.bin b/checkpoint-600/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..50da3ecc482897be429de5c403de884fb904998c --- /dev/null +++ b/checkpoint-600/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:947ea0a96c8aa76817edf0a84cbf14b7efcd019ff9db7514c2e7e1b4367c7573 +size 3493125 diff --git a/checkpoint-600/pytorch_model.bin b/checkpoint-600/pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..9fdab0ebb549a4cb9939816e27326f4b0d573246 --- /dev/null +++ b/checkpoint-600/pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:08ae0b0e60438aaed25a91a9fadaf94b5f63fcf5fbacca658778c4afd9e53072 +size 3417037 diff --git a/checkpoint-600/random_states_0.pkl b/checkpoint-600/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..8a151d6b1c9eac76a3e8db1b758dd6482ce7caff --- /dev/null +++ b/checkpoint-600/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:9b61d285e5b0bfaf29ffe8b3fb57aa3b35139fd6c923c2db6c0e5926a19ba179 +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..7385580ee10a19c6d13bd54b6d2e5bbe00d9f8ab --- /dev/null +++ b/checkpoint-700/custom_checkpoint_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:5f67eb3810da39d2c3b7c3024618abdff3a21b6860bf582453086e60df59da76 +size 3418585 diff --git a/checkpoint-700/optimizer.bin b/checkpoint-700/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..08260e6d50a2ceb6261935ea77f2b8becf840d8f --- /dev/null +++ b/checkpoint-700/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:a188e66af2bac0584da44fb990b37872d0278a0ed56c02c47b219515209088c5 +size 3493125 diff --git a/checkpoint-700/pytorch_model.bin b/checkpoint-700/pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..d9016fdc57fd298984f071ac1119c2a49d36dcf4 --- /dev/null +++ b/checkpoint-700/pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:90d96604e340a96df6153a44298a035ac0188710f9145ed38790f88f5978cc73 +size 3417037 diff --git a/checkpoint-700/random_states_0.pkl b/checkpoint-700/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..fa47acd746bf4c9e95e28aa7ea4df3ee93b2406e --- /dev/null +++ b/checkpoint-700/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:d07b6fcf362819dccb22bdcd1ef6c1346076d1470c2e4fbcf4be4533ba505835 +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..92d2d65de4002e6dc9e8e7b976d9681cf5cd77d0 --- /dev/null +++ b/checkpoint-800/custom_checkpoint_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:a80f3aecfba6e749dd63fbc55edc0fd9ba6eca00ffaaa39c205c03a64a6833fc +size 3418585 diff --git a/checkpoint-800/optimizer.bin b/checkpoint-800/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..22b63c853826cbb3d9f6f99395b3174da0095b96 --- /dev/null +++ b/checkpoint-800/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:f2d988befc4a13cc4a6786e9bb661f5bdb83926aa203471cce9c48cce46a1ebf +size 3493125 diff --git a/checkpoint-800/pytorch_model.bin b/checkpoint-800/pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..57b929b06ecefd4fd5569258111f211c26f42fee --- /dev/null +++ b/checkpoint-800/pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:3e845e496672a807239bc7ad5f0d1cf7ca72b1acb9ce6520fa982a9c3fb2cf91 +size 3417037 diff --git a/checkpoint-800/random_states_0.pkl b/checkpoint-800/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..01ca2e6dac8f32022a0ca8f1718dcdc56f67a6a8 --- /dev/null +++ b/checkpoint-800/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:600455a4a8c49ce21ce82feec3edc19573b3f63c90cd315401af6309d8e62faf +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..9e48d742b65349a538c75bc6f32ea9961dea88f3 --- /dev/null +++ b/checkpoint-900/custom_checkpoint_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:6a9e3b5319dd61cdd215a8ea99f311093f8d8709e0d37bd313afe38e0da8f1a7 +size 3418585 diff --git a/checkpoint-900/optimizer.bin b/checkpoint-900/optimizer.bin new file mode 100644 index 0000000000000000000000000000000000000000..12f364f2151baec95c16e728369ecf6d5ae90b79 --- /dev/null +++ b/checkpoint-900/optimizer.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:fee971020ac8f4905629305ca394ceb168d9333eeeadd000842b2a05e9a6c3db +size 3493125 diff --git a/checkpoint-900/pytorch_model.bin b/checkpoint-900/pytorch_model.bin new file mode 100644 index 0000000000000000000000000000000000000000..921e90bb65bbd46f9583cf5d5c27c099e538b0af --- /dev/null +++ b/checkpoint-900/pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:5fc4c8a91bbebcd443625356f2ba4d79cb353b9e160ac14b2f9acecc756d6083 +size 3417037 diff --git a/checkpoint-900/random_states_0.pkl b/checkpoint-900/random_states_0.pkl new file mode 100644 index 0000000000000000000000000000000000000000..6c3e0079c835f875fc9c2897f4778750cdcfa5d8 --- /dev/null +++ b/checkpoint-900/random_states_0.pkl @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:78bfcd4bb66182cb9d5f6b5bd25ae445dd27a59bbf1735e4d63e246ad30b1504 +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/1676085415.9586148/events.out.tfevents.1676085415.s-antonellaavad-lora-dreambooth-training-ui-33f5a-7bf69c5dcr2f6.1381.1 b/logs/dreambooth-lora/1676085415.9586148/events.out.tfevents.1676085415.s-antonellaavad-lora-dreambooth-training-ui-33f5a-7bf69c5dcr2f6.1381.1 new file mode 100644 index 0000000000000000000000000000000000000000..5740c7e194f64a89cb7c187aecbb1d193b8c7fe1 --- /dev/null +++ b/logs/dreambooth-lora/1676085415.9586148/events.out.tfevents.1676085415.s-antonellaavad-lora-dreambooth-training-ui-33f5a-7bf69c5dcr2f6.1381.1 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b95de63b9f4f2b9c90fcbab8427636fb5209c20011d236e073764433f53d6741 +size 2437 diff --git a/logs/dreambooth-lora/1676085415.960793/hparams.yml b/logs/dreambooth-lora/1676085415.960793/hparams.yml new file mode 100644 index 0000000000000000000000000000000000000000..5f37a62860ab29822d95f5406c796481d6813e4f --- /dev/null +++ b/logs/dreambooth-lora/1676085415.960793/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/walter-white +instance_prompt: break bad +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: 39 +num_validation_images: 4 +output_dir: /home/user/app/experiments/walter-white +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: break bad +with_prior_preservation: false diff --git a/logs/dreambooth-lora/events.out.tfevents.1676085415.s-antonellaavad-lora-dreambooth-training-ui-33f5a-7bf69c5dcr2f6.1381.0 b/logs/dreambooth-lora/events.out.tfevents.1676085415.s-antonellaavad-lora-dreambooth-training-ui-33f5a-7bf69c5dcr2f6.1381.0 new file mode 100644 index 0000000000000000000000000000000000000000..a99e4c327fff16396365c7db6c964626b3fdc24a --- /dev/null +++ b/logs/dreambooth-lora/events.out.tfevents.1676085415.s-antonellaavad-lora-dreambooth-training-ui-33f5a-7bf69c5dcr2f6.1381.0 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:574d54b22ea3386e3513e124348a0a4f898301081cefb96e27455dd250049dbe +size 3692553 diff --git a/pytorch_lora_weights.bin b/pytorch_lora_weights.bin new file mode 100644 index 0000000000000000000000000000000000000000..76f19965db03e00a8880e8e49c0f2292a72ac40c --- /dev/null +++ b/pytorch_lora_weights.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:d9a07b5fedee331419c23fd8f6b0e638e8d1d78c0c9b5b998ee3767477a08763 +size 3418843 diff --git a/test_images/image_0.png b/test_images/image_0.png new file mode 100644 index 0000000000000000000000000000000000000000..ac050113eb9a69f5ae5c64ecae58e660f76e9b36 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..337f8912630a9ecd3d9ffb7812d7606bd22bfbc2 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..7b369262f184147736755aa7e1de75db3dc19346 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..8f64ab7f2804abf1e4ac64254a7b4e18033ed9c2 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..f000209c5ac2bb866ee5b17d6ddd69f5c2563d32 --- /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/walter-white --output_dir=/home/user/app/experiments/walter-white --instance_prompt="break bad" --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="break bad" --validation_epochs=100 --seed=0 --mixed_precision fp16 --use_8bit_adam \ No newline at end of file