Feature/simple training

JoePenna / Dreambooth-Stable-Diffusion

Implementation of Dreambooth (https://arxiv.org/abs/2208.12242) by way of Textual Inversion (https://arxiv.org/abs/2208.01618) for Stable Diffusion (https://arxiv.org/abs/2112.10752). Tweaks focused on training faces, objects, and styles.

MIT License

3.19k stars 558 forks source link

Feature/simple training #159

Closed djbielejeski closed 1 year ago

djbielejeski commented 1 year ago

Creating a new jupyter notebook with a much improved UI

1) Build Env output (also includes a progress bar that is not included in the screenshot)

2) Downloading models

3) Training Setup

4) Training

Roughly the same, except it loads its settings from the config file generated in step 3. No more manually editing variables.

5) Copy and name checkpoint files

Also roughly the same, just cleaner to look at