Commit Graph

32 Commits

Author SHA1 Message Date
David Bielejeski aadc5ebea0 Fixing for duplicate arg error. 2022-09-28 16:17:25 -05:00
David Bielejeski 1929b564bb Fixing ipywidgets 2022-09-28 16:05:58 -05:00
David Bielejeski b7d4fcec6e Clarifying calling 2022-09-28 15:27:04 -05:00
David Bielejeski 77c8431c2c Forgot a \ 2022-09-28 15:25:43 -05:00
David Bielejeski 5f471dc3ab Minor fix for pruning and moving. 2022-09-28 15:25:00 -05:00
David Bielejeski fce2d1f17d Fixing batch_size to be max_steps. Turning --no-test on by default. 2022-09-28 15:19:31 -05:00
JoePenna 722fd01fcf Update dreambooth_runpod_joepenna.ipynb
batch_size >> max_steps
2022-09-28 07:12:33 -07:00
David Bielejeski 1c7c50f9d2 Fixing multiple runs erroring on creating the "trained_models" folder. 2022-09-27 09:25:20 -05:00
David Bielejeski 884cb2586a Minor fix for the reg data. 2022-09-26 16:42:14 -05:00
David Bielejeski 8761ce781c Cleanup 2022-09-26 16:37:20 -05:00
David B 41674ab978
Update dreambooth_runpod_joepenna.ipynb 2022-09-26 15:14:56 -05:00
David Bielejeski 5b15f9058e Adding note. 2022-09-26 15:13:56 -05:00
David Bielejeski fb9f6c75e7 Revert "Update dreambooth_runpod_joepenna.ipynb"
This reverts commit 358ef69295.
2022-09-26 15:11:20 -05:00
David B 358ef69295
Update dreambooth_runpod_joepenna.ipynb
Remove output
2022-09-26 15:08:21 -05:00
David Bielejeski 20456602dc Fixing reg_data_root. 2022-09-26 13:53:54 -05:00
David Bielejeski 8ea1695848 Resetting back to person_ddim 2022-09-26 13:34:44 -05:00
David Bielejeski df01213d32 Removing unused/not working dropdown. 2022-09-26 13:08:04 -05:00
David Bielejeski 67154ad9b5 Minor updates 2022-09-26 12:41:03 -05:00
David Bielejeski 2671369079 Attempting auto-prune and checkpoint saving 2022-09-26 08:27:10 -05:00
David Bielejeski fd298537c2 Porting changes over from my branch. 2022-09-26 08:01:18 -05:00
JoePenna 225bec8b3f
Update dreambooth_runpod_joepenna.ipynb 2022-09-25 20:46:29 -07:00
JoePenna 6304662726 Update dreambooth_runpod_joepenna.ipynb 2022-09-24 02:25:36 -07:00
JoePenna d47c06f25c Update dreambooth_runpod_joepenna.ipynb 2022-09-24 01:59:04 -07:00
JoePenna fb31ab5bd5 Update dreambooth_runpod_joepenna.ipynb 2022-09-24 01:58:23 -07:00
JoePenna 2515f2d1c3 Update dreambooth_runpod_joepenna.ipynb 2022-09-24 01:50:37 -07:00
JoePenna c2ad5aec7c Update dreambooth_runpod_joepenna.ipynb 2022-09-24 01:44:25 -07:00
JoePenna 7a4292a108 Update dreambooth_runpod_joepenna.ipynb 2022-09-24 01:12:14 -07:00
JoePenna 07a8231839 Update dreambooth_runpod_joepenna.ipynb 2022-09-24 01:09:31 -07:00
JoePenna 2d06c67408 Update dreambooth_runpod_joepenna.ipynb 2022-09-24 01:07:44 -07:00
JoePenna ce7d1dffe7 Improvements by David Bielejeski, Niko Pueringer, Maki99 2022-09-24 00:32:50 -07:00
JoePenna 33be8bb255 Adding pruning option
Adding a version of the main.py script that automatically prunes about 10GB off the ckpt.

prune.py is from waifu diffusion repo
2022-09-20 13:39:57 -07:00
JoePenna 99c2e27053 Tweaks for training people
A few tweaks to the repo to help training faces, though I've also used these same settings to train objects and styles.

Also adding a notebook to make it easier to run the code on vast.ai / runpod / other cloud computing platforms.
2022-09-17 16:00:31 -07:00