Model reference · open weights
FLUX.1-schnell is an open-weight image model from frankjoshua. AxForge deploys and operates it for you on dedicated EU-owned hardware — with the licence handled where one is required.
Available as managed deployment — configured and operated for you on dedicated EU hardware, quoted per deployment.
What it is
| Released by | frankjoshua |
|---|---|
| Type | Image models |
| Task | Text→image |
| Parameters (lead) | 11.9B |
| Runs with | diffusers |
| Released | 2024-11-12 |
| Popularity | 2k downloads / month |
| Licence | Open weights |
About
FLUX.1 [schnell] is a 12 billion parameter rectified flow transformer capable of generating images from text descriptions.
For more information, please read our blog post.
FLUX.1 [schnell] can generate high-quality images in only 1 to 4 steps.apache-2.0 licence, the model can be used for personal, scientific, and commercial purposes.We provide a reference implementation of FLUX.1 [schnell], as well as sampling code, in a dedicated github repository.
Developers and creatives looking to build on top of FLUX.1 [schnell] are encouraged to use this as a starting point.
The FLUX.1 models are also available via API from the following sources
FLUX.1 [pro])FLUX.1 [schnell] is also available in Comfy UI for local inference with a node-based workflow.
To use FLUX.1 [schnell] with the 🧨 diffusers python library, first install or upgrade diffusers
pip install -U diffusers
Then you can use FluxPipeline to run the model
import torch
from diffusers import FluxPipeline
pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=torch.bfloat16)
pipe.enable_model_cpu_offload() #save some VRAM by offloading the model to CPU. Remove this if you have enough GPU power
prompt = "A cat holding a sign that says hello world"
image = pipe(
prompt,
guidance_scale=0.0,
num_inference_steps=4,
max_sequence_length=256,
generator=torch.Generator("cpu").manual_seed(0)
).images[0]
image.save("flux-schnell.png")
To learn more check out the diffusers documentation
The model and its derivatives may not be used
From the published model card. Full card on the HuggingFace links in the sidebar.
Using it via the API
Once AxForge deploys frankjoshua-flux-1-schnell for you, it answers on the OpenAI-compatible API — the same base URL and keys as every other model. (frankjoshua-flux-1-schnell below is illustrative; you get the exact model name on deployment.)
$ curl -sS https://api.axforge.ai/v1/images/generations \
-H "Authorization: Bearer $AXFORGE_API_KEY" \
-H "Content-Type: application/json" \
-d '{"model":"frankjoshua-flux-1-schnell","prompt":"a red bicycle","size":"1024x1024"}'
Create an account — your API key is available in the console. 3M free tokens every 30 days with every new account.