Commit Graph

3 Commits

Author SHA1 Message Date
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