Update README.md
Browse files
README.md
CHANGED
@@ -1 +1,16 @@
|
|
1 |
-
This is a dummy model used for testing `NeuronStableDiffusionPipeline` of `optimum-neuron`.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
This is a dummy model used for testing `NeuronStableDiffusionPipeline` of `optimum-neuron`.
|
2 |
+
|
3 |
+
```python
|
4 |
+
from optimum.neuron import NeuronStableDiffusionPipeline
|
5 |
+
|
6 |
+
model_id = "optimum/tiny-stable-diffusion-neuronx"
|
7 |
+
input_shapes = {"batch_size": 2, "sequence_length": 18, "num_channels": 4, "height": 64, "width": 64}
|
8 |
+
|
9 |
+
stable_diffusion = NeuronStableDiffusionPipeline.from_pretrained(model_id, export=True, **input_shapes)
|
10 |
+
save_directory = "sd_neuron/"
|
11 |
+
stable_diffusion.save_pretrained(save_directory)
|
12 |
+
|
13 |
+
|
14 |
+
prompt = "sailing ship in storm by Leonardo da Vinci"
|
15 |
+
image = stable_diffusion(prompt).images[0]
|
16 |
+
```
|