本文档由 AI 生成。如果您发现任何错误或有改进建议,欢迎贡献! 在 GitHub 上编辑此节点旨在为 DPMPP_2M_SDE 模型生成采样器,支持根据指定的求解器类型、噪声水平和计算设备偏好创建样本。它抽象了采样器配置的复杂性,为使用自定义设置生成样本提供了简化的接口。
输入
| 参数名 | 数据类型 | 描述 |
|---|---|---|
solver_type | COMBO[STRING] | 指定采样过程中使用的求解器类型,提供 ‘midpoint’ 和 ‘heun’ 两种选项。此选择会影响采样过程中应用的数值积分方法。 |
eta | FLOAT | 确定数值积分中的步长,影响采样过程的粒度。数值越大表示步长越大。 |
s_noise | FLOAT | 控制采样过程中引入的噪声水平,影响生成样本的变异性。 |
noise_device | COMBO[STRING] | 指示执行噪声生成过程的计算设备(‘gpu’ 或 ‘cpu’),影响性能和效率。 |
输出
| 参数名 | 数据类型 | 描述 |
|---|---|---|
sampler | SAMPLER | 输出为根据指定参数配置的采样器,可用于生成样本。 |