huggingface diffusers

已跟踪

Diffusers:Hugging Face 提供的模块化扩展库,包含用于图像、音频及 3D 生成的预训练扩散模型与流水线。

作者 Hugging Face 开源时间 2022-05-30 最近提交 未知

Diffusers 是 Hugging Face 的旗舰库,提供用于图像、视频和音频生成的最先进扩散模型(基于 PyTorch)。它提供模块化的预训练模型和流水线工具箱,兼顾推理与训练需求,强调可用性和可定制性。

主要特性

  • 开箱即用的文本到图像、图像到图像、图像修复和视频生成流水线
  • 可替换的调度器和模块化模型组件,允许在采样质量和速度之间精细调节
  • 与 Hugging Face Hub 深度集成,可访问大量预训练检查点
  • 兼容多种主流硬件后端,支持可选的硬件特定优化
  • 流水线、调度器、模型和工具组件均可独立组合和扩展

使用场景

  • 在研究和创意应用中快速原型验证生成模型
  • 构建大规模图像和媒体生成的生产推理流水线
  • 使用自定义调度器和组件训练或微调扩散模型以满足特定需求
  • 通过统一的易用 API 实验最新的生成式 AI 技术

技术特点

  • 基于 Python 和 PyTorch 构建,开发者体验优先
  • 依托 Hugging Face Hub 生态系统进行模型发现与分发
  • 由活跃社区维护并提供丰富文档和频繁更新