Adapted from https://note.com/kohya_ss/n/nbf7ce8d80f29 for easier cloning - View it on GitHub