Pricing per million tokens, context window, capabilities — pulled from each provider's public docs. All 2 are available via the same AIgateway OpenAI-compatible endpoint; flip the model string to switch.
FLUX.1 [schnell] is a 12 billion parameter rectified flow transformer capable of generating images from text descriptions.
FLUX.2 [flex] is Black Forest Labs' fine-grained control variant of FLUX.2 — exposes tunable inference steps, guidance, and prompt upsampling for typography-heavy and production workflows.
from openai import OpenAI
client = OpenAI(
base_url="https://api.aigateway.sh/v1",
api_key="sk-aig-...",
)
# Flux-1-Schnell
client.chat.completions.create(
model="black-forest-labs/flux-1-schnell",
messages=[{"role":"user","content":"hello"}],
)
# FLUX.2 [Flex]
client.chat.completions.create(
model="black-forest-labs/flux-2-flex",
messages=[{"role":"user","content":"hello"}],
)