Access FLUX image generation models through the Black Forest Labs integration.
Setup
Add Provider
mixtrain provider add blackforestlabsOr via SDK:
from mixtrain import MixClient
client = MixClient()
client.create_model_provider(
provider_type="blackforestlabs",
secrets={"api_key": "..."}
)Once added, Black Forest Labs models are available via mixtrain model catalog.
Usage
from mixtrain import get_model
model = get_model("flux-pro")
result = model.run({
"prompt": "A photorealistic portrait of a robot artist",
"width": 1024,
"height": 1024
})Available Models
| Model ID | Description |
|---|---|
flux-pro-1.1 | Professional quality model |
flux-dev | Development/testing model |
flux-schnell | Fast generation model |
Configuration Options
| Parameter | Description |
|---|---|
prompt | Text description of the image |
width | Output width in pixels |
height | Output height in pixels |
steps | Number of inference steps |
seed | Random seed for reproducibility |
Getting an API Key
- Sign up at blackforestlabs.ai
- Navigate to API settings
- Generate an API key
- Add it to Mixtrain using the setup steps above