Downloads · 30 days
0
Jingya/FLUX.1-Krea-dev-neuronx-tp8
FLUX.1-Krea-dev-neuronx-tp8 is a machine learning model from Jingya. Use it for the machine learning task on the model card, and read the license before you ship it in a product. The card lists the license as apache-2.0.
Downloads · 30 days
0
Access
Public
Updated Aug 7, 2025
Repo size
36.8 GB
Likes
0
Public
Click a slice to open those files.
.safetensors23.8 GB · 65%
From the Hugging Face model README
Exported with (Inferentia 2 only):
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
from optimum.neuron import NeuronFluxPipeline
pipe = NeuronFluxPipeline.from_pretrained("Jingya/FLUX.1-Krea-dev-neuronx-tp8")
prompt = "A frog holding a sign that says hello world"
image = pipe(
prompt,
guidance_scale=4.5,
).images[0]
image.save("flux-krea-dev.png")