Views
No views yet
1from optimum.neuron import NeuronStableDiffusionXLPipeline
2
3# Load pipeline from Hugging Face repository
4pipeline = NeuronStableDiffusionXLPipeline.from_pretrained(model_dir, device_ids=[0, 1])
5
6# prompt for generating images
7prompt = "a photo of an astronaut riding a horse on mars"
8# generate image
9image = stable_diffusion(prompt).images[0]
10image.save("generated_image.jpg")1{
2 "auto_cast": "all",
3 "auto_cast_type": "bf16"
4}1{
2 "height": 1024,
3 "width": 1024,
4 "num_images_per_prompt": 1,
5 "batch_size": 1
6}