guided-diffusion-keras

用 Keras 玩转文本生成图像，轻松训练扩散模型

guided-diffusion-keras 是一个基于 Keras 框架实现的扩散模型库，专注于文本到图像的生成任务。它提供了从零开始构建和训练扩散模型的完整代码，包括条件生成（如类别条件、文本条件）的支持，并兼容 Stable Diffusion 等主流模型的架构。该项目旨在降低扩散模型的研究和开发门槛，让开发者无需依赖 PyTorch 即可在 TensorFlow/Keras 生态中实现和实验最新的生成模型。核心能力包括：扩散模型的训练与采样、引导（guidance）机制、文本编码与融合，以及灵活的模型配置。对于希望深入理解扩散模型原理或需要在生产环境中部署 Keras 模型的团队，该项目提供了一个轻量级且可定制的起点。