stabilityai

/

stable-diffusion-xl-base-1.0

like 5.94k

Follow

Stability AI 6,003

Has anyone got the result like this by executing with the onnx runtime?

Code:

from optimum.onnxruntime import ORTStableDiffusionXLPipeline

model_id = "stabilityai/stable-diffusion-xl-base-1.0"
pipeline = ORTStableDiffusionXLPipeline.from_pretrained(model_id)
prompt = "A majestic lion jumping from a big stone at night"
image = pipeline(prompt).images[0]

And this is the output(I have tried different texts)

Also it cannot run this codepipeline.to("cuda") with the following error:

ValueError: Asked to use CUDAExecutionProvider as an ONNX Runtime execution provider, but the available execution providers are ['AzureExecutionProvider', 'CPUExecutionProvider'].