Skip to main content
DeepInfra supports the OpenAI-compatible image generation API. The default model is FLUX Schnell. The endpoint is:
POST https://api.deepinfra.com/v1/openai/images/generations
Browse all text-to-image models.

Example

import io
import base64
from PIL import Image
from openai import OpenAI

client = OpenAI(
    api_key="$DEEPINFRA_TOKEN",
    base_url="https://api.deepinfra.com/v1/openai"
)

response = client.images.generate(
    prompt="A photo of an astronaut riding a horse on Mars.",
    size="1024x1024",
    n=1,
)

b64_json = response.data[0].b64_json
image_bytes = base64.b64decode(b64_json)
image = Image.open(io.BytesIO(image_bytes))
image.save("output.png")

Supported parameters

ParameterNotes
promptText description of the image
modelDefaults to FLUX Schnell
sizeImage dimensions (e.g., "1024x1024")
nNumber of images to generate
response_formatOnly b64_json supported
quality, styleAvailable for compatibility only

LoRA image adapters

You can also use custom LoRA adapters for image generation — see LoRA for Image Generation.

Tutorial

For a deeper example including advanced options, see the Stable Diffusion tutorial.