|
This is a neuron compiled model from https://huggingface.co/Lykon/dreamshaper-xl-1-0 |
|
|
|
It was compiled on the 20240123 version of the Hugging Face Deep Learing AMI. Using SDK 2.16.1. It should run on an inf2.8xlarge and maybe an inf2.xlarge. |
|
|
|
To run: |
|
|
|
``` |
|
from optimum.neuron import NeuronStableDiffusionXLPipeline |
|
pipeline = NeuronStableDiffusionXLPipeline.from_pretrained("jburtoft/Lykon-dreamshaper-xl-1-0-Neuron", device_ids=[0, 1]) |
|
prompt = "a photo of an astronaut thinking about diffusion" |
|
# generate image |
|
image = pipeline(prompt).images[0] |
|
image.save("generated_image3.jpg") |
|
``` |