Update README.md
Browse files
README.md
CHANGED
@@ -1,3 +1,20 @@
|
|
1 |
-
---
|
2 |
-
license: openrail++
|
3 |
-
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
---
|
2 |
+
license: openrail++
|
3 |
+
---
|
4 |
+
|
5 |
+
Exported with
|
6 |
+
|
7 |
+
```bash
|
8 |
+
optimum-cli export neuron --model PixArt-alpha/PixArt-XL-2-512x512 --batch_size 1 --height 512 --width 512 --num_images_per_prompt 1 --torch_dtype bfloat16 --sequence_length 120 pixart_alpha_neuron_512/
|
9 |
+
```
|
10 |
+
|
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."
|
18 |
+
image = neuron_model(prompt=prompt).images[0]
|
19 |
+
image.save("out.png")
|
20 |
+
```
|