模型

🤗 Diffusers 提供了用于流行算法和模块的预训练模型，以创建自定义扩散系统。模型的主要功能是根据分布 \(p_{\theta}(x_{t-1}|x_{t})\) 对输入样本进行去噪。

所有模型都基于 [ModelMixin] 类构建，该类是一个 torch.nn.Module，提供了保存和加载模型的基本功能，既可以在本地进行，也可以从 Hugging Face Hub 加载。