flux model family
Tasks
Supported architectures and task heads
The Hugging Face values below are copied from checkpoint metadata at the recorded revision. The TRTMC task contract comes from the exact E2E recipe. Architecture identity selects or describes a source graph; it does not imply that TRTMC reproduces every Hugging Face head with the same model_type. For example, an encoder recipe may consume the base model and intentionally return hidden states instead of the checkpoint's pretraining or classification logits.
HF model_type | HF architecture / pipeline class | TRTMC task contract | Exact recipe profiles |
|---|---|---|---|
| — | Flux2PipelineSource: model_index._class_nameMetadata: model_index.json at 26afe3a78bb2 | diffusion_media_generation | flux-2-devflux-2-dev-fp8flux-2-dev-fp8-l0flux-2-dev-fp8-l0-tp4flux-2-dev-l0flux-2-dev-l0-tp4 |
| — | FluxPipelineSource: model_index._class_nameMetadata: model_index.json at 741f7c3ce8b3 | diffusion_media_generation | flux-schnellflux-schnell-l0flux-schnell-l0-batch2flux-schnell-l0-cp4flux-schnell-l0-tp4 |
Declared recipes
Each row comes from a model-owned E2E manifest. It declares a test recipe; it is not a live pass receipt for every hardware target.
| Recipe | Exact Hugging Face checkpoint | Task | Build configuration | Declared E2E cases | Manifest |
|---|---|---|---|---|---|
flux-2-dev | black-forest-labs/FLUX.2-dev | bf16; single device | flux-2-dev | tests/e2e/models/flux/manifests/flux-2-dev.json | |
flux-2-dev-fp8 | black-forest-labs/FLUX.2-dev | fp16; single device | flux-2-dev-fp8 | tests/e2e/models/flux/manifests/flux-2-dev-fp8.json | |
flux-2-dev-fp8-l0 | black-forest-labs/FLUX.2-dev | fp16; single device | flux-2-dev-fp8-l0 | tests/e2e/models/flux/manifests/flux-2-dev-fp8-l0.json | |
flux-2-dev-fp8-l0-tp4 | black-forest-labs/FLUX.2-dev | fp16; TP4 | flux-2-dev-fp8-l0-tp4 | tests/e2e/models/flux/manifests/flux-2-dev-fp8-l0-tp4.json | |
flux-2-dev-l0 | black-forest-labs/FLUX.2-dev | bf16; single device | flux-2-dev-l0 | tests/e2e/models/flux/manifests/flux-2-dev-l0.json | |
flux-2-dev-l0-tp4 | black-forest-labs/FLUX.2-dev | fp16; TP4 | flux-2-dev-l0-tp4 | tests/e2e/models/flux/manifests/flux-2-dev-l0-tp4.json | |
flux-schnell | black-forest-labs/FLUX.1-schnell | fp16; single device | flux-schnell | tests/e2e/models/flux/manifests/flux-schnell.json | |
flux-schnell-l0 | black-forest-labs/FLUX.1-schnell | fp16; single device | flux-schnell-l0 | tests/e2e/models/flux/manifests/flux-schnell-l0.json | |
flux-schnell-l0-batch2 | black-forest-labs/FLUX.1-schnell | fp16; single device | flux-schnell-l0-batch2 | tests/e2e/models/flux/manifests/flux-schnell-l0-batch2.json | |
flux-schnell-l0-cp4 | black-forest-labs/FLUX.1-schnell | family default; CP4 | flux-schnell-l0-cp4 | tests/e2e/models/flux/manifests/flux-schnell-l0-cp4.json | |
flux-schnell-l0-tp4 | black-forest-labs/FLUX.1-schnell | family default; TP4 | flux-schnell-l0-tp4 | tests/e2e/models/flux/manifests/flux-schnell-l0-tp4.json |
Family-owned configuration
This family does not declare a family-owned --set namespace. Use the explicit CLI options shown below and the shared configuration namespaces documented in Configure Runtime Behavior.
Family-specific CLI contracts
Inputs and options below are filtered by the declared E2E task and the methods and configuration fields used by that family's native runtime implementation. The global CLI parser accepts a wider union of flags; flags absent here are not declared for this family.
trtmc run
Generate an image from a text prompt.
Declared recipes: flux-2-dev, flux-2-dev-fp8, flux-2-dev-fp8-l0, flux-2-dev-fp8-l0-tp4, flux-2-dev-l0, flux-2-dev-l0-tp4, flux-schnell, flux-schnell-l0, flux-schnell-l0-batch2, flux-schnell-l0-cp4, flux-schnell-l0-tp4
trtmc run <bundle.bundle> --prompt "<text>" --output <output.png> [generation options]| Supported input or option | Requirement | Runtime behavior |
|---|---|---|
--prompt <TEXT> | Required | Text conditioning input. |
--output <PATH> | Optional | Output PNG path or directory. |
--num-images <N> | Optional | Generate an image batch from one prompt. |
--prompts-file <PATH> | Optional | Generate one image per line in a prompt file. |
--seed <N> | Optional | Set the sampling or diffusion seed. |
--num-steps <N> | Optional | Set the number of diffusion or flow-matching steps. |
--guidance-scale <F> | Optional | Set classifier-free guidance strength. |
--initial-latents-raw <PATH> | Optional | Load packed float32 initial latents. |
Code basis
Runtime provider: flux
include/trtmc/pipeline.hsrc/cli/main.cppsrc/runtime/models/flux/pipeline.cppsrc/runtime/models/flux/pipeline.htests/e2e/models/flux/manifests/flux-2-dev-fp8-l0-tp4.jsontests/e2e/models/flux/manifests/flux-2-dev-fp8-l0.jsontests/e2e/models/flux/manifests/flux-2-dev-fp8.jsontests/e2e/models/flux/manifests/flux-2-dev-l0-tp4.jsontests/e2e/models/flux/manifests/flux-2-dev-l0.jsontests/e2e/models/flux/manifests/flux-2-dev.jsontests/e2e/models/flux/manifests/flux-schnell-l0-batch2.jsontests/e2e/models/flux/manifests/flux-schnell-l0-cp4.jsontests/e2e/models/flux/manifests/flux-schnell-l0-tp4.jsontests/e2e/models/flux/manifests/flux-schnell-l0.jsontests/e2e/models/flux/manifests/flux-schnell.json
Bundle lifecycle commands
These commands apply to every declared recipe in this family; they do not add task inputs or model capabilities.
trtmc build
Build one exact checkpoint into a TensorRT-Model-Connect bundle.
trtmc build <hf-id> --output <bundle.bundle>trtmc inspect
Inspect bundle metadata, runtime identity, and packaged sections.
trtmc inspect <bundle.bundle>See the CLI Reference for all options and limitations.