huggingface/diffusers
🤗 Diffusers: State-of-the-art diffusion models for image, video, and audio generation in PyTorch.
What it solves
Diffusers is a library designed to simplify the use and training of state-of-the-art pretrained diffusion models. It provides a modular toolbox that allows users to generate images, audio, and 3D molecular structures without needing to build complex systems from scratch.
How it works
The library is built around three core components:
- Diffusion Pipelines: High-level APIs that allow users to run inference with just a few lines of code.
- Noise Schedulers: Interchangeable components that control the diffusion speed and the quality of the output.
- Pretrained Models: Modular building blocks that can be combined with schedulers to create custom end-to-end diffusion systems.
Who it’s for
It is intended for developers and researchers who want to implement diffusion-based generation tasks—such as text-to-image, image-to-image, inpainting, and super-resolution—either through simple inference or by training their own models.
Highlights
- Broad Modality Support: Supports generating images, audio, and 3D molecular structures.
- Modular Design: Prioritizes customizability and usability over strict abstractions.
- Extensive Model Hub: Access to over 30,000 pretrained checkpoints via the Hugging Face Hub.
- Optimization Tools: Includes guides and tools for reducing memory consumption and increasing inference speed.
Related
- Dispatch
- Project
- Project
- Project
- Project