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.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Is there a way to add in the notebook or scripts so that the training step can use more than 1 GPU?

yuanbit opened this issue · comments

Is there a way to add in the notebook or scripts so that the training step can use more than 1 GPU?

commented

Should be in training yaml