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/dep update #160

Closed djbielejeski closed 1 year ago

djbielejeski commented 1 year ago

Updating dependencies. Cleaning up main. Fixing the checkpoint directory so you can browse it in jupyterlabs.

Note: you will need to update your image to use pytoch:1.13.1

Example in vast.ai image

To test on vast.ai

!git clone https://github.com/djbielejeski/Dreambooth-Stable-Diffusion
%cd Dreambooth-Stable-Diffusion
!git checkout feature/dep-update
%cd ..