File size: 601 Bytes
d5341b9
 
 
 
 
 
 
 
 
 
 
 
 
 
 
fc69b23
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
This is a dummy model used for testing `NeuronStableDiffusionPipeline` of `optimum-neuron`.

```python
from optimum.neuron import NeuronStableDiffusionPipeline

model_id = "optimum/tiny-stable-diffusion-neuronx"
input_shapes = {"batch_size": 2, "sequence_length": 18, "num_channels": 4, "height": 64, "width": 64}

stable_diffusion = NeuronStableDiffusionPipeline.from_pretrained(model_id, export=True, **input_shapes)
save_directory = "sd_neuron/"
stable_diffusion.save_pretrained(save_directory)


prompt = "sailing ship in storm by Leonardo da Vinci"
image = stable_diffusion(prompt).images[0]
```