Hub 文档

在 Hugging Face 使用 🧨 diffusers

Hugging Face's logo
加入 Hugging Face 社区

并获得增强型文档体验

开始使用

在 Hugging Face 中使用 🧨 diffusers

Diffusers 是用于生成图像、音频甚至分子 3D 结构的最先进预训练扩散模型的首选库。无论您是寻找简单的推理解决方案还是想要训练自己的扩散模型,Diffusers 都是支持这两种功能的模块化工具箱。该库的设计重点是可用性而不是性能,简单而不是易用,以及可定制性而不是抽象性。

在 Hub 中探索 Diffusers

在 Hub 上有超过 10,000 个与 diffusers 兼容的管道,您可以在 模型页面 的左侧过滤找到它们。扩散系统通常由多个组件组成,例如文本编码器、UNet、VAE 和调度器。即使它们不是独立模型,管道抽象也使它们易于用于推理或训练。

您可以找到适用于许多不同任务的扩散管道

如果您想在不下载它们的情况下测试这些模型,可以使用浏览器内小部件直接在浏览器中尝试这些模型!

使用现有管道

所有 diffusers 管道距离使用只有一行代码!要运行生成,建议始终从 DiffusionPipeline 开始

from diffusers import DiffusionPipeline

pipeline = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0")

如果您想加载特定的管道组件(例如 UNet),您可以通过以下方式进行

from diffusers import UNet2DConditionModel

unet = UNet2DConditionModel.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", subfolder="unet")

分享您的管道和模型

所有 管道类模型类调度器类 与 Hub 完全兼容。更具体地说,它们可以使用 from_pretrained() 方法轻松地从 Hub 加载,并可以使用 push_to_hub() 方法与他人共享。

有关更多详细信息,请查看 文档

其他资源

< > 更新 在 GitHub 上