Views
No views yet
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/1from diffusers.utils import load_image
2
3from optimum.neuron import NeuronFluxKontextPipeline
4
5
6pipe = NeuronFluxKontextPipeline.from_pretrained("Jingya/Flux.1-Kontext-dev-1024x1024-neuronx-tp8")
7input_image = load_image("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/diffusers/cat.png")
8image = pipe(
9 image=input_image,
10 prompt="Add a hat to the cat",
11 guidance_scale=2.5
12).images[0]
13image = image.resize(input_image.size)
14image.save("flux_kontext.png")