跳转到主要内容
本文档由 AI 生成。如果您发现任何错误或有改进建议,欢迎贡献! 在 GitHub 上编辑
SamplerSASolver 节点为扩散模型实现了一种自定义采样算法。它采用预测器-校正器方法,通过可配置的阶数设置和随机微分方程(SDE)参数,从输入模型中生成样本。

输入参数

参数数据类型必填取值范围描述
modelMODEL-用于采样的扩散模型
etaFLOAT0.0 - 10.0控制步长缩放因子(默认值:1.0)
sde_start_percentFLOAT0.0 - 1.0SDE 采样的起始百分比(默认值:0.2)
sde_end_percentFLOAT0.0 - 1.0SDE 采样的结束百分比(默认值:0.8)
s_noiseFLOAT0.0 - 100.0控制采样过程中添加的噪声量(默认值:1.0)
predictor_orderINT1 - 6求解器中预测器组件的阶数(默认值:3)
corrector_orderINT0 - 6求解器中校正器组件的阶数(默认值:4)
use_peceBOOLEAN-启用或禁用 PECE(预测-评估-校正-评估)方法
simple_order_2BOOLEAN-启用或禁用简化的二阶计算

输出结果

输出名称数据类型描述
samplerSAMPLER已配置的采样器对象,可用于扩散模型