Update README.md
Browse files
README.md
CHANGED
|
@@ -11,7 +11,7 @@ optimum-cli export neuron --model PixArt-alpha/PixArt-XL-2-512x512 --batch_size
|
|
| 11 |
Inference
|
| 12 |
|
| 13 |
```python
|
| 14 |
-
|
| 15 |
|
| 16 |
neuron_model = NeuronPixArtAlphaPipeline.from_pretrained(save_directory)
|
| 17 |
prompt = "Mario got caught in a McDonald's while eating hamburgers with pickles."
|
|
|
|
| 11 |
Inference
|
| 12 |
|
| 13 |
```python
|
| 14 |
+
from optimum.neuron import NeuronPixArtAlphaPipeline
|
| 15 |
|
| 16 |
neuron_model = NeuronPixArtAlphaPipeline.from_pretrained(save_directory)
|
| 17 |
prompt = "Mario got caught in a McDonald's while eating hamburgers with pickles."
|