Adding a version of the main.py script that automatically prunes about 10GB off the ckpt. prune.py is from waifu diffusion repo
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.