3 ms·
I can't see any progress on AMD/Intel GPU support :( Would love to see Vulkan or at least ROCm support. With SD1 you could follow some guides online to make it
by tormeh 4y ago
I can't see any progress on AMD/Intel GPU support :( Would love to see Vulkan or at least ROCm support. With SD1 you could follow some guides online to make it work, since PyTorch itself supports ROCm, but the state of non-Nvidia GPU support in the DL space is quite sad.
- magic_at_nodai 4y agoTry SHARK on your AMD GPUs for SD. Follow the setup here: https://github.com/nod-ai/SHARK/tree/main/shark/examples/shark_inference/stable_diffusion https://github.com/nod-ai/SHARK/tree/main/shark/examples/sha.... It works with Pytorch -> torch-mlir -> MLIR / IREE -> vulkan. Works on both Windows and Linux. And has a simple gradio web UI https://github.com/nod-ai/SHARK/tree/main/web https://github.com/nod-ai/SHARK/tree/main/web but we plan to enable better UI integrations very soon. Join us on discord https://discord.gg/RUqY2h2s9u https://discord.gg/RUqY2h2s9u if you have any trouble. Appreciate any / all feedback.