dminhk's picture
Update README.md
85bfddb
metadata
license: openrail++

Amazon EC2 Inf2 Instances

How to use Diffusers library for inference

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]