![FLUX.1 [schnell] Grid](https://cnb.cool/ai-models/black-forest-labs/FLUX.1-schnell/-/git/raw/master/schnell_grid.jpeg)
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