search

XavierXiao / Dreambooth-Stable-Diffusion

Implementation of Dreambooth (https://arxiv.org/abs/2208.12242) with Stable Diffusion
MIT License
7.6k stars 795 forks source link

Update main.py #66

Open Tuxius opened 2 years ago

Tuxius commented 2 years ago

See issue #65 (under Windows NCCL is not available, instead GLOO needs to be used) With this change GLOO is used per default if the OS is Windows