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