Diffusers
一、项目简介
Diffusers 是 Hugging Face 的扩散模型库,把 Stable Diffusion 等复杂生成流程封装为可组合的模块(模型、调度器、管线),是二次开发文生图应用的主流底座。
二、核心特性
- 丰富管线:文生图、图生图、局部重绘、超分、视频生成;
- 调度器可换:DDIM、DPM++、Euler 等,直接影响速度与质量;
- 模型可插拔:支持 SD1.5 / SDXL / Flux,以及 LoRA、ControlNet;
- 训练支持:Dreambooth、LoRA、Textual Inversion 脚本;
- 生态协同:与 Transformers、Accelerate 无缝配合。
三、适用场景
- 文生图应用开发与定制;
- 微调专属风格模型;
- 扩散采样算法研究。
四、快速上手
pip install diffusers transformers accelerate
from diffusers import StableDiffusionPipeline
pipe = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5")
pipe.to("cuda").save("output.png")
五、注意事项
- 显存消耗大,可启用
enable_attention_slicing()与 fp16; - 生成结果有随机性,需固定 seed 复现;
- 商用注意模型许可与内容合规。