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 |
|