🧨Diffusers

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.

1. The Pipeline class provides an easy and unified way to perform inference with many models.

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")

2. Diffusers also provides models and schedulers that you can mix and match to build or train your own diffusion systems.

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! 🤗