search

d8ahazard / sd_dreambooth_extension

Other
1.85k stars 283 forks source link

Support AMD with 2 lines of code #1424

Closed jolyons123 closed 6 months ago

jolyons123 commented 6 months ago

Kindly read the entire form below and fill it out with the requested information.

WHAT AMD compatibility can be achieved by commenting out two lines.

HOW

  1. During setup, skip the bitsandbytes check. Comment out https://github.com/d8ahazard/sd_dreambooth_extension/blob/main/postinstall.py#L45
  2. During class image generation, xformers is used by default. Comment out https://github.com/d8ahazard/sd_dreambooth_extension/blob/main/helpers/image_builder.py#L109

The above listed lines could be executed whenever NVIDIA GPU is detected (similar to https://github.com/AUTOMATIC1111/stable-diffusion-webui/blob/master/webui.sh#L126)

d8ahazard commented 6 months ago

Suggested changes implemented with a1f5a5442cc92d89a60dfa247faeade6c023ac52. Please try and let me know if you have issues.