Stable Diffusion, AMD, and nod.ai

·

1 min read

Background

Previously, I failed to get anything running using AMD and ROCm for stable diffusion. One suggestion online was using nod.ai SHARK, which uses IREE to run ML through Vulkan. SHARK supports Radeon out of the box, so I decided to give it a try. As with most things, I do things the "developer way," so those are the directions I followed. SHARK is a complete solution though, so it's different cobbling and building stuff like llama.cpp or Gaussian Splatting.

git clone https://github.com/nod-ai/SHARK.git
cd .\SHARK\
.\setup_venv.ps1
 cd .\apps\stable_diffusion\web\
python .\index.py

This, you can point your browser to your localhost:8080 and put in prompts like "Harry Potter on a horse"

This 512x512 image took about 20 secs to generate.