dminhk/

stable-diffusion-xl-base-1.0-neuronx

from optimum.neuron import NeuronStableDiffusionXLPipeline

prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
stable_diffusion_xl = NeuronStableDiffusionXLPipeline.from_pretrained(
    model_id="dminhk/stable-diffusion-xl-base-1.0-neuronx", device_ids=[0, 1]
)
image = stable_diffusion_xl(prompt).images[0]