跳转到主要内容
本文档由 AI 生成。如果您发现任何错误或有改进建议,欢迎贡献! 在 GitHub 上编辑
此节点旨在为 DPMPP_2M_SDE 模型生成采样器,支持根据指定的求解器类型、噪声水平和计算设备偏好创建样本。它抽象了采样器配置的复杂性,为使用自定义设置生成样本提供了简化的接口。

输入

参数名数据类型描述
solver_typeCOMBO[STRING]指定采样过程中使用的求解器类型,提供 ‘midpoint’ 和 ‘heun’ 两种选项。此选择会影响采样过程中应用的数值积分方法。
etaFLOAT确定数值积分中的步长,影响采样过程的粒度。数值越大表示步长越大。
s_noiseFLOAT控制采样过程中引入的噪声水平,影响生成样本的变异性。
noise_deviceCOMBO[STRING]指示执行噪声生成过程的计算设备(‘gpu’ 或 ‘cpu’),影响性能和效率。

输出

参数名数据类型描述
samplerSAMPLER输出为根据指定参数配置的采样器,可用于生成样本。