Make PwW compatible with "mps" device (Apple silicon (M1/M2) hardware.)

cloneofsimo / paint-with-words-sd

Implementation of Paint-with-words with Stable Diffusion : method from eDiff-I that let you generate image from text-labeled segmentation map.

MIT License

636 stars 50 forks source link

Make PwW compatible with "mps" device (Apple silicon (M1/M2) hardware.) #28

Closed lwchen6309 closed 1 year ago

lwchen6309 commented 1 year ago

As the title, I slightly modify the script to enable PwW on "mps" device. I benchmarked the script on "MacBook pro M1 pro" and "RTX3090Ti" to ensure it can be executed on both devices.