Stable diffusion timeout

Hello, I am trying to generate an image within my worker, but I am getting

InferenceUpstreamError: ERROR 3007: Request timeout
at InferenceSession.run (index.js:3364:17)

my call is basic as in docs
await ai.run(“@cf/stabilityai/stable-diffusion-xl-base-1.0”, {
prompt: userPrompt,
num_steps: 10
});

even with 10 steps I got timed out

Do you guys are able to generate anything? Is it bug in my code somehow? Do I need context.waitUntil?

Not a bug in your code. As Workers AI is in beta state there may be some issues like this from time to time. You may consider reporting to the product team in the #workers-ai room of Cloudflare Developers