Jingya
/

FLUX.1-Krea-dev-neuronx-tp8

Model card Files Files and versions

Jingya HF Staff commited on Aug 7

Commit

1faad62

·

verified ·

1 Parent(s): 646462e

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -5,7 +5,7 @@ license: apache-2.0
 Exported with (Inferentia 2 only):
 ```bash
-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/
 ```
 Inference

 Exported with (Inferentia 2 only):
 ```bash
+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_krea_neuron_1024_tp8/
 ```
 Inference