← 返回 AI 开源导航

Diffusers

图像生成协议 Apache-2.0★ 34.8kPython

一、项目简介

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 复现;
  • 商用注意模型许可与内容合规。