Diffusers is a library of state-of-the-art pretrained diffusion models for all of your generative AI needs and use cases.
The library provides three main classes.
import torch
from diffusers import FluxPipeline
pipeline = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=torch.bfloat16)
pipeline.enable_model_cpu_offload()
prompt = "A cat holding a sign that says hello world"
image = pipeline(prompt, guidance_scale=0.0, output_type="pil", num_inference_steps=4, max_sequence_length=256, generator=torch.Generator("cpu").manual_seed(33)).images[0]
image.save("flux-schnell.png")
In addition to these classes, Diffusers is invested in lowering the barrier for everyone. The library is optimized to run on memory-constrained hardware, accelerate inference on PyTorch, hardware (GPU/CPU/TPUs), and model accelerators.
Visit the documentation if you're interested in learning more. We're excited to see what you create with Diffusers! 🤗