昇腾社区首页
中文
注册

类说明

OpenSora采样器,用于在扩散模型中管理时间步和噪声的添加,继承自DiffusionScheduler,该函数在MindIE SD仓的路径为:mindiesd/schedulers/rectified_flow.py。

成员

成员名称

描述

__init__

初始化RFlowScheduler类的实例。

add_noise

在原始样本中添加噪声。

step

更新噪声。