Jingya HF Staff commited on
Commit
646462e
·
verified ·
1 Parent(s): c104964

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +6 -1
README.md CHANGED
@@ -5,7 +5,12 @@ license: apache-2.0
5
  Exported with (Inferentia 2 only):
6
 
7
  ```bash
8
- optimum-cli export neuron --model black-forest-labs/FLUX.1-Kontext-dev --tensor_parallel_size 8 --batch_size 1 --height 1024 --width 1024 --num_images_per_prompt 1 --sequence_length 512 --torch_dtype bfloat16 flux_kontext_neuron_1024_tp8/
9
  ```
10
 
 
 
 
 
 
11
 
 
5
  Exported with (Inferentia 2 only):
6
 
7
  ```bash
8
+ optimum-cli export neuron --model black-forest-labs/FLUX.1-Krea-dev --tensor_parallel_size 8 --batch_size 1 --height 1024 --width 1024 --num_images_per_prompt 1 --sequence_length 512 --torch_dtype bfloat16 flux_kontext_neuron_1024_tp8/
9
  ```
10
 
11
+ Inference
12
+
13
+ ```python
14
+
15
+ ```
16