Update README.md
Browse files
README.md
CHANGED
|
@@ -11,6 +11,14 @@ optimum-cli export neuron --model black-forest-labs/FLUX.1-Krea-dev --tensor_par
|
|
| 11 |
Inference
|
| 12 |
|
| 13 |
```python
|
|
|
|
| 14 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 15 |
```
|
| 16 |
|
|
|
|
| 11 |
Inference
|
| 12 |
|
| 13 |
```python
|
| 14 |
+
from optimum.neuron import NeuronFluxPipeline
|
| 15 |
|
| 16 |
+
pipe = NeuronFluxPipeline.from_pretrained("Jingya/FLUX.1-Krea-dev-neuronx-tp8")
|
| 17 |
+
prompt = "A frog holding a sign that says hello world"
|
| 18 |
+
image = pipe(
|
| 19 |
+
prompt,
|
| 20 |
+
guidance_scale=4.5,
|
| 21 |
+
).images[0]
|
| 22 |
+
image.save("flux-krea-dev.png")
|
| 23 |
```
|
| 24 |
|