diffusion

Jan 19 2026

diffusion

GitHub: https://github.com/mosaicml/diffusion

⭐ 星星數： 712

摘要： 這是一個基於MosaicML框架構建的擴散模型訓練和推理工具包，專注於圖像生成任務。

技術堆疊： PyTorch, Diffusers, Transformers, HuggingFace Hub, MosaicML, Hydra, Gradio, PEFT, xFormers

架構： 專案採用模組化設計，包含模型、資料集、調度器、推理和評估等獨立模組，使用Hydra進行配置管理，支援多種擴散模型訓練和微調技術。