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 [klein] is an ultra-fast, distilled image model. It unifies image generation and editing in a single model, delivering state-of-the-art quality enabling interactive workflows, real-time previews, and latency-critical applications.
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-Klein-4b
client.chat.completions.create(
model="black-forest-labs/flux-2-klein-4b",
messages=[{"role":"user","content":"hello"}],
)