Jingya HF Staff commited on
Commit
3652fcb
·
verified ·
1 Parent(s): 1faad62

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +8 -0
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