跳转到主要内容
此节点旨在为 DPMPP_2M_SDE 模型生成采样器,支持根据指定的求解器类型、噪声水平和计算设备偏好创建样本。它抽象了采样器配置的复杂性,为使用自定义设置生成样本提供了简化的接口。

输入

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

输出

参数名描述数据类型
sampler输出为根据指定参数配置的采样器,可用于生成样本。SAMPLER
本文档由 AI 生成。如果您发现任何错误或有改进建议,欢迎贡献! 在 GitHub 上编辑